一個(gè)非常實(shí)用的小方法

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

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

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

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

重寫save方法

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

假如我們定義了model如下：

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

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

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

      if not self.execuser:
        raise ValidationError('當(dāng)執(zhí)行主機(jī)存在時(shí)執(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ù)的時(shí)候就會(huì)先去執(zhí)行save中的校驗(yàn)方法進(jìn)行校驗(yàn)了

TempTask.objects.create(**postdata)

update踩坑

就當(dāng)我以為一切都要結(jié)束準(zhǔn)備起身沖杯咖啡的時(shí)候，我發(fā)現(xiàn)新加數(shù)據(jù)可以正常進(jìn)行校驗(yàn)，但更新數(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)使用查詢集批量更新對(duì)象時(shí)，將不會(huì)為每個(gè)對(duì)象調(diào)用save()方法，連pre_save和post_save也不會(huì)被調(diào)用。與save()類似的還有model的delete()方法，當(dāng)批量刪除的時(shí)候，同樣不會(huì)調(diào)用model的delete()方法，但delete是可以使用pre_delete或post_delete信號(hào)的

解決這個(gè)問題的方法很簡(jiǎn)單，那就是將更新的代碼換成下邊這種，保證調(diào)用到save方法

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

補(bǔ)充知識(shí)：django model save方法對(duì)未更改的字段依然進(jìn)行了保存

看代碼吧~

obj = Obj.objects.get(id=1)
print obj.name #此時(shí)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()保存時(shí),雖然沒有更改其它字段，但依然會(huì)將內(nèi)存中的值，再次存入數(shù)據(jù)庫，子函數(shù)和其它進(jìn)程更改的值會(huì)被覆蓋。

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

最新評(píng)論