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"]

mailing/models.py

@@ -4,6 +4,8 @@
 from django.db import models
 from .constants import get_default_mailing_schema
 
+from django_stubs_ext.db.models import TypedModelMeta
+
 
 def get_template_path(instance, filename):
     ext = filename.split(".")[-1]
@@ -16,7 +18,7 @@ class MailingSchema(models.Model):
     schema = models.JSONField(default=get_default_mailing_schema, null=True, blank=True)
     template = models.TextField()
 
-    class Meta:
+    class Meta(TypedModelMeta):
         verbose_name = "Схема шаблона письма"
         verbose_name_plural = "Схемы шаблонов писем"
 

mailing/views.py

@@ -2,14 +2,15 @@
 from django.http import JsonResponse
 from django.shortcuts import render
 from rest_framework.views import APIView
+from rest_framework.request import Request
 
 from users.models import CustomUser
 from .utils import send_mass_mail
 from .models import MailingSchema
 
 
 class SendMailView(APIView):
-    def post(self, request):
+    def post(self, request: Request) -> JsonResponse:
         users = request.POST.getlist("users[]")
         schema_id = request.POST["schemas"]
         subject = request.POST["subject"]
@@ -30,7 +31,7 @@ def get(self, request):
 
 
 class TemplateFieldsView(APIView):
-    def get(self, request, schema_id):
+    def get(self, request: Request, schema_id: int) -> JsonResponse:
         return JsonResponse(
             dict(MailingTemplateRender().get_template_fields_context(schema_id))
         )