Fix a crash for Enum class decorated with dataclass (#9104)

* Fix a crash when an enum class which is also decorated with a ``dataclasses.dataclass`` decorator is defined. Closes #9100 * Update test to mention the `astroid` issue that addresses the missing ``__members__`` object. * Use an `if` instead of the `try/except` block. * Update pylint/checkers/utils.py Co-authored-by: Daniël van Noord <[email protected]> --------- Co-authored-by: Daniël van Noord <[email protected]>
pylint-dev · Oct 4, 2023 · 2c3425d · 2c3425d
1 parent cd4b292
commit 2c3425d
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 3 deletions.
diff --git a/doc/whatsnew/fragments/9100.other b/doc/whatsnew/fragments/9100.other
@@ -0,0 +1,3 @@
+Fix a crash when an enum class which is also decorated with a ``dataclasses.dataclass`` decorator is defined.
+
+Closes #9100
diff --git a/pylint/checkers/utils.py b/pylint/checkers/utils.py
@@ -2281,5 +2281,8 @@ def is_enum_member(node: nodes.AssignName) -> bool:
     ):
         return False
 
-    enum_member_objects = frame.locals.get("__members__")[0].items
-    return node.name in [name_obj.name for value, name_obj in enum_member_objects]
+    members = frame.locals.get("__members__")
+    # A dataclass is one known case for when `members` can be `None`
+    if members is None:
+        return False
+    return node.name in [name_obj.name for value, name_obj in members[0].items]
diff --git a/tests/functional/i/invalid/invalid_name/invalid_name_enum.py b/tests/functional/i/invalid/invalid_name/invalid_name_enum.py
@@ -2,6 +2,7 @@
 # pylint: disable=too-few-public-methods
 
 
+from dataclasses import dataclass
 from enum import Enum
 
 
@@ -28,3 +29,11 @@ def __init__(self, red: int, green: int, blue: int) -> None:
     def as_hex(self) -> str:
         """Get hex 'abcdef' representation for a color."""
         return f'{self.red:0{2}x}{self.green:0{2}x}{self.blue:0{2}x}'
+
+
+@dataclass
+class Something(str, Enum):
+    """ A false positive for ``invalid-name``
+    which should be fixed by https://github.com/pylint-dev/astroid/issues/2317
+    """
+    ASD: str = 1  # [invalid-name]
diff --git a/tests/functional/i/invalid/invalid_name/invalid_name_enum.txt b/tests/functional/i/invalid/invalid_name/invalid_name_enum.txt
@@ -1 +1,2 @@
-invalid-name:16:4:16:14:Color:"Class constant name ""aquamarine"" doesn't conform to UPPER_CASE naming style":HIGH
+invalid-name:17:4:17:14:Color:"Class constant name ""aquamarine"" doesn't conform to UPPER_CASE naming style":HIGH
+invalid-name:39:4:None:None:Something:"Attribute name ""ASD"" doesn't conform to snake_case naming style":HIGH
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		Fix a crash when an enum class which is also decorated with a ``dataclasses.dataclass`` decorator is defined.

		Closes #9100