mirror of
https://github.com/sphinx-doc/sphinx.git
synced 2025-02-25 18:55:22 -06:00
refactor: pycode: visit_Import
This commit is contained in:
parent
02b8547a96
commit
17f64e3829
@ -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."""
|
||||||
|
Loading…
Reference in New Issue
Block a user