Try to fix the failing tests by processing instance attributes and class level attributes separate

Author: Julfried

pylint/pyreverse/inspector.py

@@ -186,6 +186,14 @@ def visit_classdef(self, node: nodes.ClassDef) -> None:
                     self.compositions_handler.handle(assignattr, node)
                     self.handle_assignattr_type(assignattr, node)
 
+        # Process class attributes
+        for local_nodes in node.locals.values():
+            for local_node in local_nodes:
+                if isinstance(local_node, nodes.AssignName) and isinstance(
+                    local_node.parent, nodes.Assign
+                ):
+                    self.compositions_handler.handle(local_node, node)
+
     def visit_functiondef(self, node: nodes.FunctionDef) -> None:
         """Visit an astroid.Function node.
 
@@ -307,7 +315,9 @@ def set_next(
         pass
 
     @abstractmethod
-    def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
+    def handle(
+        self, node: nodes.AssignAttr | nodes.AssignName, parent: nodes.ClassDef
+    ) -> None:
         pass
 
 
@@ -332,21 +342,29 @@ def set_next(
         return handler
 
     @abstractmethod
-    def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
+    def handle(
+        self, node: nodes.AssignAttr | nodes.AssignName, parent: nodes.ClassDef
+    ) -> None:
         if self._next_handler:
             self._next_handler.handle(node, parent)
 
 
 class CompositionsHandler(AbstractRelationshipHandler):
     """Handle composition relationships where parent creates child objects."""
 
-    def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
+    def handle(
+        self, node: nodes.AssignAttr | nodes.AssignName, parent: nodes.ClassDef
+    ) -> None:
+        # If the node is not part of an assignment, pass to next handler
         if not isinstance(node.parent, (nodes.AnnAssign, nodes.Assign)):
             super().handle(node, parent)
             return
 
         value = node.parent.value
 
+        # Extract the name to handle both AssignAttr and AssignName nodes
+        name = node.attrname if isinstance(node, nodes.AssignAttr) else node.name
+
         # Composition: direct object creation (self.x = P())
         if isinstance(value, nodes.Call):
             inferred_types = utils.infer_node(node)
@@ -355,8 +373,8 @@ def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
             # Resolve nodes to actual class definitions
             resolved_types = resolve_to_class_def(element_types)
 
-            current = set(parent.compositions_type[node.attrname])
-            parent.compositions_type[node.attrname] = list(current | resolved_types)
+            current = set(parent.compositions_type[name])
+            parent.compositions_type[name] = list(current | resolved_types)
             return
 
         # Composition: comprehensions with object creation (self.x = [P() for ...])
@@ -376,8 +394,8 @@ def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
                 # Resolve nodes to actual class definitions
                 resolved_types = resolve_to_class_def(element_types)
 
-                current = set(parent.compositions_type[node.attrname])
-                parent.compositions_type[node.attrname] = list(current | resolved_types)
+                current = set(parent.compositions_type[name])
+                parent.compositions_type[name] = list(current | resolved_types)
                 return
 
         # Not a composition, pass to next handler
@@ -387,13 +405,19 @@ def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
 class AggregationsHandler(AbstractRelationshipHandler):
     """Handle aggregation relationships where parent receives child objects."""
 
-    def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
+    def handle(
+        self, node: nodes.AssignAttr | nodes.AssignName, parent: nodes.ClassDef
+    ) -> None:
+        # If the node is not part of an assignment, pass to next handler
         if not isinstance(node.parent, (nodes.AnnAssign, nodes.Assign)):
             super().handle(node, parent)
             return
 
         value = node.parent.value
 
+        # Extract the name to handle both AssignAttr and AssignName nodes
+        name = node.attrname if isinstance(node, nodes.AssignAttr) else node.name
+
         # Aggregation: direct assignment (self.x = x)
         if isinstance(value, nodes.Name):
             inferred_types = utils.infer_node(node)
@@ -402,8 +426,8 @@ def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
             # Resolve nodes to actual class definitions
             resolved_types = resolve_to_class_def(element_types)
 
-            current = set(parent.aggregations_type[node.attrname])
-            parent.aggregations_type[node.attrname] = list(current | resolved_types)
+            current = set(parent.aggregations_type[name])
+            parent.aggregations_type[name] = list(current | resolved_types)
             return
 
         # Aggregation: comprehensions without object creation (self.x = [existing_obj for ...])
@@ -423,8 +447,8 @@ def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
                 # Resolve nodes to actual class definitions
                 resolved_types = resolve_to_class_def(element_types)
 
-                current = set(parent.aggregations_type[node.attrname])
-                parent.aggregations_type[node.attrname] = list(current | resolved_types)
+                current = set(parent.aggregations_type[name])
+                parent.aggregations_type[name] = list(current | resolved_types)
                 return
 
         # Not an aggregation, pass to next handler
@@ -434,7 +458,12 @@ def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
 class AssociationsHandler(AbstractRelationshipHandler):
     """Handle regular association relationships."""
 
-    def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
+    def handle(
+        self, node: nodes.AssignAttr | nodes.AssignName, parent: nodes.ClassDef
+    ) -> None:
+        # Extract the name to handle both AssignAttr and AssignName nodes
+        name = node.attrname if isinstance(node, nodes.AssignAttr) else node.name
+
         # Type annotation only (x: P) -> Association
         # BUT only if there's no actual assignment (to avoid duplicates)
         if isinstance(node.parent, nodes.AnnAssign) and node.parent.value is None:
@@ -444,18 +473,18 @@ def handle(self, node: nodes.AssignAttr, parent: nodes.ClassDef) -> None:
             # Resolve nodes to actual class definitions
             resolved_types = resolve_to_class_def(element_types)
 
-            current = set(parent.associations_type[node.attrname])
-            parent.associations_type[node.attrname] = list(current | resolved_types)
+            current = set(parent.associations_type[name])
+            parent.associations_type[name] = list(current | resolved_types)
             return
 
         # Everything else is also association (fallback)
-        current = set(parent.associations_type[node.attrname])
+        current = set(parent.associations_type[name])
         inferred_types = utils.infer_node(node)
         element_types = extract_element_types(inferred_types)
 
         # Resolve Name nodes to actual class definitions
         resolved_types = resolve_to_class_def(element_types)
-        parent.associations_type[node.attrname] = list(current | resolved_types)
+        parent.associations_type[name] = list(current | resolved_types)
 
 
 def resolve_to_class_def(types: set[nodes.NodeNG]) -> set[nodes.ClassDef]: