fix datetime module detection

pjknkda · Apr 25, 2019 · b8894e4 · b8894e4
1 parent 90e1144
commit b8894e4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/.flake8 b/.flake8
diff --git a/flake8_datetimez.py b/flake8_datetimez.py
@@ -45,10 +45,17 @@ def __init__(self):
         self.errors = []
 
     def visit_Call(self, node):
-        is_datetime_class = (isinstance(node.func.value, ast.Name)
+        # ex: `datetime.something()``
+        is_datetime_class = (isinstance(node.func, ast.Attribute)
+                             and isinstance(node.func.value, ast.Name)
                              and node.func.value.id == 'datetime')
-        is_datetime_module_n_class = (isinstance(node.func.value, ast.Attribute)
-                                      and node.func.value.attr == 'datetime')
+
+        # ex: `datetime.datetime.something()``
+        is_datetime_module_n_class = (isinstance(node.func, ast.Attribute)
+                                      and isinstance(node.func.value, ast.Attribute)
+                                      and node.func.value.attr == 'datetime'
+                                      and isinstance(node.func.value.value, ast.Name)
+                                      and node.func.value.value.id == 'datetime')
 
         if is_datetime_class or is_datetime_module_n_class:
             if node.func.attr == 'utcnow':
@@ -109,8 +116,8 @@ def visit_Call(self, node):
         self.generic_visit(node)
 
 
-error = namedtuple('error', ['lineno', 'col', 'message', 'type', 'vars'])
-Error = partial(partial, error, type=DateTimeZChecker, vars=())
+error = namedtuple('error', ['lineno', 'col', 'message', 'type'])
+Error = partial(partial, error, type=DateTimeZChecker)
 
 
 DTZ001 = Error(