django 自定义.save()方法

在django中，我们有时候需要自己控制数据库的存储。这就需要我们重写模型的.save()方法。

一般来说，我们可以这样写：

class Student(models.Model):
    username = models.CharField('学生姓名', max_length=16)
    age = models.IntegerField('年龄')

    def save(self, *args, **kwargs):
        do_something()
        super().save(*args, **kwargs)  # 执行真正的 .save() 方法.
        do_something_else()

一、举个例子，使用do_something_else()的场景：

我们创建订单，订单有一个编号the_id是根据创建日期和主键pk生成的。

class Order(models.Model):

    def _create_the_id(self):
        """生成订单编号"""
        return self.created.date().strftime('%Y%m%d') + '_' + str(self.id)

    created = models.DateTimeField('创建时间', auto_now_add=True)
    the_id = models.CharField('订单编号', max_length=32, unique=True, null=True)
    title = models.CharField('订单名称', max_length=32, default='')
    total = models.FloatField('订单金额', default=0.0)

在这里，我们的the_id属性依赖于id，所以我们必须先执行.save()然后再进行更新。

>>> order = Order(title='订单的标题****', total=25.5)
>>> order.save()  # 保存进数据库，获取 id

>>> order.the_id = order._create_the_id()
>>> order.save() # 更新 the_id 属性

这里就有一个问题，我们需要在每个.save()执行后都执行

>>> order.the_id = order._create_the_id()

更好的方法是重写.save()方法

    def save(self, force_insert=False, force_update=False, using=None,
             update_fields=None):
        # 执行 save(), 将数据保存进数据库
        super().save(
            force_insert=force_insert,
            force_update=force_update,
            using=using,
            update_fields=update_fields
        )
        self.the_id = self._create_the_id()

        # 再次执行 save(), 将数据更新到数据库
        # 注意这里的参数，必须设置 force_update=True,否则会创建新的数据
        super().save(
            force_insert=False,
            force_update=True,
            using=using,
            update_fields=['the_id']
        )

当我们重写.save()后，就不用考虑会不会忘记执行更新the_id操作了。现在直接执行.save()就会自动帮我们更新the_id

二、在.save()之前执行do_something()

在数据库中，有些属性属于派生属性，也就是说它们是依据其它属性生成的。

class Order(models.Model):
    created = models.DateTimeField('创建时间', auto_now_add=True)
    title = models.CharField('订单名称', max_length=32, default='')
    num = models.IntegerField('商品数量')
    price = models.FloatField('商品价格')
    total = models.FloatField('订单金额', default=0.0)

在上述订单模型中，total应该是在后端存储时自动计算，而不是接收前端传给我们的数据。

    def save(self, force_insert=False, force_update=False, using=None,
             update_fields=None):

        self.total = self.num * self.price

        # 执行 save(), 将数据保存进数据库
        super().save(
            force_insert=force_insert,
            force_update=force_update,
            using=using,
            update_fields=update_fields
        )