Merge pull request #3 from Stafil0/master

Added await support

Author: Melevir

flake8_expression_complexity/utils/complexity.py

@@ -24,6 +24,7 @@ def get_complexity_increase_for_node_type(node_type_sid: str) -> float:
         'assign': 1,
         'featured_assign': 1,
         'call': .5,
+        'await': .5,
         'sized': 1,
         'dict': 1,
         'dict_comprehension': 1,
@@ -59,6 +60,7 @@ def get_expression_part_info(node: ast.AST) -> Mapping[str, Any]:
         (ast.Assign, 'assign'),
         ((ast.AugAssign, ast.AnnAssign), 'featured_assign'),
         (ast.Call, 'call'),
+        (ast.Await, 'await'),
         ((ast.List, ast.Set, ast.Tuple), 'sized'),
         (ast.Dict, 'dict'),
         (ast.DictComp, 'dict_comprehension'),
@@ -108,6 +110,7 @@ def _get_sub_nodes(node: Any, node_type_sid: str) -> List[ast.AST]:
         'assign': lambda n: node.targets + [node.value],
         'featured_assign': lambda n: [n.target, n.value],
         'call': lambda n: node.args + [n.func],
+        'await': lambda n: [node.value],
         'sized': lambda n: node.elts,
         'dict': lambda n: itertools.chain(node.keys, node.values),
         'dict_comprehension': lambda n: node.generators + [n.key, n.value],

tests/test_files/long_expressions.py

@@ -36,8 +36,13 @@ async def foo():
             > today - datetime.timedelta(days=10)
         )
     ):
+        await bar()
         pass
 
 
+async def bar():
+    return 'bar'
+
+
 weird_container = []
 sublist = weird_container[10:datetime.datetime.today(), None]