type-hints industries

files/models.py

@@ -3,6 +3,8 @@
 from django.contrib.auth import get_user_model
 from django.db import models
 
+from django_stubs_ext.db.models import TypedModelMeta
+
 User = get_user_model()
 
 
@@ -31,7 +33,7 @@ def __str__(self):
         filename_with_extension = f"{self.name}.{self.extension}"
         return f"UserFile<{reprlib.repr(filename_with_extension)}>"
 
-    class Meta:
+    class Meta(TypedModelMeta):
         verbose_name = "Файл"
         verbose_name_plural = "Файлы"
         ordering = ["datetime_uploaded"]

files/serializers.py

@@ -3,7 +3,7 @@
 from files.models import UserFile
 
 
-class UserFileSerializer(ModelSerializer):
+class UserFileSerializer(ModelSerializer[UserFile]):
     class Meta:
         model = UserFile
         fields = [

industries/models.py

@@ -1,4 +1,5 @@
 from django.db import models
+from django_stubs_ext.db.models import TypedModelMeta
 
 
 class Industry(models.Model):
@@ -20,7 +21,7 @@ class Industry(models.Model):
     def __str__(self):
         return f"Industry<{self.id}> - {self.name}"
 
-    class Meta:
+    class Meta(TypedModelMeta):
         verbose_name = "Индустрия"
         verbose_name_plural = "Индустрии"
         ordering = ["name"]

industries/serializers.py

@@ -1,9 +1,10 @@
+from django_stubs_ext.db.models import TypedModelMeta
 from rest_framework import serializers
 
 from industries.models import Industry
 
 
 class IndustrySerializer(serializers.ModelSerializer):
-    class Meta:
+    class Meta(TypedModelMeta):
         model = Industry
         fields = ["id", "name", "datetime_created"]

industries/tests.py

@@ -92,7 +92,7 @@ def test_industry_update_with_wrong_data(self):
 
         self.assertEqual(response.status_code, 400)
 
-    def _user_create(self):
+    def _user_create(self) -> CustomUser:
         request = self.factory.post("auth/users/", USER_CREATE_DATA)
         response = self.user_list_view(request)
         user_id = response.data["id"]