python枚举法例子_Django在Model中枚举ENUM表字段创建的例子——PYTHON

本文介绍Django开发中通过model创建MYSQL数据库中包含ENUM枚举类型字段的表，方法如下：

from django.db import models

class EnumField(models.Field):

def __init__(self, *args, **kwargs):

super(EnumField, self).__init__(*args, **kwargs)

assert self.choices, “Need choices for enumeration”

def db_type(self, connection):

if not all(isinstance(col, basestring) for col, _ in self.choices):

raise ValueError(“MySQL ENUM values should be strings”)

return “ENUM({})”.format(‘,’.join(“‘{}'”.format(col)

for col, _ in self.choices))

class IceCreamFlavor(EnumField, models.CharField):

def __init__(self, *args, **kwargs):

flavors = [(‘chocolate’, ‘Chocolate’),

(‘vanilla’, ‘Vanilla’),

]

super(IceCreamFlavor, self).__init__(*args, choices=flavors, **kwargs)

class IceCream(models.Model):

price = models.DecimalField(max_