Don't warn about missing __slots__ for string Enums

Author: domdfcoding

flake8_slots/__init__.py

@@ -28,6 +28,7 @@
 
 # stdlib
 import ast
+import re
 from typing import Iterator, Union
 
 # 3rd party
@@ -81,6 +82,9 @@ def resolve_dotted_name(name: Union[str, ast.Name, ast.Attribute, ast.Call, ast.
 		yield from resolve_dotted_name(name.func)  # type: ignore
 
 
+_enum_re = re.compile(r"\.?Enum")
+
+
 class Visitor(flake8_helper.Visitor):
 	"""
 	AST visitor to identify missing ``__slots__`` for subclasses of immutable types.
@@ -90,7 +94,11 @@ def visit_ClassDef(self, node: ast.ClassDef):  # noqa: D102
 
 		bases = ['.'.join(resolve_dotted_name(base)) for base in node.bases]  # type: ignore
 
-		is_str = "str" in bases or "builtins.str" in bases  # SLOT000
+		is_str = False
+
+		if "str" in bases or "builtins.str" in bases:  # SLOT000
+			if not any(map(_enum_re.match, bases)):
+				is_str = True
 
 		if "tuple" in bases or "builtins.tuple" in bases:  # SLOT001
 			is_tuple = True

tests/test_good_code.py

@@ -43,6 +43,14 @@ class Compound(Counter):
 	pass
 """
 
+enum = """\
+from enum import Enum
+class Protocols(str, Enum):
+	HTTP = "http"
+	FTP = "ftp"
+	SSH = "SSH"
+"""
+
 sources = pytest.mark.parametrize(
 		"source",
 		[
@@ -59,6 +67,7 @@ class Compound(Counter):
 				typing_namedtuple_a,
 				typing_namedtuple_b,
 				mutable,
+				enum,
 				]
 		)