Skip to content

Commit a77be30

Browse files
authored
Refs #36005 -- Made OperationCategory subclass StrEnum.
1 parent 2c87f51 commit a77be30

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

  • django/db/migrations/operations

django/db/migrations/operations/base.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
1-
import enum
1+
from enum import StrEnum
22

33
from django.db import router
44
from django.utils.inspect import get_func_args
55

66

7-
class OperationCategory(str, enum.Enum):
7+
class OperationCategory(StrEnum):
88
ADDITION = "+"
99
REMOVAL = "-"
1010
ALTERATION = "~"
@@ -113,8 +113,8 @@ def formatted_description(self):
113113
"""Output a description prefixed by a category symbol."""
114114
description = self.describe()
115115
if self.category is None:
116-
return f"{OperationCategory.MIXED.value} {description}"
117-
return f"{self.category.value} {description}"
116+
return f"{OperationCategory.MIXED} {description}"
117+
return f"{self.category} {description}"
118118

119119
@property
120120
def migration_name_fragment(self):

0 commit comments

Comments
 (0)