• Django model重写save方法及update踩坑记录


    一个非常实用的小方法

    试想一下,Django中如果我们想对保存进数据库的数据做校验,有哪些实现的方法?

    我们可以在view中去处理,每当view接收请求,就对提交的数据做校验,校验不通过直接返回错误,不写数据库,校验通过再调用createupdate方法写入数据库

    以上方式比较简单,容易理解,但随之又带来了麻烦,我们需在所有接收数据的地方都要去校验,那么有没有更加优雅的方式呢?如果你看过我之前的文章『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_savepost_save也不会被调用。与save()类似的还有model的delete()方法,当批量删除的时候,同样不会调用model的delete()方法,但delete是可以使用pre_deletepost_delete信号的

    解决这个问题的方法很简单,那就是将更新的代码换成下边这种,保证调用到save方法

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

    扫码关注公众号查看更多实用文章

  • 相关阅读:
    用pelican搭建完美博客
    对比MySQL,什么场景MongoDB更适用
    客官,您的 Flask 全家桶请收好
    虚拟机安装macos 分辨率不正常修改不了,不能全屏如何解决
    mac开启HiDPI
    虚拟机安装MacOS|unlocker解锁出现闪退问题!
    下载com.vmware.fusion.tools.darwin.zip.tar慢
    VMware虚拟机安装黑苹果MacOS Mojave系统详细教程
    Python格式化输出
    ubuntu进入initramfs,系统黑屏
  • 原文地址:https://www.cnblogs.com/37Y37/p/12840890.html
Copyright © 2020-2023  润新知