diff options
| author | Mariusz Felisiak <felisiak.mariusz@gmail.com> | 2021-04-07 10:28:40 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-04-07 10:28:40 +0200 |
| commit | c6859f1a684edec7bb33038b4408046a4db0c16d (patch) | |
| tree | bf539dfe0c54996c9ead3e87feb44145340b12a8 | |
| parent | 5b05a45c62f4702a6039cd3de290320c232cb808 (diff) | |
Refs #32074 -- Backported Enum.__repr__() from Python 3.10.
Enum.__repr__() was changed in [1], we should use the same format in
Python < 3.10.
[1] https://bugs.python.org/issue40066
| -rw-r--r-- | django/db/models/constraints.py | 8 | ||||
| -rw-r--r-- | django/db/models/enums.py | 6 | ||||
| -rw-r--r-- | tests/model_enums/tests.py | 4 |
3 files changed, 15 insertions, 3 deletions
diff --git a/django/db/models/constraints.py b/django/db/models/constraints.py index b073df1763..6dfc42942f 100644 --- a/django/db/models/constraints.py +++ b/django/db/models/constraints.py @@ -4,6 +4,7 @@ from django.db.models.expressions import ExpressionList, F from django.db.models.indexes import IndexExpression from django.db.models.query_utils import Q from django.db.models.sql.query import Query +from django.utils.version import PY310 __all__ = ['CheckConstraint', 'Deferrable', 'UniqueConstraint'] @@ -85,6 +86,11 @@ class Deferrable(Enum): DEFERRED = 'deferred' IMMEDIATE = 'immediate' + # A similar format is used in Python 3.10+. + if not PY310: + def __repr__(self): + return '%s.%s' % (self.__class__.__qualname__, self._name_) + class UniqueConstraint(BaseConstraint): def __init__( @@ -218,7 +224,7 @@ class UniqueConstraint(BaseConstraint): '' if not self.expressions else ' expressions=%s' % repr(self.expressions), ' name=%s' % repr(self.name), '' if self.condition is None else ' condition=%s' % self.condition, - '' if self.deferrable is None else ' deferrable=%s' % self.deferrable, + '' if self.deferrable is None else ' deferrable=%r' % self.deferrable, '' if not self.include else ' include=%s' % repr(self.include), '' if not self.opclasses else ' opclasses=%s' % repr(self.opclasses), ) diff --git a/django/db/models/enums.py b/django/db/models/enums.py index 7082a397c2..dd9088597d 100644 --- a/django/db/models/enums.py +++ b/django/db/models/enums.py @@ -2,6 +2,7 @@ import enum from types import DynamicClassAttribute from django.utils.functional import Promise +from django.utils.version import PY310 __all__ = ['Choices', 'IntegerChoices', 'TextChoices'] @@ -74,6 +75,11 @@ class Choices(enum.Enum, metaclass=ChoicesMeta): """ return str(self.value) + # A similar format is used in Python 3.10+. + if not PY310: + def __repr__(self): + return '%s.%s' % (self.__class__.__qualname__, self._name_) + class IntegerChoices(int, Choices): """Class for creating enumerated integer choices.""" diff --git a/tests/model_enums/tests.py b/tests/model_enums/tests.py index 78f8b146be..cda835010d 100644 --- a/tests/model_enums/tests.py +++ b/tests/model_enums/tests.py @@ -48,7 +48,7 @@ class ChoicesTests(SimpleTestCase): self.assertEqual(Suit.values, [1, 2, 3, 4]) self.assertEqual(Suit.names, ['DIAMOND', 'SPADE', 'HEART', 'CLUB']) - self.assertEqual(repr(Suit.DIAMOND), '<Suit.DIAMOND: 1>') + self.assertEqual(repr(Suit.DIAMOND), 'Suit.DIAMOND') self.assertEqual(Suit.DIAMOND.label, 'Diamond') self.assertEqual(Suit.DIAMOND.value, 1) self.assertEqual(Suit['DIAMOND'], Suit.DIAMOND) @@ -89,7 +89,7 @@ class ChoicesTests(SimpleTestCase): self.assertEqual(YearInSchool.values, ['FR', 'SO', 'JR', 'SR', 'GR']) self.assertEqual(YearInSchool.names, ['FRESHMAN', 'SOPHOMORE', 'JUNIOR', 'SENIOR', 'GRADUATE']) - self.assertEqual(repr(YearInSchool.FRESHMAN), "<YearInSchool.FRESHMAN: 'FR'>") + self.assertEqual(repr(YearInSchool.FRESHMAN), 'YearInSchool.FRESHMAN') self.assertEqual(YearInSchool.FRESHMAN.label, 'Freshman') self.assertEqual(YearInSchool.FRESHMAN.value, 'FR') self.assertEqual(YearInSchool['FRESHMAN'], YearInSchool.FRESHMAN) |
