Correctly handle annotated assignments.

Author: domdfcoding

flake8_slots/__init__.py

@@ -63,6 +63,10 @@ def visit_Assign(self, node: ast.Assign):  # noqa: D102
 			if isinstance(target, ast.Name) and target.id == "__slots__":
 				self.has_slots = True
 
+	def visit_AnnAssign(self, node: ast.AnnAssign):  # noqa: D102
+		if isinstance(node.target, ast.Name) and node.target.id == "__slots__":
+			self.has_slots = True
+
 
 def resolve_dotted_name(name: Union[str, ast.Name, ast.Attribute, ast.Call, ast.Subscript]) -> Iterator[str]:
 	if isinstance(name, str):

tests/test_good_code.py

@@ -17,6 +17,12 @@
 namedtuple_a = "import collections\nclass Person(collections.namedtuple('foo', 'name, age')):  __slots__ = ()"
 namedtuple_b = "from collections import namedtuple\nclass Person(namedtuple('foo', 'name, age')):  __slots__ = ()"
 
+namedtuple_c = """\
+from collections import namedtuple
+class Person(namedtuple('foo', 'name, age')):
+	__slots__: Tuple[str] = ()
+"""
+
 typing_namedtuple_a = """\
 import typing
 class Person(typing.NamedTuple):
@@ -33,6 +39,28 @@ class Person(NamedTuple):
 	age: int
 """
 
+typing_namedtuple_c = """\
+from typing import NamedTuple
+class Person(NamedTuple):
+	__slots__: Tuple[str] = ()
+	name: str
+	age: int
+"""
+
+typing_namedtuple_d = """\
+from typing import NamedTuple
+class Person(NamedTuple):
+	__slots__: List[str] = []
+	name: str
+	age: int
+"""
+
+mutable = """\
+from collections import Counter
+class Compound(Counter):
+	pass
+"""
+
 sources = pytest.mark.parametrize(
 		"source",
 		[
@@ -45,8 +73,12 @@ class Person(NamedTuple):
 				typing_tuple_c,
 				namedtuple_a,
 				namedtuple_b,
+				namedtuple_c,
 				typing_namedtuple_a,
 				typing_namedtuple_b,
+				typing_namedtuple_c,
+				typing_namedtuple_d,
+				mutable,
 				]
 		)