现状

在实际测试中发现，需要插入大量的测试数据或者有依赖关系的数据 来达到测试的目的，这时我们可以使用python来简化和规范化该操作。

采用 peewee 来将SQL代码转化为Python代码实现ORM

peewee 是一个轻量级的 python ORM 库。内建对 SQLite、 MySQL 和 PostgreSQL 的支持

peewee文档地址：

http://docs.peewee-orm.com/en/latest/

中文版文档：

https://www.osgeo.cn/peewee/

生成数据模型

在安装peewee后可以在命令行中输入指令，快速的将已有的 数据库 转化成数据模型

python -m pwiz -e 数据库类型 -p 端口 -u 账号 -H IP地址 --password 密码 -o -i 库名 > 生成的py文件名.py

例如

python -m pwiz -e mysql -p 3306 -u doctor_core -H 数据库ip --password 数据库密码 -o -i 数据库账号 > 导出的文件.py

查看doctor_info表

-- auto-generated definition
create table doctor_info
    id                      bigint(22) unsigned auto_increment comment '主键'
        primary key,
    user_id                 bigint unsigned                not null comment '用户编号',
    login_id                varchar(128)                   null comment '登录名',
    mobile_no               varchar(11)                    null comment '手机号码',
    telephone_no            varchar(50)                    null comment '医生固话',
    title_name              varchar(200)                   null comment '行政职称名',
    dept_uuid               varchar(40)                    null comment '科室uuid',
    dept_name               varchar(100)                   null comment '科室名称',
    hospital_uuid           varchar(40)                    null comment '医院uuid',
    hospital_name           varchar(100)                   null comment '医院名称',
    property                tinyint(1) unsigned            null comment '医院性质',
    expert_uuid             varchar(45)                    null comment '专家uuid',
    expert_name             varchar(40)                    null comment '专家姓名',
    favorite_dept           varchar(1000)                  null comment '医生擅长科室，多个以英文逗号分隔',
    job_number              varchar(40)                    null comment '医院端医生工号',
    canal_online            tinyint(1) unsigned default 1  null comment '渠道：1为线上，0为线下，2为兼职，3为赠险注册',
    source                  bigint(11) unsigned            null comment '应用来源',
    reg_type                tinyint(2) unsigned            not null comment '注册来源: ',
    state                   tinyint(4) unsigned default 0  not null comment '状态，0：未认证，1：已认证',
    is_close                tinyint(1) unsigned default 1  not null comment ' 0,关闭，1，开通',
    is_deleted              tinyint unsigned    default 0  not null comment '是否删除',
    gmt_created             datetime                       not null comment '创建时间',
    gmt_modified            datetime                       not null comment '修改时间',
    project_source_id       int                            null comment '对应项目来源表的外键',
    source_group_id         bigint                         null comment '来源分组',
    source_group_name       varchar(200)                   null comment '来源分组名称',
    role_id                 varchar(100)                   null comment '医生关联角色',
    grade                   tinyint(3)          default 0  null comment '医生资质等级',
    is_special              tinyint(2)          default 0  null comment '是否特批医生',
    relation_state          tinyint(4) unsigned            null comment '关联状态',
    relation_remark         varchar(256)                   null comment '关联备注',
    gmt_relation            datetime                       null comment '关联时间',
    is_auto_relation        tinyint(1) unsigned            null comment '是否自动关联',
    supervise_record_status tinyint(2) unsigned default 0  not null comment '监管备案状态',
    supervise_record_remark varchar(255)        default '' not null comment '监管备案失败原因',
    constraint uniq_user_id
        unique (user_id)
    comment '医生信息表' charset = utf8;
create index idx_expert_name
    on doctor_info (expert_name);
create index idx_expert_uuid
    on doctor_info (expert_uuid);
create index idx_gmtCreated
    on doctor_info (gmt_created);
create index idx_gmtModified
    on doctor_info (gmt_modified);
create index idx_hospital_uuid_job_number
    on doctor_info (hospital_uuid, job_number);
create index idx_job_number
    on doctor_info (job_number);
create index idx_loginId
    on doctor_info (login_id);
create index idx_mobile_no
    on doctor_info (mobile_no);

查看它转换后的代码为:

class DoctorInfo(BaseModel):
    id = BigAutoField()
    user_id = BigIntegerField(unique=True)
    login_id = CharField(index=True, null=True)
    mobile_no = CharField(index=True, null=True)
    telephone_no = CharField(null=True)
    title_name = CharField(null=True)
    dept_uuid = CharField(null=True)
    dept_name = CharField(null=True)
    hospital_uuid = CharField(null=True)
    hospital_name = CharField(null=True)
    property = IntegerField(null=True)
    expert_uuid = CharField(index=True, null=True)
    expert_name = CharField(index=True, null=True)
    favorite_dept = CharField(null=True)
    job_number = CharField(index=True, null=True)
    canal_online = IntegerField(constraints=[SQL("DEFAULT 1")], null=True)
    source = BigIntegerField(null=True)
    reg_type = IntegerField()
    state = IntegerField(constraints=[SQL("DEFAULT 0")])
    is_close = IntegerField(constraints=[SQL("DEFAULT 1")])
    is_deleted = IntegerField(constraints=[SQL("DEFAULT 0")])
    gmt_created = DateTimeField(index=True)
    gmt_modified = DateTimeField(index=True)
    project_source_id = IntegerField(null=True)
    source_group_id = BigIntegerField(null=True)
    source_group_name = CharField(null=True)
    role_id = CharField(null=True)
    grade = IntegerField(constraints=[SQL("DEFAULT 0")], null=True)
    is_special = IntegerField(constraints=[SQL("DEFAULT 0")], null=True)
    relation_state = IntegerField(null=True)
    relation_remark = CharField(null=True)
    gmt_relation = DateTimeField(null=True)
    is_auto_relation = IntegerField(null=True)
    supervise_record_status = IntegerField(constraints=[SQL("DEFAULT 0")])
    supervise_record_remark = CharField(constraints=[SQL("DEFAULT ''")])
    class Meta:
        table_name = 'doctor_info'
        indexes = (
            (('hospital_uuid', 'job_number'), False),

我们使用这段代码来完成一次CRUD操作

增

DoctorInfo(user_id=xx,login_id=xx,....).save()

删

DoctorInfo.delete().where(DoctorInfo.expert_name == '张三').execute()

改

DoctorInfo.expert_name = '李四'