如何在django模型中使用枚举作为选择字段
回答问题 我有一个模型类,我希望其中两个字段作为选择字段,因此要填充这些选择,我使用了下面列出的枚举 #models.py class Transaction(models.Model): trasaction_status = models.CharField(max_length=255, choices=TransactionStatus.choices()) transaction_typ
·
回答问题
我有一个模型类,我希望其中两个字段作为选择字段,因此要填充这些选择,我使用了下面列出的枚举
#models.py
class Transaction(models.Model):
trasaction_status = models.CharField(max_length=255, choices=TransactionStatus.choices())
transaction_type = models.CharField(max_length=255, choices=TransactionType.choices())
#enums.py
class TransactionType(Enum):
IN = "IN",
OUT = "OUT"
@classmethod
def choices(cls):
print(tuple((i.name, i.value) for i in cls))
return tuple((i.name, i.value) for i in cls)
class TransactionStatus(Enum):
INITIATED = "INITIATED",
PENDING = "PENDING",
COMPLETED = "COMPLETED",
FAILED = "FAILED"
ERROR = "ERROR"
@classmethod
def choices(cls):
print(tuple((i.name, i.value) for i in cls))
return tuple((i.name, i.value) for i in cls)
但是,当我尝试通过管理员访问此模型时,我收到以下错误:
Django Version: 1.11
Exception Type: ValueError
Exception Value:
too many values to unpack (expected 2)
我关注了两篇描述如何使用枚举的文章:
-
https://hackernoon.com/using-enum-as-model-field-choice-in-django-92d8b97aaa63
-
https://blog.richard.do/2014/02/18/how-to-use-enums-for-django-field-choices/
Answers
您的代码中的问题是INITIATED = "INITIATED",
在INITIATED
选项和其他选项之后的逗号。当我们在任何字符串后添加逗号时,它将成为一个元组。请参阅下面的示例
s = 'my str'
print(type(s))
# output: str
s = 'my str',
print(type(s))
# output: tuple
#models.py
class Transaction(models.Model):
trasaction_status = models.CharField(max_length=255, choices=TransactionStatus.choices())
transaction_type = models.CharField(max_length=255, choices=TransactionType.choices())
#枚举.py
class TransactionType(Enum):
IN = "IN"
OUT = "OUT"
@classmethod
def choices(cls):
print(tuple((i.name, i.value) for i in cls))
return tuple((i.name, i.value) for i in cls)
class TransactionStatus(Enum):
INITIATED = "INITIATED"
PENDING = "PENDING"
COMPLETED = "COMPLETED"
FAILED = "FAILED"
ERROR = "ERROR"
@classmethod
def choices(cls):
print(tuple((i.name, i.value) for i in cls))
return tuple((i.name, i.value) for i in cls)
对于 django > 3.0https://docs.djangoproject.com/en/4.0/ref/models/fields/#field-choices-enum-types
更多推荐
已为社区贡献126483条内容
所有评论(0)