don't remove objects / DoesNotExist / etc. from model classes

django-stubs fails to restore these in too many cases. see upstream issue typeddjango#1744 (not upstreamed)
getsentry · Oct 31, 2023 · c122d5c · c122d5c
1 parent b431663
commit c122d5c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/django-stubs/contrib/sessions/base_session.pyi b/django-stubs/contrib/sessions/base_session.pyi
@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Any, Literal, TypeVar
+from typing import Any, ClassVar, Literal, TypeVar
 
 from django.contrib.sessions.backends.base import SessionBase
 from django.db import models
@@ -14,7 +14,7 @@ class AbstractBaseSession(models.Model):
     expire_date: datetime
     session_data: str
     session_key: str
-    objects: Any
+    objects: ClassVar[Any]
 
     class Meta:
         abstract: Literal[True]

diff --git a/mypy_django_plugin/main.py b/mypy_django_plugin/main.py
@@ -224,7 +224,7 @@ def get_method_hook(self, fullname: str) -> Optional[Callable[[MethodContext], M
 
     def get_customize_class_mro_hook(self, fullname: str) -> Optional[Callable[[ClassDefContext], None]]:
         if fullname == fullnames.MODEL_CLASS_FULLNAME:
-            return MetaclassAdjustments.adjust_model_class
+            return None
 
         sym = self.lookup_fully_qualified(fullname)
         if (