一個非常實用的小方法

試想一下，Django中如果我們想對保存進數(shù)據(jù)庫的數(shù)據(jù)做校驗，有哪些實現(xiàn)的方法？

我們可以在view中去處理，每當(dāng)view接收請求，就對提交的數(shù)據(jù)做校驗，校驗不通過直接返回錯誤，不寫數(shù)據(jù)庫，校驗通過再調(diào)用create或update方法寫入數(shù)據(jù)庫

以上方式比較簡單，容易理解，但隨之又帶來了麻煩，我們需在所有接收數(shù)據(jù)的地方都要去校驗，那么有沒有更加優(yōu)雅的方式呢？如果你看過我之前的文章『Django使用Signals監(jiān)測model字段變化發(fā)送通知』]就能想到可以通過signals信號來處理，添加一個pre_save的信號，每當(dāng)數(shù)據(jù)庫數(shù)據(jù)變更前都會觸發(fā)pre_save方法，可以在這里進行校驗，免去了view中多個地方校驗的麻煩

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

重寫save方法

save方法的主要作用就是將一個對象保存到數(shù)據(jù)庫。如果我們想在數(shù)據(jù)入庫之前做一些處理，除了上邊提到的signals之外，還可以通過重寫save方法來實現(xiàn)。具體實現(xiàn)方式看下面這個例子

假如我們定義了model如下：

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

exechost默認為Localhost，execuser默認為空，現(xiàn)有需求：當(dāng)exechost不為localhost時，他必須符合ip:port的格式，且

execuser不能為空。這是一個比較復(fù)雜的校驗方式，我們可以通過重寫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('執(zhí)行主機格式錯誤，應(yīng)為ip:port格式')

      if not self.execuser:
        raise ValidationError('當(dāng)執(zhí)行主機存在時執(zhí)行用戶不能為空')

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

我們可以在save函數(shù)內(nèi)執(zhí)行各種自定義邏輯，但需要注意的是，最后必須要調(diào)用super().save()方法來保證執(zhí)行了父類的save()，這樣才能保證數(shù)據(jù)寫入了數(shù)據(jù)庫。

這樣在當(dāng)我們執(zhí)行create語句插入數(shù)據(jù)的時候就會先去執(zhí)行save中的校驗方法進行校驗了

TempTask.objects.create(**postdata)

update踩坑

就當(dāng)我以為一切都要結(jié)束準(zhǔn)備起身沖杯咖啡的時候，我發(fā)現(xiàn)新加數(shù)據(jù)可以正常進行校驗，但更新數(shù)據(jù)卻不行，更新的代碼如下：

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

經(jīng)過一番查找發(fā)現(xiàn)了問題所在，官方文檔中有這么一句話

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

也就是說，當(dāng)使用查詢集批量更新對象時，將不會為每個對象調(diào)用save()方法，連pre_save和post_save也不會被調(diào)用。與save()類似的還有model的delete()方法，當(dāng)批量刪除的時候，同樣不會調(diào)用model的delete()方法，但delete是可以使用pre_delete或post_delete信號的

解決這個問題的方法很簡單，那就是將更新的代碼換成下邊這種，保證調(diào)用到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結(jié)果依然為'abc'。save()保存時,雖然沒有更改其它字段，但依然會將內(nèi)存中的值，再次存入數(shù)據(jù)庫，子函數(shù)和其它進程更改的值會被覆蓋。

以上這篇Django model重寫save方法及update踩坑詳解就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

最新評論