如何引入独特的数据冲突?

时间:2015-01-08 05:02:19

标签: django django-1.7 django-database django-migrations

我为我的网站编写了一个小qna脚本,并阻止用户开始讨论我希望每个用户只能回复一次。

class Q(models.Model):
  text = models.TextField()
  user = models.ForeignKeyField('auth.User')

class A(models.Model):
  text = models.TextField()
  user = models.ForeignKeyField('auth.User')
  q = models.ForeignKeyField('Q')
  class Meta:
    unique_together = (('user','q'),)

现在迁移给了我:

return Database.Cursor.execute(self, query, params)
  django.db.utils.IntegrityError: columns user_id, q_id are not unique

当然,与现有数据的独特冲突。我现在需要知道的是如何告诉迁移删除冲突的答案。保持第一个发现的愚蠢解决方案已经是一个很大的帮助。更好的方法是通过自定义函数比较冲突的A.

我正在使用新的迁移系统运行Django-1.7 - 而不是南方。

感谢您的帮助!

1 个答案:

答案 0 :(得分:4)

您只需要创建一个数据迁移,您可以在其中编写自定义函数以使用您想要的任何逻辑。有关详细信息,请参阅the documentation

例如,仅保留最低Answer id(应该是最早答案的良好代理)的数据迁移可能如下所示:

from django.db import models, migrations

def make_unique(apps, schema_editor):
    A = apps.get_model("yourappname", "A")

    # get all unique pairs of (user, question)
    all_user_qs = A.objects.values_list("user_id", "q_id").distinct()

    # for each pair, delete all but the first
    for user_id, q_id in all_user_qs:
        late_answers = A.objects.filter(user_id=user_id, q_id=q_id).order_by('id')[1:]
        A.objects.filter(id__in=list(late_answers)).delete()

class Migration(migrations.Migration): 

    dependencies = [ 
        ('yourappname', '0001_initial'),
    ]

    operations = [ 
        migrations.RunPython(make_unique), 
    ]

(这是我的头脑,当然是破坏性的,所以请以此为例。您可能希望在执行所有删除操作之前先查看备份数据。)

回顾一下:删除您尝试运行的迁移并删除唯一约束。按文档中的说明创建空数据迁移。编写一个函数来删除数据库中当前存在的非唯一数据。然后重新添加唯一约束并运行makemigrations。最后,使用migrate运行两次迁移。