Add type annotations

filter_bug_repro/settings.py

@@ -12,6 +12,11 @@
 
 from pathlib import Path
 
+import django_stubs_ext
+from rest_framework import viewsets
+
+django_stubs_ext.monkeypatch(extra_classes=(viewsets.ModelViewSet,))
+
 # Build paths inside the project like this: BASE_DIR / 'subdir'.
 BASE_DIR = Path(__file__).resolve().parent.parent
 
@@ -25,7 +30,7 @@
 # SECURITY WARNING: don't run with debug turned on in production!
 DEBUG = True
 
-ALLOWED_HOSTS = []
+ALLOWED_HOSTS: list[str] = []
 
 
 # Application definition

filter_bug_repro/urls.py

@@ -24,23 +24,23 @@ class Document(models.Model):
     name = models.CharField(max_length=255)
     owner = models.ForeignKey("auth.User", on_delete=models.CASCADE)
 
-    def __str__(self):
+    def __str__(self) -> str:
         return self.name
 
 
-class DocumentFilterSet(filters.FilterSet):
+class DocumentFilterSet(filters.FilterSet):  # type: ignore
     class Meta:
         model = Document
         fields = ("owner",)
 
 
-class DocumentSerializer(serializers.ModelSerializer):
+class DocumentSerializer(serializers.ModelSerializer[Document]):
     class Meta:
         model = Document
         fields = ("id", "name", "owner")
 
 
-class DocumentViewSet(viewsets.ModelViewSet):
+class DocumentViewSet(viewsets.ModelViewSet[Document]):
     queryset = Document.objects.all()
     serializer_class = DocumentSerializer
     filterset_class = DocumentFilterSet

mypy.ini

@@ -0,0 +1,28 @@
+[mypy]
+python_version = 3.10
+
+warn_return_any = True
+warn_unused_configs = True
+warn_unused_ignores = True
+disallow_untyped_defs = True
+disallow_incomplete_defs = True
+disallow_untyped_decorators = True
+disallow_any_expr = False
+disallow_any_decorated = True
+disallow_subclassing_any = True
+disallow_any_generics = True
+disallow_any_explicit = False
+disallow_any_unimported = True
+no_implicit_optional = True
+strict_equality = True
+warn_redundant_casts = True
+
+plugins =
+  mypy_django_plugin.main,
+  mypy_drf_plugin.main
+
+[mypy.plugins.django-stubs]
+django_settings_module = filter_bug_repro.settings
+
+[mypy-django_filters.*]
+ignore_missing_imports = True