django - 通过数据迁移添加默认标签

Question

我正在使用hierarchical tags与 taggit 一起使用和 treebeard 。我正在尝试进行数据迁移，以便可以定义将出现在应用程序的所有实例中的标签。

我定义了这个方法:

def define_tags(apps, schema_editor):
    HierarchicalTag = apps.get_model("aion", "HierarchicalTag")
    root = HierarchicalTag.add_root(name='root')
    root.save()
    leaf = HierarchicalTag.objects.get(pk=root.pk).add_child(name='ook')
    leaf.save()

它应该创建两个标签“root”和一个子标签“ook”。但是，当我运行迁移时，出现此错误:

AttributeError: type object 'HierarchicalTag' has no attribute 'add_root'

方法add_root是来自treebeard的MP_Node的类方法。 .

我该如何解决这个问题？

Answer 1

TLDR:如果您使用 app.get_model 而不是导入它，则无法使用 Treebeard API。

当您使用时

HierarchicalTag = apps.get_model("aion", "HierarchicalTag")

要获取 HeirarchicalTag 模型而不是导入它，实际上您并没有获取 Treebeard API。您没有获得代码中定义的模型，而是获得了 django 根据您的迁移推断的模型版本。这就是为什么在您删除某个模型的实际代码后 app.get_model 将继续工作的原因(这就是您使用 app.get_model 的原因)在迁移而不是导入中)。

解决方案:

1. 编写一个注定会失败的迁移，仅导入 HierarchicalTag。
2. 继续使用 app.get_model，但随后手动插入根(请参阅 the django-treebeard api 以找出需要手动填写哪些字段)。