FIX #7: add async with support and with variables nodes validation

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

flake8_expression_complexity/__init__.py

@@ -1 +1 @@
-__version__ = '0.0.5'
+__version__ = '0.0.6'

flake8_expression_complexity/utils/ast.py

@@ -5,17 +5,20 @@
 
 def iterate_over_expressions(node: ast.AST) -> Iterable[ast.AST]:
     additionals_subnodes_info: List[Tuple[Tuple, Callable]] = [
-        ((ast.If, ast.While), lambda n: n.test),
-        ((ast.For, ), lambda n: n.iter),
+        ((ast.If, ast.While), lambda n: [n.test]),
+        ((ast.For, ), lambda n: [n.iter]),
+        ((ast.With, ast.AsyncWith), lambda n: [s.context_expr for s in n.items]),
     ]
     nodes_with_subnodes = (
         ast.FunctionDef, ast.AsyncFunctionDef,
         ast.If, ast.For, ast.Module,
-        ast.ClassDef, ast.Try, ast.With, ast.While,
+        ast.ClassDef, ast.Try, ast.With, ast.AsyncWith,
+        ast.While,
     )
     for bases, subnodes_getter in additionals_subnodes_info:
         if isinstance(node, bases):
-            yield subnodes_getter(node)
+            for subitem in subnodes_getter(node):
+                yield subitem
     nodes_to_iter = (
         _get_try_node_children(node)
         if isinstance(node, ast.Try)

requirements_dev.txt

@@ -2,7 +2,7 @@ mypy==0.740
 mypy-extensions==0.4.1
 typing==3.6.4
 typing-extensions==3.7.4
-flake8==3.7.8
+flake8==3.7.9
 flake8-annotations-complexity==0.0.2
 flake8-blind-except==0.1.1
 flake8-broken-line==0.1.1

setup.cfg

@@ -3,7 +3,7 @@ ignore = D, W503
 max-line-length = 100
 use_class_attributes_order_strict_mode = True
 max_function_length = 50
-max-complexity = 6
+max-complexity = 7
 
 
 [mypy]

tests/test_complexity.py

@@ -3,4 +3,4 @@
 
 def test_fails():
     errors = run_validator_for_test_file('long_expressions.py', max_expression_compexity=3)
-    assert len(errors) == 3
+    assert len(errors) == 4

tests/test_files/long_expressions.py

@@ -41,8 +41,13 @@ async def foo():
 
 
 async def bar():
-    return 'bar'
+    async with foo:
+        return 'bar'
 
 
 weird_container = []
 sublist = weird_container[10:datetime.datetime.today(), None]
+
+
+with weird_container[10:str(datetime.datetime.today().date())[:100], None]:
+    pass