加载包含自然键的Django灯具

时间:2012-03-01 00:48:08

标签: python django django-models natural-key

如何加载Django夹具,以便通过自然键引用的模型不会与预先存在的记录发生冲突?

我试图加载这样的工具,但我从MySQL后端获取了IntegrityErrors,抱怨Django尝试插入重复记录,这没有任何意义。

由于我了解Django的自然关键功能,为了完全支持dumpdata和loaddata使用,您需要在模型中定义natural_key方法,并在其中定义get_by_natural_key方法模特的经理。

所以,例如,我有两个模型:

class PersonManager(models.Manager):
    def get_by_natural_key(self, name):
        return self.get(name=name)

class Person(models.Model):

    objects = PersonManager()

    name = models.CharField(max_length=255, unique=True)

    def natural_key(self):
        return (self.name,)

class BookManager(models.Manager):
    def get_by_natural_key(self, title, *person_key):
        person = Person.objects.get_by_natural_key(*person_key)
        return self.get(title=title, person=person)

class Book(models.Model):

    objects = BookManager()

    author = models.ForeignKey(Person)

    title = models.CharField(max_length=255)

    def natural_key(self):
        return (self.title,) + self.author.natural_key()
    natural_key.dependencies = ['myapp.Person']

我的测试数据库已经包含一个示例Person和Book记录,我用它来生成夹具:

[
    {
        "pk": null, 
        "model": "myapp.person", 
        "fields": {
            "name": "bob"
        }
    }, 
    {
        "pk": null, 
        "model": "myapp.book", 
        "fields": {
            "author": [
                "bob"
            ], 
            "title": "bob's book", 
        }
    }
]

我希望能够使用此fixture并将其加载到我的数据库的任何实例中以重新创建记录,无论它们是否已存在于数据库中。

然而,当我运行python manage.py loaddata myfixture.json时,我收到错误:

IntegrityError: (1062, "Duplicate entry '1-1' for key 'myapp_person_name_uniq'")

为什么Django会尝试重新创建Person记录而不是重用那些已经存在的记录?

2 个答案:

答案 0 :(得分:1)

原来这个解决方案需要一个非常小的补丁来处理Django的loaddata命令。由于Django开发人员不太可能接受这样的补丁,因此我在各种Django管理员相关增强功能的包中都有forked it

密钥代码更改(loaddatanaturally.py的第189-201行)只需要调用get_natural_key()来查找循环内遍历反序列化对象的任何现有pk。

答案 1 :(得分:0)

实际上,loaddata不适用于数据库中的现有数据,通常用于初始加载模型。 看看这个问题的另一种方法:Import data into Django model with existing data?