From a90b9a1deadc8f6b73af7a0dae2cf3c23ded57b9 Mon Sep 17 00:00:00 2001 From: Naoto Mizuno Date: Sat, 19 Sep 2020 21:13:27 +0900 Subject: [PATCH] Fix expander --- atcoder/__main__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/atcoder/__main__.py b/atcoder/__main__.py index 9ce7b14..03856c0 100644 --- a/atcoder/__main__.py +++ b/atcoder/__main__.py @@ -35,7 +35,8 @@ def iter_child_nodes( if re.match(r'^atcoder\.?', name.name): import_info = ImportInfo(node.lineno, node.end_lineno) elif isinstance(node, ast.ImportFrom): - import_info = ImportInfo(node.lineno, node.end_lineno, node.module) + if re.match(r'^atcoder\.?', node.module): + import_info = ImportInfo(node.lineno, node.end_lineno, node.module) for child in ast.iter_child_nodes(node): result += iter_child_nodes(child, import_info)