refactor: pycode: visit_Import

This commit is contained in:
Takeshi KOMIYA 2020-04-26 20:21:19 +09:00
parent 02b8547a96
commit 17f64e3829

View File

@ -280,18 +280,12 @@ class VariableCommentPicker(ast.NodeVisitor):
def visit_Import(self, node: ast.Import) -> None: def visit_Import(self, node: ast.Import) -> None:
"""Handles Import node and record it to definition orders.""" """Handles Import node and record it to definition orders."""
for name in node.names: for name in node.names:
if name.asname: self.add_entry(name.asname or name.name)
self.add_entry(name.asname)
else:
self.add_entry(name.name)
def visit_ImportFrom(self, node: ast.ImportFrom) -> None: def visit_ImportFrom(self, node: ast.ImportFrom) -> None:
"""Handles Import node and record it to definition orders.""" """Handles Import node and record it to definition orders."""
for name in node.names: for name in node.names:
if name.asname: self.add_entry(name.asname or name.name)
self.add_entry(name.asname)
else:
self.add_entry(name.name)
def visit_Assign(self, node: ast.Assign) -> None: def visit_Assign(self, node: ast.Assign) -> None:
"""Handles Assign node and pick up a variable comment.""" """Handles Assign node and pick up a variable comment."""