django model save - 在迁移期间未调用的覆盖方法

Question

我的模型类中有一个保存覆盖方法,每次保存一个对象时都会生成一个新的slug.

def save(self, *args, **kwargs):
    if self.column2:
        self.slug = slugify(self.column1 + " " + self.column2)
    else:
        self.slug = slugify(self.column1)
    print slug
    super(MyModel, self).save(*args, **kwargs)

当我尝试通过登录python shell创建一个新对象时,我看到正在调用save方法.

python manage.py shell

>>> MyModel(column1="test",column2="2015").save()
slug is test-2015

但是,当我运行迁移时,不会调用此保存覆盖方法.这是我的迁移脚本的一部分..

...
def add_myModel_details(apps, schema_editor):
    x = apps.get_model("myapp","myModel")
    MyModel(column1 = "test", column2="2015" ).save()
.....

slug为空,因为没有调用保存覆盖.

Answer 1

自定义模型方法在迁移期间不可用。

相反，您可以在RunPython函数中运行 put 代码，以自定义方式修改模型实例save()。

参考：

• 这个回答

Answer 2

发生这种情况是因为迁移不会调用您的save方法。

我认为 save 方法不是生成 slug 的最佳位置。使用AutoSlugField或Signals会更好。

1. 信号：

在您的情况下，您可以使用信号pre_save。

例子：

@receiver(pre_save, sender=MyModel)
def my_handler(sender, **kwargs):
    my_model = kwargs.get('instance')
    if my_model.column2:
        my_model.slug = slugify(my_model.column1 + " " + my_model.column2)
    else:
        my_model.slug = slugify(my_model.column1)
    print my_model.slug

2. AutoSlugField：

它不是一个标准字段，但很多库都实现了它。我使用django-extensions中的AutoSlugField。该字段也使用信号。

例子：

slug = AutoSlugField(populate_from=("column1", "column2"))

3.save方法和迁移

但如果您仍然想使用save生成 slug 的方法，我建议您创建数据迁移并手动添加 slug。

数据迁移 django >= 1.7

数据向南迁移