我有一个很长的数据迁移,我正在做的是纠正早期的错误迁移,其中某些行创建不正确。我正在尝试根据旧列为新列分配值,但是,有时这会导致完整性错误。当发生这种情况时,我想扔掉造成完整性错误的那个
这是一段代码:
def load_data(apps, schema_editor):
MyClass = apps.get_model('my_app', 'MyClass')
new_col_mapping = {old_val1: new_val1, ....}
for inst in MyClass.objects.filter(old_col=c):
try:
inst.new_col = new_col_mapping[c]
inst.save()
except IntegrityError:
inst.delete()
然后在我的Migration
课程的操作中
operations = [
migrations.RunPython(load_data)
]
运行迁移时出现以下错误
django.db.transaction.TransactionManagementError: An error occurred in the current transaction. You can't execute queries until the end of the 'atomic' block
我觉得这样做
with transaction.atomic():
某处是我的解决方案,但我不确定正确的地方在哪里。更重要的是,我想了解为什么这是必要的
这类似于example in the docs。
首先,如果您还没有,请添加所需的导入。
from django.db import transaction
然后包装可能在原子块中引发完整性错误的代码。
try:
with transaction.atomic():
inst.new_col = new_col_mapping[c]
inst.save()
except IntegrityError:
inst.delete()
警告块“避免在原子内捕获异常!”中解释了错误的原因。在文档中。一旦Django遇到数据库错误,它将回滚原子块。尝试更多的数据库查询将导致你看到的TransactionManagementError
。通过将代码包装在原子块中,只会回滚该代码,并且可以在块之外执行查询。
每次迁移都围绕一个事务处理,因此在迁移期间出现故障时,所有操作都将被取消。因此,每次事务失败的事务都不能接受新的查询(无论如何都会被取消)。
用with transaction.atomic():
包装一些操作并不是一个好的解决方案,因为当某些东西失败时你将无法取消该操作。而不是那样,通过在保存数据之前进行更多检查来避免完整性错误。
似乎同一个异常可能有多种原因。在我的情况下,它是由一个无效的模型字段名称引起的:我在我的字段名称中使用了希腊字母delta 𐤃
。
它似乎工作正常,所有应用程序运行良好(也许我只是没有尝试任何更复杂的用例)。然而,测试提出了TransactionManagementError
。
我通过从字段名称和所有迁移文件中删除𐤃
来解决问题。