From 6b29918068c436b2a1fb7801f53d90478bdf86cc Mon Sep 17 00:00:00 2001
From: Petter Friberg <petter_friberg@hotmail.com>
Date: Thu, 28 Sep 2023 20:52:54 +0200
Subject: [PATCH] Add missing `_default_manager` symbol to generated through
 model

---
 mypy_django_plugin/transformers/models.py | 6 ++++++
 tests/typecheck/fields/test_related.yml   | 5 ++++-
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/mypy_django_plugin/transformers/models.py b/mypy_django_plugin/transformers/models.py
index c8cf52a68..e9e4568b0 100644
--- a/mypy_django_plugin/transformers/models.py
+++ b/mypy_django_plugin/transformers/models.py
@@ -740,6 +740,12 @@ def run(self) -> None:
                     name="objects",
                     sym_type=Instance(manager_info, [Instance(through_model, [])]),
                 )
+                # Also add manager as '_default_manager' attribute
+                helpers.add_new_sym_for_info(
+                    through_model,
+                    name="_default_manager",
+                    sym_type=Instance(manager_info, [Instance(through_model, [])]),
+                )
 
     @cached_property
     def default_pk_instance(self) -> Instance:
diff --git a/tests/typecheck/fields/test_related.yml b/tests/typecheck/fields/test_related.yml
index 8f21557e1..6c10333e5 100644
--- a/tests/typecheck/fields/test_related.yml
+++ b/tests/typecheck/fields/test_related.yml
@@ -337,6 +337,7 @@
         from myapp.models import User
         reveal_type(User().friends)  # N: Revealed type is "django.db.models.fields.related_descriptors.ManyRelatedManager[myapp.models.User]"
         reveal_type(User.friends.through.objects.get())  # N: Revealed type is "myapp.models.User_friends"
+        reveal_type(User.friends.through._default_manager)  # N: Revealed type is "django.db.models.manager.Manager[myapp.models.User_friends]"
         reveal_type(User.friends.through().from_user)  # N: Revealed type is "myapp.models.User"
         reveal_type(User.friends.through().from_user_id)  # N: Revealed type is "builtins.int"
         reveal_type(User.friends.through().to_user)  # N: Revealed type is "myapp.models.User"
@@ -1088,6 +1089,7 @@
 
         reveal_type(MyModel.auto_through.through)
         reveal_type(MyModel.auto_through.through.mymodel)
+        reveal_type(MyModel.auto_through.through._default_manager)
 
         reveal_type(MyModel.other_again.through)
     out: |
@@ -1106,7 +1108,8 @@
         main:17: note: Revealed type is "builtins.int"
         main:19: note: Revealed type is "Type[myapp.models.MyModel_auto_through]"
         main:20: note: Revealed type is "django.db.models.fields.related_descriptors.ForwardManyToOneDescriptor[django.db.models.fields.related.ForeignKey[Union[myapp.models.MyModel, django.db.models.expressions.Combinable], myapp.models.MyModel]]"
-        main:22: note: Revealed type is "Type[myapp.models.MyModel_other_again]"
+        main:21: note: Revealed type is "django.db.models.manager.Manager[myapp.models.MyModel_auto_through]"
+        main:23: note: Revealed type is "Type[myapp.models.MyModel_other_again]"
     installed_apps:
         -   myapp
     files: