Make self-cls-assignment check tuple assignment (#5268)

Co-authored-by: Marc Mueller <[email protected]>

Author: DanielNoord

ChangeLog

@@ -127,6 +127,8 @@ Release date: TBA
 
   Closes #4426
 
+* ``self-cls-assignment`` now also considers tuple assignment
+
 * Fix ``missing-function-docstring`` not being able to check ``__init__`` and other
   magic methods even if the ``no-docstring-rgx`` setting was set to do so
 

doc/whatsnew/2.12.rst

@@ -112,6 +112,8 @@ Other Changes
 
   Fixes part of #3688
 
+* ``self-cls-assignment`` now also considers tuple assignment
+
 * ``undefined-variable`` now correctly triggers for assignment expressions in if ... else statements
   This includes a basic form of control flow inference for if ... else statements using
   constant boolean values

pylint/checkers/variables.py

@@ -61,7 +61,7 @@
 import re
 import sys
 from functools import lru_cache
-from typing import DefaultDict, List, Tuple
+from typing import DefaultDict, List, Optional, Set, Tuple
 
 import astroid
 from astroid import nodes
@@ -2022,18 +2022,19 @@ def _store_type_annotation_names(self, node):
             return
         self._store_type_annotation_node(node.type_annotation)
 
-    def _check_self_cls_assign(self, node):
+    def _check_self_cls_assign(self, node: nodes.Assign) -> None:
         """Check that self/cls don't get assigned"""
-        assign_names = {
-            target.name
-            for target in node.targets
-            if isinstance(target, nodes.AssignName)
-        }
+        assign_names: Set[Optional[str]] = set()
+        for target in node.targets:
+            if isinstance(target, nodes.AssignName):
+                assign_names.add(target.name)
+            elif isinstance(target, nodes.Tuple):
+                assign_names.update(
+                    elt.name for elt in target.elts if isinstance(elt, nodes.AssignName)
+                )
         scope = node.scope()
         nonlocals_with_same_name = any(
-            child
-            for child in scope.body
-            if isinstance(child, nodes.Nonlocal) and assign_names & set(child.names)
+            child for child in scope.body if isinstance(child, nodes.Nonlocal)
         )
         if nonlocals_with_same_name:
             scope = node.scope().parent.scope()
@@ -2048,12 +2049,7 @@ def _check_self_cls_assign(self, node):
         if not argument_names:
             return
         self_cls_name = argument_names[0]
-        target_assign_names = (
-            target.name
-            for target in node.targets
-            if isinstance(target, nodes.AssignName)
-        )
-        if self_cls_name in target_assign_names:
+        if self_cls_name in assign_names:
             self.add_message("self-cls-assignment", node=node, args=(self_cls_name,))
 
     def _check_unpacking(self, inferred, node, targets):

tests/functional/s/self/self_cls_assignment.py

@@ -14,6 +14,8 @@ def self_foo(bar_):
     def self_foofoo(self, lala):
         """Instance method, should warn for self"""
         self = lala  # [self-cls-assignment]
+        self, var = lala, 1  # [self-cls-assignment]
+        print(var)
 
     @classmethod
     def cls_foo(cls):

tests/functional/s/self/self_cls_assignment.txt

@@ -1,4 +1,5 @@
-self-cls-assignment:11:8:Foo.self_foo:Invalid assignment to bar_ in method
-self-cls-assignment:16:8:Foo.self_foofoo:Invalid assignment to self in method
-self-cls-assignment:21:8:Foo.cls_foo:Invalid assignment to cls in method
-self-cls-assignment:44:12:TestNonLocal.function._set_param:Invalid assignment to self in method
+self-cls-assignment:11:8:Foo.self_foo:Invalid assignment to bar_ in method:HIGH
+self-cls-assignment:16:8:Foo.self_foofoo:Invalid assignment to self in method:HIGH
+self-cls-assignment:17:8:Foo.self_foofoo:Invalid assignment to self in method:HIGH
+self-cls-assignment:23:8:Foo.cls_foo:Invalid assignment to cls in method:HIGH
+self-cls-assignment:46:12:TestNonLocal.function._set_param:Invalid assignment to self in method:HIGH