如何動態設置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)
發佈了45 篇原創文章 · 獲贊 2 · 訪問量 6萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章