diff options
| author | Collin Anderson <cmawebsite@gmail.com> | 2022-04-22 08:13:12 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-04-22 14:13:12 +0200 |
| commit | 0de89b6f8d4d8f5fd232d6c5412260b0d79a760a (patch) | |
| tree | 4553b6b8ae2fce981f459d0ba4d97d2168026c68 | |
| parent | eeb0bb63795f7500156abaed2a94aae681a9fc4a (diff) | |
Refs #31223 -- Added __class_getitem__() to ForeignKey.
| -rw-r--r-- | django/db/models/fields/related.py | 3 | ||||
| -rw-r--r-- | tests/model_fields/test_foreignkey.py | 3 |
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) |
