Python编程：利用peewee的model_to_dict进行数据迁移

"""
需求：两个结构相同的表A, B，需要从 A表 将数据迁移到 B表
"""

from peewee import *
from playhouse.shortcuts import model_to_dict
from conf import BaseModel  # 配置好的数据库信息
class AModel(BaseModel): name = CharField(default="") age = IntegerField(default=0) class Meta: db_table = 'a_table'
class BModel(BaseModel): name = CharField(default="") age = IntegerField(default=0) class Meta: db_table = 'b_table'
if __name__ == '__main__': for page in range(1, 50):  # 页数，要比实际页数多一页，以免数据漏掉 rows = AModel.select().paginate(page=page, paginate_by=1000) # 随机取出一定数量的数据 lst = [] for row in rows: dct = model_to_dict(row) dct.pop("id")  # 删除字典中的id字段，这个默认会自增 # 通过 md5 字段判断，避免重复插入 ret = BModel.filter(BModel.md5 == dct["md5"]).first() if ret is None: lst.append(dct) if len(lst) > 0: BModel.insert_many(lst).execute() print "数据拷贝完成", "取出数据:%d" % len(rows), "成功插入%d" % len(lst) print "数据迁移完成"
  
 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
  
14
  
15
  
16
  
17
  
18
  
19
  
20
  
21
  
22
  
23
  
24
  
25
  
26
  
27
  
28
  
29
  
30
  
31
  
32
  
33
  
34
  
35
  
36
  
37
  
38
  
39
  
40
  
41
  
42
  
43
  
44
  
45
 

关于model_to_dict的用法，参考：
Python编程：playhouse模块转peewee的model对象为字典dict

文章来源: pengshiyu.blog.csdn.net，作者：彭世瑜，版权归原作者所有，如需转载，请联系作者。

原文链接：pengshiyu.blog.csdn.net/article/details/81077382