summaryrefslogtreecommitdiff
path: root/sphinx/pycode/parser.py
diff options
context:
space:
mode:
authorTakeshi KOMIYA <i.tkomiya@gmail.com>2020-04-26 20:21:19 +0900
committerTakeshi KOMIYA <i.tkomiya@gmail.com>2020-04-26 20:22:17 +0900
commit17f64e3829059873401fd2038af8eceae7a357fb (patch)
tree35335c45451efff3345f62db50fdc776311559c8 /sphinx/pycode/parser.py
parent02b8547a96630ca9ff3a28709ff793eaa79c6402 (diff)
downloadsphinx-git-17f64e3829059873401fd2038af8eceae7a357fb.tar.gz
refactor: pycode: visit_Import
Diffstat (limited to 'sphinx/pycode/parser.py')
-rw-r--r--sphinx/pycode/parser.py10
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."""