Django复制记录的方法 – Perchouli 的 Blog

最近的Django项目中有复制记录的需求。数据库里有一张名为Party的表，记录用户创建的party，现在要让用户能够复制一个新的party。本身非常简单的一个功能，但运行的时候出错了。我以为是复制过程出错，所以测试了一下Django中复制记录可能遇到的情况(后来发现是其他代码的逻辑出了问题），结果整理如下。

1.Reset pk

在Django中复制一条记录最简单的方法应该是先取出这条记录，将它的主键设置为None，然后再保存一次。


old_party = Party.objects.get(pk=1)
party.pk = None
party.save()

2.Keyword Arguments

如果某张表除了主键还有其他AutoField字段，就要换一种方式了，虽然也可以手动根据字段名将其指定为空，但直接遍历出字段名和相应的值，把其中的AutoField字段赋值为None会比较简单。


data = {}
for f in old_party._meta.fields:
    if f.__class__.__name__ != 'AutoField':
        data[f.name] = getattr(old_party, f.name)
    else:
        data[f.name] = None
party = Party(**data)
party.save()

3.Deepcopy

Django没有直接复制记录的方法，但是Python提供了copy模块用来拷贝一个对象，所以也可以这样复制Django的记录：


import copy
party = copy.deepcopy(old_party)
party.pk = None
party.save()

Except

在关系型数据库中，存在相关联的表是很常见的事。Django中有三种关系型字段：ForeignKey/ManyToManyField/OneToOneField，这些就是需要例外处理的情况了。

ForeignKey在数据库中是用整数型来存放id，所以只要清空主键，直接保存也不会出错。假设party有一个外键关联到creator，如果要把这个关联的外键也复制一份，可以直接Deepcopy：


creator =  copy.deepcopy(party.creator)
creator.pk = None
creator.save()

OneToOneField一般都是处理用户Profile和其他比较特殊的数据，所以还没有遇到过要复制OneToOneField的情况，纯测试了一下，前面提供的方法就可以解决，这里就不赘述了。

ManyToManyField比较特殊，关联的一般是多条记录，需要做一些处理：


creator = party.creator.all()
party.pk = None
party.save()
party.creator = creator