我为我的网站编写了一个小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 - 而不是南方。
感谢您的帮助!
答案 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
运行两次迁移。