在 django 中测试时的模型创建顺序？

Question

我有这两个模型（示例），当我试图运行我的测试时 - 它错误地说：no such table: my_app_modelA - 如果我向上滚动我可以看到它在创建 modelB 时被炸毁了（我认为这是由于应用了default）。有没有办法订购这些，以便modelA 总是在modelB 之前创建？还是我不应该将该方法作为默认属性引用？只是想让我的测试正常工作，这是我的症结所在。

我的模型如下所示：

class modelA(models.Model):
    attribute = models.IntegerField()
    active = models.BooleanField(default=False)

    @classmethod
    def get_active_attribute(cls):
        return modelA.objects.get(active=True).attribute

class modelB(models.Model):
    attribute = models.IntegerField(default=modelA.get_active_attribute())

我的问题是：

• 这是可以接受的做法吗？默认调用另一个模型方法？

• 有没有办法处理这些模型的创建，我可以保证首先创建 modelA，以便 modelB 可以在我的测试中成功创建？

Answer 1

首先，迁移按照创建迁移文件时定义的顺序进行。

# 0001_initial.py
...
operations = [
    migrations.CreateModel(
        name=modelA,
        ....
    ),
    migrations.CreateModel(
        name=modelB,
        ....
    ),
]

您可以检查您的迁移文件并确保modelA 在modelB 之前。

其次，modelA.get_active_attribute() 需要一个 DB 条目才能返回某些内容。运行迁移时，您不会插入数据。所以你不应该用其他模型的对象声明default。 您应该改写 save() 以确保默认值基于 modelA 的属性。

class modelB(models.Model):
    attribute = models.IntegerField()

    def save(self, *args, **kwargs):
        if self.attribute is None:
            self.attribute = modelA.get_active_attribute()
        super(modelB, self).save(*args, **kwargs)