From f78a35afb76c38f4141714130bbd1a35ee430ea1 Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Wed, 18 Aug 2021 09:48:25 +0200 Subject: Use alias for astroid.nodes 02 (#4863) * Use alias for astroid nodes * Resolve name conflicts --- pylint/checkers/similar.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) (limited to 'pylint/checkers/similar.py') diff --git a/pylint/checkers/similar.py b/pylint/checkers/similar.py index b44081924..e9792ebeb 100644 --- a/pylint/checkers/similar.py +++ b/pylint/checkers/similar.py @@ -64,7 +64,7 @@ from typing import ( ) import astroid -from astroid.node_classes import NodeNG +from astroid import nodes from pylint.checkers import BaseChecker, MapReduceMixin, table_lines_from_stats from pylint.interfaces import IRawChecker @@ -576,7 +576,7 @@ def stripped_lines( tree = astroid.parse("".join(lines)) if ignore_imports: node_is_import_by_lineno = ( - (node.lineno, isinstance(node, (astroid.Import, astroid.ImportFrom))) + (node.lineno, isinstance(node, (nodes.Import, nodes.ImportFrom))) for node in tree.body ) line_begins_import = { @@ -588,16 +588,18 @@ def stripped_lines( current_line_is_import = False if ignore_signatures: - def _get_functions(functions: List[NodeNG], tree: NodeNG) -> List[NodeNG]: + def _get_functions( + functions: List[nodes.NodeNG], tree: nodes.NodeNG + ) -> List[nodes.NodeNG]: """Recursively get all functions including nested in the classes from the tree.""" for node in tree.body: - if isinstance(node, (astroid.FunctionDef, astroid.AsyncFunctionDef)): + if isinstance(node, (nodes.FunctionDef, nodes.AsyncFunctionDef)): functions.append(node) if isinstance( node, - (astroid.ClassDef, astroid.FunctionDef, astroid.AsyncFunctionDef), + (nodes.ClassDef, nodes.FunctionDef, nodes.AsyncFunctionDef), ): _get_functions(functions, node) -- cgit v1.2.1