[FIX] changes for flake8 compliance

Author: auphofBSF

influxdb/line_protocol.py

@@ -9,11 +9,9 @@
 from datetime import datetime
 from numbers import Integral
 
-from pytz import UTC
-from dateutil.parser import parse
 from six import iteritems, binary_type, text_type, integer_types, PY2
 
-import pandas as pd # Provide for ns timestamps
+import pandas as pd  # Provide for ns timestamps
 
 EPOCH = pd.Timestamp(0, tz='UTC')
 
@@ -25,15 +23,15 @@ def _convert_timestamp(timestamp, precision=None):
     if isinstance(_get_unicode(timestamp), text_type):
         timestamp = pd.Timestamp(timestamp)
 
-    if isinstance(timestamp, datetime): # change to pandas.Timestamp
+    if isinstance(timestamp, datetime):  # change to pandas.Timestamp
         if not timestamp.tzinfo:
             timestamp = pd.Timestamp(timestamp, tz='UTC')
         else:
             timestamp = pd.Timestamp(timestamp)
 
     if isinstance(timestamp, pd._libs.tslib.Timestamp):
-        if not timestamp.tzinfo: # set to UTC for time since EPOCH
-            timestamp = pd.Timestamp(timestamp, tz = 'UTC')
+        if not timestamp.tzinfo:  # set to UTC for time since EPOCH
+            timestamp = pd.Timestamp(timestamp, tz='UTC')
         else:
             timestamp = timestamp.astimezone('UTC')
 

influxdb/tests/test_line_protocol.py

@@ -13,6 +13,7 @@
 from influxdb import line_protocol
 import pandas as pd
 
+
 class TestLineProtocol(unittest.TestCase):
     """Define the LineProtocol test object."""
 
@@ -55,43 +56,44 @@ def test_timezone(self):
         eastern = berlin.astimezone(timezone('US/Eastern'))
         # pandas ns timestamp tests
         pddt = pd.Timestamp('2009-11-10 23:00:00.123456789')
-        pdutc = pd.Timestamp(pddt, tz = 'UTC')
+        pdutc = pd.Timestamp(pddt, tz='UTC')
         pdberlin = pdutc.astimezone('Europe/Berlin')
         pdeastern = pdberlin.astimezone('US/Eastern')
 
-        data = {  "points": [
-                {"measurement": "A", "fields": {"val": 1},
-                "time": 0},
-                #string representations
-                {"measurement": "A", "fields": {"val": 1},
-                "time": "2009-11-10T23:00:00.123456Z"},   # String version for datetime
-                {"measurement": "A", "fields": {"val": 1},
-                "time": "2009-11-10 23:00:00.123456789"},   # String version for pandas ns timestamp
-                # datetime 
-                {"measurement": "A", "fields": {"val": 1}, "time": dt},
-                {"measurement": "A", "fields": {"val": 1}, "time": utc},
-                {"measurement": "A", "fields": {"val": 1}, "time": berlin},
-                {"measurement": "A", "fields": {"val": 1}, "time": eastern},
-                # pandas timestamp
-                {"measurement": "A", "fields": {"val": 1}, "time": pddt},
-                {"measurement": "A", "fields": {"val": 1}, "time": pdutc},
-                {"measurement": "A", "fields": {"val": 1}, "time": pdberlin},
-                {"measurement": "A", "fields": {"val": 1}, "time": pdeastern},
-        ]}
-        
+        data = {"points": [
+            {"measurement": "A", "fields": {"val": 1}, "time": 0},
+            # string representations
+            # String version for datetime
+            {"measurement": "A", "fields": {"val": 1},
+             "time": "2009-11-10T23:00:00.123456Z"},
+            # String version for pandas ns timestamp
+            {"measurement": "A", "fields": {"val": 1},
+             "time": "2009-11-10 23:00:00.123456789"},
+            # datetime
+            {"measurement": "A", "fields": {"val": 1}, "time": dt},
+            {"measurement": "A", "fields": {"val": 1}, "time": utc},
+            {"measurement": "A", "fields": {"val": 1}, "time": berlin},
+            {"measurement": "A", "fields": {"val": 1}, "time": eastern},
+            # pandas timestamp
+            {"measurement": "A", "fields": {"val": 1}, "time": pddt},
+            {"measurement": "A", "fields": {"val": 1}, "time": pdutc},
+            {"measurement": "A", "fields": {"val": 1}, "time": pdberlin},
+            {"measurement": "A", "fields": {"val": 1}, "time": pdeastern},
+        ]
+        }
 
         self.assertEqual(
             line_protocol.make_lines(data),
             '\n'.join([
                 'A val=1i 0',
                 'A val=1i 1257894000123456000',
                 'A val=1i 1257894000123456789',
-                #datetime results
+                # datetime results
                 'A val=1i 1257894000123456000',
                 'A val=1i 1257894000123456000',
                 'A val=1i 1257890400123456000',
                 'A val=1i 1257890400123456000',
-                #pandas ns timestamp results
+                # pandas ns timestamp results
                 'A val=1i 1257894000123456789',
                 'A val=1i 1257894000123456789',
                 'A val=1i 1257894000123456789',