Import itertools functions directly

Jamie Hewland · Jamie Hewland · commit d6e7d1cf57b6 · 2019-09-10T16:21:36.000+02:00
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -6,9 +6,9 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 
-import itertools
 import time
 import random
+from itertools import chain, islice
 
 import json
 import socket
@@ -553,8 +553,8 @@ def _batches(iterable, size):
             except StopIteration:
                 return  # ...so that we can stop if there isn't one
             # Otherwise, lazily slice the rest of the batch
-            rest = itertools.islice(iterator, size - 1)
-            yield itertools.chain(head, rest)
+            rest = islice(iterator, size - 1)
+            yield chain(head, rest)
 
     def _write_points(self,
                       points,
