枚举类型django和rest框架

Question

我对其余框架和一个枚举字段有疑问。我收到此错误：

  File "/usr/local/lib/python2.7/dist-packages/coreapi/codecs/corejson.py", line 48, in decode_schema_from_corejson
    return schema_cls(title=title, description=description)
TypeError: __init__() takes exactly 2 arguments (1 given)

当模型有这个字段时：

class PuddleAtomStatus(ChoiceEnum):
    INVALID = "INVALID"
    OK = "OK"

class PuddleAtom(Model):
    name = models.CharField(max_length=255, primary_key=True)
    ingestion_flow = models.CharField(max_length=256)
    status = EnumChoiceField(enm_class=PuddleAtomStaus) 

但是当

status = models.CharField(max_length=100)

我正在使用 Django 1.11 和 python 2.7

有没有办法解决这个问题并让模型保留 enum 字段？

Answer 1

在高层次上，我可以说这就像 DRF 不知道这个特殊的 EnumField 是什么一样简单。因此，您可能必须编写自己的序列化程序。

Answer 2

我建议这样做：

class PuddleAtom(Model):
    STATUS_CHOICES = (
                        ('INVALID', 'INVALID'),
                        ('OK', 'OK'),
                     )
    name = models.CharField(max_length=255, primary_key=True)
    ingestion_flow = models.CharField(max_length=256)
    status = models.CharField(max_length=100, choices=STATUS_CHOICES)

我希望这会有所帮助。 这里是对应的documentation