summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorCollin Anderson <cmawebsite@gmail.com>2022-04-22 08:13:12 -0400
committerGitHub <noreply@github.com>2022-04-22 14:13:12 +0200
commit0de89b6f8d4d8f5fd232d6c5412260b0d79a760a (patch)
tree4553b6b8ae2fce981f459d0ba4d97d2168026c68
parenteeb0bb63795f7500156abaed2a94aae681a9fc4a (diff)
Refs #31223 -- Added __class_getitem__() to ForeignKey.
-rw-r--r--django/db/models/fields/related.py3
-rw-r--r--tests/model_fields/test_foreignkey.py3
2 files changed, 6 insertions, 0 deletions
diff --git a/django/db/models/fields/related.py b/django/db/models/fields/related.py
index ca4f4368b6..fad5560486 100644
--- a/django/db/models/fields/related.py
+++ b/django/db/models/fields/related.py
@@ -978,6 +978,9 @@ class ForeignKey(ForeignObject):
)
self.db_constraint = db_constraint
+ def __class_getitem__(cls, *args, **kwargs):
+ return cls
+
def check(self, **kwargs):
return [
*super().check(**kwargs),
diff --git a/tests/model_fields/test_foreignkey.py b/tests/model_fields/test_foreignkey.py
index a5b646ae86..ca8eff3540 100644
--- a/tests/model_fields/test_foreignkey.py
+++ b/tests/model_fields/test_foreignkey.py
@@ -164,3 +164,6 @@ class ForeignKeyTests(TestCase):
class MyModel(models.Model):
child = models.ForeignKey(1, models.CASCADE)
+
+ def test_manager_class_getitem(self):
+ self.assertIs(models.ForeignKey["Foo"], models.ForeignKey)