一個非常實用的小方法

試想一下，Django中如果我們想對儲存進資料庫的資料做校驗，有哪些實現的方法？

我們可以在view中去處理，每當view接收請求，就對提交的資料做校驗，校驗不通過直接返回錯誤，不寫資料庫，校驗通過再呼叫create或update方法寫入資料庫

以上方式比較簡單，容易理解，但隨之又帶來了麻煩，我們需在所有接收資料的地方都要去校驗，那麼有沒有更加優雅的方式呢？如果你看過我之前的文章『Django使用Signals監測model欄位變化傳送通知』]就能想到可以通過signals訊號來處理，新增一個pre_save的訊號，每當資料庫資料變更前都會觸發pre_save方法，可以在這裡進行校驗，免去了view中多個地方校驗的麻煩

而今天要說的並不是signals，而是另一種比較常用的做法：重寫model的save方法

重寫save方法

save方法的主要作用就是將一個物件儲存到資料庫。如果我們想在資料入庫之前做一些處理，除了上邊提到的signals之外，還可以通過重寫save方法來實現。具體實現方式看下面這個例子

假如我們定義了model如下：

class TempTask(models.Model):
  ...
  
  exechost = models.CharField(max_length=64,default='localhost',verbose_name='執行主機')
  execuser = models.ForeignKey(ExecUser,null=True,on_delete=models.PROTECT,db_constraint=False)

exechost預設為Localhost，execuser預設為空，現有需求：當exechost不為localhost時，他必須符合ip:port的格式，且

execuser不能為空。這是一個比較複雜的校驗方式，我們可以通過重寫save方法來處理

class TempTask(models.Model):
  ...

  def save(self,*args,**kwargs):
    if self.exechost and (self.exechost.strip() != 'localhost'):
      if len(self.exechost.split(':')) != 2:
        raise ValidationError('執行主機格式錯誤，應為ip:port格式')

      if not self.execuser:
        raise ValidationError('當執行主機存在時執行使用者不能為空')

    super().save(*args,**kwargs)

我們可以在save函式內執行各種自定義邏輯，但需要注意的是，最後必須要呼叫super().save()方法來保證執行了父類的save()，這樣才能保證資料寫入了資料庫。

這樣在當我們執行create語句插入資料的時候就會先去執行save中的校驗方法進行校驗了

TempTask.objects.create(**postdata)

update踩坑

就當我以為一切都要結束準備起身衝杯咖啡的時候，我發現新加資料可以正常進行校驗，但更新資料卻不行，更新的程式碼如下：

TempTask.objects.filter(id=pk).update(**postdata)

經過一番查詢發現了問題所在，官方文件中有這麼一句話

Unfortunately,there isn't a workaround when creating or updating objects in bulk,since none of save(),pre_save,and post_save are called.

也就是說，當使用查詢集批量更新物件時，將不會為每個物件呼叫save()方法，連pre_save和post_save也不會被呼叫。與save()類似的還有model的delete()方法，當批量刪除的時候，同樣不會呼叫model的delete()方法，但delete是可以使用pre_delete或post_delete訊號的

解決這個問題的方法很簡單，那就是將更新的程式碼換成下邊這種，保證呼叫到save方法

_t = TempTask.objects.get(id=pk)
_t.__dict__.update(**postdata)
_t.save()

補充知識：django model save方法對未更改的欄位依然進行了儲存

看程式碼吧~

obj = Obj.objects.get(id=1)
print obj.name #此時name的值假定為'abc'
 
def handler(oid):
  obj = Obj.objects.get(id=oid)
  obj.name = '123'
  obj.save()
handler(obj.id)
obj.age = 10
obj.save()
print obj.name 

最終的name結果依然為'abc'。save()儲存時,雖然沒有更改其它欄位，但依然會將記憶體中的值，再次存入資料庫，子函式和其它程序更改的值會被覆蓋。

以上這篇Django model重寫save方法及update踩坑詳解就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支援我們。