Merge pull request #1313 from CXuesong/master

get_module_names should include top-level async functions when all_scopes=False.
2019-04-13 00:43:58 +02:00
parent e843c6108d 2724ac9e07
commit b2f6758a9c
2 changed files with 40 additions and 0 deletions
@@ -190,6 +190,9 @@ def get_module_names(module, all_scopes):
                # but that would be a big change that could break type inference, whereas for now
                # this discrepancy looks like only a problem for "get_module_names".
                parent_scope = parent_scope.parent
+            # async functions have an extra wrapper. Strip it.
+            if parent_scope and parent_scope.type == 'async_stmt':
+                parent_scope = parent_scope.parent
            return parent_scope in (module, None)

        names = [n for n in names if is_module_scope_name(n)]