使用模型方法作为 Django 模型的默认值？

Question

我的模型中实际上有这个方法：

def speed_score_compute(self):
  # Speed score:                                                                                                     
  #                                                                                                                  
  #   - 8 point for every % of time spent in                                                                         
  #     high intensity running phase.                                                                                
  #   - Average Speed in High Intensity                                                                              
  #     running phase (30km/h = 50 points                                                                            
  #     0-15km/h = 15 points )

  try:
    high_intensity_running_ratio = ((
      (self.h_i_run_time * 100)/self.training_length) * 8)
  except ZeroDivisionError:
    return 0
  high_intensity_running_ratio = min(50, high_intensity_running_ratio)
  if self.h_i_average_speed < 15:
    average_speed_score = 10
  else:
    average_speed_score = self.cross_multiplication(
      30, self.h_i_average_speed, 50)
  final_speed_score = high_intensity_running_ratio + average_speed_score
  return final_speed_score

我想像这样将它用作我的模型的默认值：

speed_score = models.IntegerField(default=speed_score_compute)

但这不起作用（请参阅下面的错误消息）。我检查了不同的主题，例如 this one，但这仅适用于函数（不使用 self 属性）而不适用于方法（我必须使用方法，因为我正在使用实际的对象属性）。

Django doc. 似乎在谈论这个，但我不太清楚可能是因为我还是 Django 和 Programmation 的新手。

有没有办法做到这一点？

编辑：

我的功能在我的模型之上定义。但这是我的错误信息：

ValueError: Could not find function speed_score_compute in tournament.models. Please note that due to Python 2 limitations, you cannot serialize unbound method functions (e.g. a method declared and used in the same class body). Please move the function into the main module body to use migrations. For more information, see https://docs.djangoproject.com/en/1.8/topics/migrations/#serializing-values

错误信息很清楚，我似乎无法做到这一点。但是还有其他方法可以实现吗？

Answer 1

问题

当我们提供 default=callable 并提供来自模型的方法时，它不会被作为模型实例的 self 参数调用。

覆盖save()

我没有找到比覆盖MyModel.save()* 更好的解决方案，如下所示：

class MyModel(models.Model):
    def save(self, *args, **kwargs):
        if self.speed_score is None:
            self.speed_score = ...
            # or
            self.calculate_speed_score()

        # Now we call the actual save method
        super(MyModel, self).save(*args, **kwargs)

这样，如果您尝试保存模型，而该字段没有设置值，它会在保存之前填充。

就我个人而言，我只是更喜欢这种方法，即在模型中定义属于模型的所有内容（数据、方法、验证、默认值等）。我认为这种方法被称为胖 Django 模型方法。

_{*如果你找到更好的方法，我也想学习一下！}

使用pre_save 信号

Django 提供了一个pre_save 信号，该信号在save() 在模型上运行之前运行。信号同步运行，即 pre_save 代码需要在模型上调用 save() 之前完成运行。您将获得与覆盖 save() 相同的结果（和执行顺序）。

from django.db.models.signals import pre_save
from django.dispatch import receiver
from myapp.models import MyModel


@receiver(pre_save, sender=MyModel)
def my_handler(sender, **kwargs):
    instance = kwargs['instance']
    instance.populate_default_values()

如果您希望将默认值行为与模型分开，这种方法适合您！

什么时候调用save()？我可以在对象保存之前使用它吗？

好问题！因为我们希望能够在保存填充默认值之前使用我们的对象。

要获取一个对象，而不保存它，我们可以这样做：

instance = MyModel()

如果您创建它使用MyModel.objects.create()，那么save() 将被调用。它本质上是(see source code) 等价于：

instance = MyModel()
instance.save()

如果你感兴趣，你也可以定义一个MyModel.populate_default_values()，你可以在对象生命周期的任何阶段调用它（在创建、保存或按需，这取决于你）