diff --git a/astroid/nodes/_base_nodes.py b/astroid/nodes/_base_nodes.py index b39489d958..47c27d38b4 100644 --- a/astroid/nodes/_base_nodes.py +++ b/astroid/nodes/_base_nodes.py @@ -196,9 +196,9 @@ def _get_yield_nodes_skip_lambdas(self): yield from child_node._get_yield_nodes_skip_lambdas() @cached_property - def _get_assign_nodes(self): + def _assign_nodes_in_scope(self) -> list[nodes.Assign]: children_assign_nodes = ( - child_node._get_assign_nodes + child_node._assign_nodes_in_scope for block in self._multi_line_blocks for child_node in block ) diff --git a/astroid/nodes/node_classes.py b/astroid/nodes/node_classes.py index 33a2768fdf..35547aa84f 100644 --- a/astroid/nodes/node_classes.py +++ b/astroid/nodes/node_classes.py @@ -1272,8 +1272,8 @@ def get_children(self): yield self.value @cached_property - def _get_assign_nodes(self): - return [self] + self.value._get_assign_nodes + def _assign_nodes_in_scope(self) -> list[nodes.Assign]: + return [self] + self.value._assign_nodes_in_scope def _get_yield_nodes_skip_lambdas(self): yield from self.value._get_yield_nodes_skip_lambdas() diff --git a/astroid/nodes/node_ng.py b/astroid/nodes/node_ng.py index a3ed22d83c..b85707dafe 100644 --- a/astroid/nodes/node_ng.py +++ b/astroid/nodes/node_ng.py @@ -576,7 +576,7 @@ def nodes_of_class( # type: ignore[misc] # mypy doesn't correctly recognize the yield from child_node.nodes_of_class(klass, skip_klass) @cached_property - def _get_assign_nodes(self): + def _assign_nodes_in_scope(self) -> list[nodes.Assign]: return [] def _get_name_nodes(self): diff --git a/astroid/nodes/scoped_nodes/scoped_nodes.py b/astroid/nodes/scoped_nodes/scoped_nodes.py index d0d269e634..5349dfa824 100644 --- a/astroid/nodes/scoped_nodes/scoped_nodes.py +++ b/astroid/nodes/scoped_nodes/scoped_nodes.py @@ -1434,7 +1434,7 @@ def extra_decorators(self) -> list[node_classes.Call]: return [] decorators: list[node_classes.Call] = [] - for assign in frame._get_assign_nodes: + for assign in frame._assign_nodes_in_scope: if isinstance(assign.value, node_classes.Call) and isinstance( assign.value.func, node_classes.Name ): @@ -3067,9 +3067,9 @@ def get_children(self): yield from self.body @cached_property - def _get_assign_nodes(self): + def _assign_nodes_in_scope(self): children_assign_nodes = ( - child_node._get_assign_nodes for child_node in self.body + child_node._assign_nodes_in_scope for child_node in self.body ) return list(itertools.chain.from_iterable(children_assign_nodes))