Changes batching to allow all iterables

influxdb/client.py

@@ -7,6 +7,7 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 
+import itertools
 import json
 import socket
 import requests
@@ -411,8 +412,10 @@ def write_points(self,
                                       tags=tags, protocol=protocol)
 
     def _batches(self, iterable, size):
-        for i in xrange(0, len(iterable), size):
-            yield iterable[i:i + size]
+        iterator = iter(iterable)
+        while True:
+            batch_iterator = itertools.islice(iterator, size)
+            yield itertools.chain([next(batch_iterator)], batch_iterator)
 
     def _write_points(self,
                       points,