如何动态设置django的model field的默认值

问题背景

django的model field需要动态设置默认值,具体案例如下:
原始代码如下,model是Application,其中字段ignore_fort的默认值设置为False

class Application(TimestampedModel):
    name = models.CharField(max_length=255, null=True)
    ignore_fort = models.BooleanField(default=False)

然而现在有这样一个需求:default需要根据某个变量ENV进行动态设置,如果ENV是UAT或者FAT(不区分大小写,具体分支,比如uaT01也可以),则default设置为True,否则设置为False

首先想到的是如下代码:

class Application(TimestampedModel):
    name = models.CharField(max_length=255, null=True)
    ignore_fort = models.BooleanField(default= 'UAT'in ENV.upper() or 'FAT' in ENV.upper())

通过python manage.py shell_plus启动shell调试,发现如果ENV本来设置的是什么值,逻辑正确,如果在shell中修改ENV的值,则新建的model的ignore_fort值并不是根据当前ENV值进行设置,而是保持原来的值,达不到需求。例如,ENV值本来设置为uat,那么新建app = Application()print app.ignore_fort结果是True,修改ENV ENV = 'hhh'app1 = Application(); print app1.ignore_fort结果还是True,而我们需要的是False。

分析

官方描述如下:
The default value for the field. This can be a value or a callable object. If callable it will be called every time a new object is created.
如果想要在创建对象时动态修改default的值,需要用callable object,可以理解为函数调用?
上述代码的default值并不是callable object,所以并不是在object创建时同台更新。

解决方法

参考:http://stackoverflow.com/questions/12649659/how-to-set-a-django-model-fields-default-value-to-a-function-call-callable-e
正解如下,用一个函数调用实现callable object,进而实现动态跟新default

def get_default_ignore_fort():
    cur_env =ENV.upper()
    return any(i in cur_env for i in ('UAT', 'FAT'))


class Application(TimestampedModel):
    name = models.CharField(max_length=255, null=True)
    ignore_fort = models.BooleanField(default=get_default_ignore_fort)

你可能感兴趣的:(python)