問題背景
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)