add BaseManager.create() typechecking

test-data/typecheck/managers.test

@@ -0,0 +1,139 @@
+[CASE test_every_model_has_objects_queryset_available]
+from django.db import models
+class User(models.Model):
+    pass
+reveal_type(User.objects)  # E: Revealed type is 'django.db.models.manager.Manager[main.User]'
+reveal_type(User.objects.get())  # E: Revealed type is 'main.User*'
+
+[CASE every_model_has_its_own_objects_queryset]
+from django.db import models
+class Parent(models.Model):
+    pass
+class Child(Parent):
+    pass
+reveal_type(Parent.objects)  # E: Revealed type is 'django.db.models.manager.Manager[main.Parent]'
+reveal_type(Child.objects)  # E: Revealed type is 'django.db.models.manager.Manager[main.Child]'
+[out]
+
+[CASE if_manager_is_defined_on_model_do_not_add_objects]
+from django.db import models
+
+class MyModel(models.Model):
+    authors = models.Manager[MyModel]()
+reveal_type(MyModel.authors)  # E: Revealed type is 'django.db.models.manager.Manager[main.MyModel]'
+reveal_type(MyModel.objects)  # E: Revealed type is 'Any'
+[out]
+
+[CASE test_model_objects_attribute_present_in_case_of_model_cls_passed_as_generic_parameter]
+from typing import TypeVar, Generic, Type
+from django.db import models
+
+_T = TypeVar('_T', bound=models.Model)
+class Base(Generic[_T]):
+    def __init__(self, model_cls: Type[_T]):
+        self.model_cls = model_cls
+        reveal_type(self.model_cls._default_manager)  # E: Revealed type is 'django.db.models.manager.Manager[django.db.models.base.Model]'
+class MyModel(models.Model):
+    pass
+base_instance = Base(MyModel)
+reveal_type(base_instance.model_cls._default_manager)  # E: Revealed type is 'django.db.models.manager.Manager[main.MyModel]'
+
+class Child(Base[MyModel]):
+    def method(self) -> None:
+        reveal_type(self.model_cls._default_manager)  # E: Revealed type is 'django.db.models.manager.Manager[main.MyModel]'
+
+[CASE if_custom_manager_defined_it_is_set_to_default_manager]
+from typing import TypeVar
+from django.db import models
+_T = TypeVar('_T', bound=models.Model)
+class CustomManager(models.Manager[_T]):
+    pass
+class MyModel(models.Model):
+    manager = CustomManager[MyModel]()
+reveal_type(MyModel._default_manager)  # E: Revealed type is 'main.CustomManager[main.MyModel]'
+
+[CASE if_default_manager_name_is_passed_set_default_manager_to_it]
+from typing import TypeVar
+from django.db import models
+
+_T = TypeVar('_T', bound=models.Model)
+class Manager1(models.Manager[_T]):
+    pass
+class Manager2(models.Manager[_T]):
+    pass
+class MyModel(models.Model):
+    class Meta:
+        default_manager_name = 'm2'
+    m1: Manager1[MyModel]
+    m2: Manager2[MyModel]
+reveal_type(MyModel._default_manager)  # E: Revealed type is 'main.Manager2[main.MyModel]'
+
+[CASE test_leave_as_is_if_objects_is_set_and_fill_typevars_with_outer_class]
+from django.db import models
+
+class UserManager(models.Manager[MyUser]):
+    def get_or_404(self) -> MyUser:
+        pass
+
+class MyUser(models.Model):
+    objects = UserManager()
+
+reveal_type(MyUser.objects)  # E: Revealed type is 'main.UserManager[main.MyUser]'
+reveal_type(MyUser.objects.get())  # E: Revealed type is 'main.MyUser*'
+reveal_type(MyUser.objects.get_or_404())  # E: Revealed type is 'main.MyUser'
+
+[CASE model_imported_from_different_file]
+from django.db import models
+from models.main import Inventory
+
+class Band(models.Model):
+    pass
+reveal_type(Inventory.objects)  # E: Revealed type is 'django.db.models.manager.Manager[models.main.Inventory]'
+reveal_type(Band.objects)  # E: Revealed type is 'django.db.models.manager.Manager[main.Band]'
+[file models/__init__.py]
+[file models/main.py]
+from django.db import models
+class Inventory(models.Model):
+    pass
+
+[CASE managers_that_defined_on_other_models_do_not_influence]
+from django.db import models
+
+class AbstractPerson(models.Model):
+    abstract_persons = models.Manager[AbstractPerson]()
+class PublishedBookManager(models.Manager[Book]):
+    pass
+class AnnotatedBookManager(models.Manager[Book]):
+    pass
+class Book(models.Model):
+    title = models.CharField(max_length=50)
+    published_objects = PublishedBookManager()
+    annotated_objects = AnnotatedBookManager()
+
+reveal_type(AbstractPerson.abstract_persons)  # E: Revealed type is 'django.db.models.manager.Manager[main.AbstractPerson]'
+reveal_type(Book.published_objects)  # E: Revealed type is 'main.PublishedBookManager[main.Book]'
+Book.published_objects.create(title='hello')
+reveal_type(Book.annotated_objects)  # E: Revealed type is 'main.AnnotatedBookManager[main.Book]'
+Book.annotated_objects.create(title='hello')
+[out]
+
+[CASE managers_inherited_from_abstract_classes_multiple_inheritance]
+from django.db import models
+class CustomManager1(models.Manager[AbstractBase1]):
+    pass
+class AbstractBase1(models.Model):
+    class Meta:
+        abstract = True
+    name = models.CharField(max_length=50)
+    manager1 = CustomManager1()
+class CustomManager2(models.Manager[AbstractBase2]):
+    pass
+class AbstractBase2(models.Model):
+    class Meta:
+        abstract = True
+    value = models.CharField(max_length=50)
+    restricted = CustomManager2()
+
+class Child(AbstractBase1, AbstractBase2):
+    pass
+[out]

test-data/typecheck/model_create.test

@@ -0,0 +1,34 @@
+[CASE default_manager_create_is_typechecked]
+from django.db import models
+
+class User(models.Model):
+    name = models.CharField(max_length=100)
+    age = models.IntegerField()
+
+User.objects.create(name='Max', age=10)
+User.objects.create(name=1010)  # E: Incompatible type for "name" of "User" (got "int", expected "Union[str, Combinable]")
+[out]
+
+[CASE model_recognises_parent_attributes]
+from django.db import models
+
+class Parent(models.Model):
+    name = models.CharField(max_length=100)
+class Child(Parent):
+    lastname = models.CharField(max_length=100)
+Child.objects.create(name='Maxim', lastname='Maxim2')
+[out]
+
+[CASE deep_multiple_inheritance_with_create]
+from django.db import models
+
+class Parent1(models.Model):
+    name1 = models.CharField(max_length=50)
+class Parent2(models.Model):
+    id2 = models.AutoField(primary_key=True)
+    name2 = models.CharField(max_length=50)
+class Child1(Parent1, Parent2):
+    value = models.IntegerField()
+class Child4(Child1):
+    value4 = models.IntegerField()
+Child4.objects.create(name1='n1', name2='n2', value=1, value4=4)

test-data/typecheck/model_init.test

@@ -149,3 +149,9 @@ MyModel(field=time())
 MyModel(field='12:00')
 MyModel(field=100)  # E: Incompatible type for "field" of "MyModel" (got "int", expected "Union[str, time]")
 
+[CASE charfield_with_integer_choices]
+from django.db import models
+class MyModel(models.Model):
+    day = models.CharField(max_length=3, choices=((1, 'Fri'), (2, 'Sat')))
+MyModel(day=1)
+[out]

test-data/typecheck/models.test

@@ -1,49 +0,0 @@
-[CASE test_every_model_has_objects_queryset_available]
-from django.db import models
-class User(models.Model):
-    pass
-reveal_type(User.objects)  # E: Revealed type is 'django.db.models.manager.Manager[main.User]'
-reveal_type(User.objects.get())  # E: Revealed type is 'main.User*'
-
-[CASE test_leave_as_is_if_objects_is_set_and_fill_typevars_with_outer_class]
-from django.db import models
-
-class UserManager(models.Manager[User]):
-    def get_or_404(self) -> User:
-        pass
-
-class User(models.Model):
-    objects = UserManager()
-
-reveal_type(User.objects)  # E: Revealed type is 'main.UserManager'
-reveal_type(User.objects.get())  # E: Revealed type is 'main.User*'
-reveal_type(User.objects.get_or_404())  # E: Revealed type is 'main.User'
-
-[CASE test_model_objects_attribute_present_in_case_of_model_cls_passed_as_parameter]
-from typing import Type
-from django.db import models
-
-class Base:
-    def __init__(self, model_cls: Type[models.Model]):
-        self.model_cls = model_cls
-class MyModel(models.Model):
-    pass
-reveal_type(Base(MyModel).model_cls.objects)  # E: Revealed type is 'django.db.models.manager.Manager[django.db.models.base.Model]'
-
-[CASE test_model_objects_attribute_present_in_case_of_model_cls_passed_as_generic_parameter]
-from typing import TypeVar, Generic, Type
-from django.db import models
-
-_T = TypeVar('_T', bound=models.Model)
-class Base(Generic[_T]):
-    def __init__(self, model_cls: Type[_T]):
-        self.model_cls = model_cls
-        reveal_type(self.model_cls.objects)  # E: Revealed type is 'django.db.models.manager.Manager[django.db.models.base.Model]'
-class MyModel(models.Model):
-    pass
-base_instance = Base(MyModel)
-reveal_type(base_instance.model_cls.objects)  # E: Revealed type is 'django.db.models.manager.Manager[main.MyModel]'
-
-class Child(Base[MyModel]):
-    def method(self) -> None:
-        reveal_type(self.model_cls.objects)  # E: Revealed type is 'django.db.models.manager.Manager[main.MyModel]'