diff options
author | Takeshi KOMIYA <i.tkomiya@gmail.com> | 2020-04-26 14:21:19 +0300 |
---|---|---|
committer | Takeshi KOMIYA <i.tkomiya@gmail.com> | 2020-04-26 14:22:17 +0300 |
commit | 17f64e3829059873401fd2038af8eceae7a357fb (patch) | |
tree | 35335c45451efff3345f62db50fdc776311559c8 /sphinx/pycode | |
parent | 02b8547a96630ca9ff3a28709ff793eaa79c6402 (diff) |
refactor: pycode: visit_Import
Diffstat (limited to 'sphinx/pycode')
-rw-r--r-- | sphinx/pycode/parser.py | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/sphinx/pycode/parser.py b/sphinx/pycode/parser.py index c8a522fb6..3ef1c4605 100644 --- a/sphinx/pycode/parser.py +++ b/sphinx/pycode/parser.py @@ -280,18 +280,12 @@ class VariableCommentPicker(ast.NodeVisitor): def visit_Import(self, node: ast.Import) -> None: """Handles Import node and record it to definition orders.""" for name in node.names: - if name.asname: - self.add_entry(name.asname) - else: - self.add_entry(name.name) + self.add_entry(name.asname or name.name) def visit_ImportFrom(self, node: ast.ImportFrom) -> None: """Handles Import node and record it to definition orders.""" for name in node.names: - if name.asname: - self.add_entry(name.asname) - else: - self.add_entry(name.name) + self.add_entry(name.asname or name.name) def visit_Assign(self, node: ast.Assign) -> None: """Handles Assign node and pick up a variable comment.""" |