fix #2: now works with async defs

Author: Илья Лебедев

flake8_expression_complexity/__init__.py

@@ -1 +1 @@
-__version__ = '0.0.3'
+__version__ = '0.0.4'

flake8_expression_complexity/utils/ast.py

@@ -9,7 +9,8 @@ def iterate_over_expressions(node: ast.AST) -> Iterable[ast.AST]:
         ((ast.For, ), lambda n: n.iter),
     ]
     nodes_with_subnodes = (
-        ast.FunctionDef, ast.If, ast.For, ast.Module,
+        ast.FunctionDef, ast.AsyncFunctionDef,
+        ast.If, ast.For, ast.Module,
         ast.ClassDef, ast.Try, ast.With, ast.While,
     )
     for bases, subnodes_getter in additionals_subnodes_info:

tests/test_files/long_expressions.py

@@ -23,16 +23,20 @@ class Sum:
     pass
 
 
-if (
-    (user and user.is_authorized)
-    and user.subscriptions.filter(start_date__lt=today, end_date__gt=today).exists()
-    and (
-        user.total_credits_added
-        - Check.objects.filter(user=user).aggregate(Sum('price'))['check__sum']
-    )
-    and UserAction.objects.filter(user=user).last().datetime > today - datetime.timedelta(days=10)
-):
-    pass
+async def foo():
+    if (
+        (user and user.is_authorized)
+        and user.subscriptions.filter(start_date__lt=today, end_date__gt=today).exists()
+        and (
+            user.total_credits_added
+            - Check.objects.filter(user=user).aggregate(Sum('price'))['check__sum']
+        )
+        and (
+            UserAction.objects.filter(user=user).last().datetime
+            > today - datetime.timedelta(days=10)
+        )
+    ):
+        pass
 
 
 weird_container = []