Django迁移RunPython无法调用模型方法

Question

我正在使用RunPython方法创建一个数据迁移。但是，当我尝试在对象上运行一个方法时，没有定义任何方法。是否可以使用RunPython调用在模型上定义的方法？

Answer 1

模型方法在迁移中不可用，包括数据迁移。

但是，有一种变通方法，它应该与调用模型方法非常相似。您可以在迁移中定义模仿您想要使用的模型方法的函数。

如果你有这种方法：

class Order(models.Model):
    '''
    order model def goes here
    '''
    def get_foo_as_bar(self):
        new_attr = 'bar: %s' % self.foo
        return new_attr

您可以在迁移脚本中编写如下函数：

def get_foo_as_bar(obj):
    new_attr = 'bar: %s' % obj.foo
    return new_attr

def save_foo_as_bar(apps, schema_editor):
    old_model = apps.get_model("order", "Order")
    for obj in old_model.objects.all():
        obj.new_bar_field = get_foo_as_bar(obj)
        obj.save()

然后在迁移中使用它：

class Migration(migrations.Migration):
    dependencies = [
        ('order', '0001_initial'),
    ]
    operations = [
        migrations.RunPython(save_foo_as_bar)
    ]

这样迁移就可以工作了。代码会有一些重复，但这并不重要，因为在应用程序的特定状态下，数据迁移应该是一次性操作。

Answer 2

您是否像文档中所说的那样调用了您的模型？

def combine_names(apps, schema_editor):
    # We can't import the Person model directly as it may be a newer
    # version than this migration expects. We use the historical version.
    Person = apps.get_model("yourappname", "Person")
    for person in Person.objects.all():
        person.name = "%s %s" % (person.first_name, person.last_name)
        person.save()

数据迁移因为在这一点上，你不能直接导入你的模型：

from yourappname.models import Person

更新

Django内部代码位于Django/db/migrations/state.py文件中django.db.migrations.state.ModelState#construct_fields

def construct_fields(self):
    "Deep-clone the fields using deconstruction"
    for name, field in self.fields:
        _, path, args, kwargs = field.deconstruct()
        field_class = import_string(path)
        yield name, field_class(*args, **kwargs)

在"伪"模型实例中只有克隆的字段：

MyModel.__module__ = '__fake__'

Github Django

Answer 3

精细打印在历史模型中

因为不可能序列化任意的Python代码，所以这些历史模型将没有您定义的任何自定义方法。

当我在迁移过程中第一次遇到它时，我感到非常惊讶，因为它似乎与他们的设计理念（围绕模型添加功能）相矛盾，所以没有阅读详细信息。

Answer 4

从Django 1.8开始，您可以通过在模型管理器上设置use_in_migrations = True来使模型管理器可用于迁移。请参阅迁移文档。

Answer 5

这不会回答OP，但可能对某些人仍然有用。

自定义模型方法不仅在迁移中不可用，而且对其他模型属性（如类"）也是如此；常数"；用于模型字段CCD_ 4。请参阅文档中的示例。

在这种特定的边缘情况下，我们无法在迁移过程中直接访问选项的历史值，但我们可以使用model_meta-api从模型字段中获取历史值，因为这些值包含在迁移中。

给定Django的Student示例：

class Student(models.Model):
    FRESHMAN = 'FR'
    ...
    YEAR_IN_SCHOOL_CHOICES = [(FRESHMAN, 'Freshman'), ...]
    year_in_school = models.CharField(
        max_length=2,
        choices=YEAR_IN_SCHOOL_CHOICES,
        default=FRESHMAN,
    )

我们可以得到迁移中Student.FRESHMAN的历史值，如下所示：

...
Student = apps.get_model('my_app', 'Student')
YEAR_IN_SCHOOL_CHOICES = Student._meta.get_field('year_in_school').choices
...

Answer 6

当你有许多复杂的方法相互调用，并且你需要通过你的对象提供它们时，对我有用的东西：

首先将这些模型方法复制到迁移文件中

def A(self):
    return self.B() + self.C()
def B(self):
    return self.name
def C(self):
    return self.description

然后在您的迁移功能中：

def do_something_to_your_objects(apps, schema_editor):
    MyModel = apps.get_model("my_app", "MyModel")
    MyModel.A = A
    MyModel.B = B
    MyModel.C = C
    
    for my_object in MyModel.objects.all():
         my_object.name_and_decription = my_object.C()
         my_object.save()
class Migration(migrations.Migration):
    dependencies = [
        ('initial', '0001_initial'),
    ]
    operations = [
        migrations.RunPython(do_something_to_your_objects)
    ]

Answer 7

如果你和我一样，谁来这里是因为你得到了错误ValueError: RunPython must be supplied with a callable这是因为你把"；（）"；在migrations.RunPython 中分配给code的函数的末尾

错误，例如migrations.RunPython(code=do_something(), reverse=noop)

它应该是：不带（）的migrations.RunPython(code=do_something, reverse=noop)