Add future argument to all NodeNG.statement() calls

astroid/brain/brain_dataclasses.py

@@ -304,7 +304,7 @@ def _looks_like_dataclass_field_call(node: Call, check_scope: bool = True) -> bo
     If check_scope is False, skips checking the statement and body.
     """
     if check_scope:
-        stmt = node.statement()
+        stmt = node.statement(future=True)
         scope = stmt.scope()
         if not (
             isinstance(stmt, AnnAssign)

astroid/brain/brain_namedtuple_enum.py

@@ -365,7 +365,7 @@ def infer_enum_class(node):
             if any(not isinstance(value, nodes.AssignName) for value in values):
                 continue
 
-            stmt = values[0].statement()
+            stmt = values[0].statement(future=True)
             if isinstance(stmt, nodes.Assign):
                 if isinstance(stmt.targets[0], nodes.Tuple):
                     targets = stmt.targets[0].itered()

astroid/mixins.py

@@ -46,7 +46,7 @@ class FilterStmtsMixin:
 
     def _get_filtered_stmts(self, _, node, _stmts, mystmt):
         """method used in _filter_stmts to get statements and trigger break"""
-        if self.statement() is mystmt:
+        if self.statement(future=True) is mystmt:
             # original node's statement is the assignment, only keep
             # current node (gen exp, list comp)
             return [node], True
@@ -64,7 +64,7 @@ def _get_filtered_stmts(self, lookup_node, node, _stmts, mystmt):
         """method used in filter_stmts"""
         if self is mystmt:
             return _stmts, True
-        if self.statement() is mystmt:
+        if self.statement(future=True) is mystmt:
             # original node's statement is the assignment, only keep
             # current node (gen exp, list comp)
             return [node], True

astroid/nodes/node_classes.py

@@ -54,6 +54,7 @@
     InferenceError,
     NoDefault,
     ParentMissingError,
+    StatementMissing,
 )
 from astroid.manager import AstroidManager
 from astroid.nodes.const import OP_PRECEDENCE
@@ -387,7 +388,7 @@ def ilookup(self, name):
         return _infer_stmts(stmts, context, frame)
 
     def _get_filtered_node_statements(self, nodes):
-        statements = [(node, node.statement()) for node in nodes]
+        statements = [(node, node.statement(future=True)) for node in nodes]
         # Next we check if we have ExceptHandlers that are parent
         # of the underlying variable, in which case the last one survives
         if len(statements) > 1 and all(
@@ -437,10 +438,18 @@ def _filter_stmts(self, stmts, frame, offset):
             #
             # def test(b=1):
             #     ...
-
-            if self.statement() is myframe and myframe.parent:
+            if (
+                self.parent
+                and self.statement(future=True) is myframe
+                and myframe.parent
+            ):
                 myframe = myframe.parent.frame()
-        mystmt = self.statement()
+
+        # nodes.Module don't have a parent attribute
+        try:
+            mystmt = self.statement(future=True)
+        except StatementMissing:
+            mystmt = self
         # line filtering if we are in the same frame
         #
         # take care node may be missing lineno information (this is the case for
@@ -1816,7 +1825,7 @@ def _get_filtered_stmts(self, lookup_node, node, stmts, mystmt):
             if isinstance(lookup_node, (Const, Name)):
                 return [lookup_node], True
 
-        elif self.statement() is mystmt:
+        elif self.statement(future=True) is mystmt:
             # original node's statement is the assignment, only keeps
             # current node (gen exp, list comp)
 

astroid/nodes/scoped_nodes.py

@@ -2587,7 +2587,7 @@ def getattr(self, name, context=None, class_context=True):
         # Look for AnnAssigns, which are not attributes in the purest sense.
         for value in values:
             if isinstance(value, node_classes.AssignName):
-                stmt = value.statement()
+                stmt = value.statement(future=True)
                 if isinstance(stmt, node_classes.AnnAssign) and stmt.value is None:
                     raise AttributeInferenceError(
                         target=self, attribute=name, context=context

astroid/protocols.py

@@ -637,7 +637,7 @@ def _determine_starred_iteration_lookups(starred, target, lookups):
                 lookups.append((index, len(element.itered())))
                 _determine_starred_iteration_lookups(starred, element, lookups)
 
-    stmt = self.statement()
+    stmt = self.statement(future=True)
     if not isinstance(stmt, (nodes.Assign, nodes.For)):
         raise InferenceError(
             "Statement {stmt!r} enclosing {node!r} " "must be an Assign or For node.",

tests/unittest_builder.py

@@ -614,7 +614,9 @@ def test_module_base_props(self) -> None:
         self.assertEqual(module.pure_python, 1)
         self.assertEqual(module.package, 0)
         self.assertFalse(module.is_statement)
-        self.assertEqual(module.statement(), module)
+        with pytest.warns(DeprecationWarning) as records:
+            self.assertEqual(module.statement(), module)
+            assert len(records) == 1
         with pytest.warns(DeprecationWarning) as records:
             module.statement()
             assert len(records) == 1