基于Python的DbFaker数据生成器源码

dbfaker Python Python, Shell 共38个文件 py: 22 md: 5 yml: 2 gitignore: 1 LICENSE: 1 bin/dbfaker: 1 bin/table2yml: 1 db: 1 sql: 1 txt: 1 基于数据库层面批量生成有逻辑关联的数据 dbfaker 介绍 基于数据库层面批量生成有逻辑关联的数据 软件架构 对于要造大量数据来讲，有几种方式，一是通过开发写的接口来模拟真实用户场景来产生数据，这种方式在某些方面来讲是比较合适的，比如因为是模拟用户场景，业务数据完整，不会丢．但不好的地方就是要处理的地方太多，接口定义，签名，加密等等，还有扩展性也很不好，一个业务场景要写一套代码．另一种方式，是已知业务产生的数据之间的依赖关系后，直接在数据库中插入相关数据，本项目就是通过这种方式来实现，好处就是生成规则通过配置文件来描述即可（yaml文件），不需要额外添加代码（对于某些字段生成规则有可能需要单独编写方法），与测试库testrunner比较类似．现在已知有些库也支持直接在数据库内造数据，但对库表之间的关联关系的处理都做得不太好． 另外对于测试人员进行自动化接口测试时，前置数据是个问题，是通过业务接口一条条跑完作为前置数据产生条件还是直接在数据库里面插入需要的数据呢？前面一种方式在业务场景复杂的时候用例的维护可能会变得极其麻烦。比如： 你要测试一个用户加入商品到购物车的一个接口；前置数据有哪些？ 用户账号 商品 通过接口来造数据的流程如下： 通过用户端注册接口来创建账号 通过管理端接口登录，拿到token（预设管理端账号是已经存在了，如果不存在你还得注册管理端账号） 登陆后通过管理端接口创建商品信息 这三个步骤看起来简单，但实际上针对某些业务场景还有接口加密、签名、而且几乎每一条测试用例都需要用到N个前置条件，这些通过接口来调用的前置条件只要有一个失败就会影响你真正要测试的接口。做过接口测试的同学应该深有体会； 实际上实现上述接口测试，真正要测的只是商品是否能正常加入到购物车；真正依赖的就只有两个数据，用户账号和商品信息。我们只需要在用户表和商品表里面各插入一条数据就可以了。通过本工具，只需要描述两个表的每个字段的生成规则就可以直接在数据库中插入两条“十分真实”的数据。用这种方式来解决测试用例前置条件的问题不说是最好的方式，但肯定是比上面的前置用例的方式好很多。 本项目数据处理流程如下： 安装 git clone https://gitee.com/guojongg/dbfaker.git cd dbfaker # 可先创建虚拟环境后再安装 python3 setup.py install # 使用pip直接安装 pip3 install dbfaker # 卸载 pip uninstall dbfaker 使用说明 １．　生成meta.yml文件，可通过内置工具table2yml.py来一键生成模板，生成后需完善模板中表字段的定义 table2yml.py文件使用说明： usage: table2yml [-h] [--connect [CONNECT]] [--table_names [TABLE_NAMES]] [--sql_file [SQL_FILE]] [--output [OUTPUT]] [--hide_command] [type] 数据库表转数据生成yaml文件格式工具 positional arguments: type 数据来源，table_name： 通过输入表名与数据库链接方式，在数据库中获取数据库建表语句； table_statement: 指定数据库建表语句的sql文件路径 optional arguments: -h, --help show this help message and exit --connect [CONNECT] 数据库连接语法，例如：mysql+mysqldb://user:password@host/dbname --table_names [TABLE_NAMES] 数据库表，多个表以“,”分割 --sql_file [SQL_FILE] 数据库建表语句的sql文件路径 --output [OUTPUT] 输出文件名，默认为数据库表名+meta.yml --hide_comment 不转换comment字段（可减少yml文件行数） 使用举例 1、在可以连接数据库的情况下直接指定数据库连接和要转换的表名来输出yaml文件 table2yml table_name --table_names stu,course,choice_course --connect mysql+mysqldb://[dbuser]:[dbpassword]@[dbhost]/[dbname] --output test.yml 2、在不能直接连接数据库，但有建表语句sql文件时可以通过下面的方法来输出yaml文件 table2yml table_statement --sql_file test.sql --output test.yml 2, 编辑meta.yml文件，文件格式如下 package: - datetime #　导入额外的包，在jinja2模板中使用（下面有用到datetime包，所以要先导入） env: id: engine: faker.uuid time_format: engine: faker.eq("%Y-%m-%d %H:%M:%S") tables: - table: stu comment: '学生表' columns: id: comment: 数据主键id engine: eq('{{ env.id }}') # 通过引用环境变量中的值 name: comment: 姓名 engine: name idcard: comment: 身份证号 engine: ssn age: comment: 年龄 engine: eq('{{ datetime.datetime.now().year - int(stu.idcard[6:10]) }}') #　通过jinja２模板直接计算 sex: comment: 性别 engine: faker.eq('{{ "man" if int(stu.idcard[-2]) % 2==1 else "female" }}') #　通过jinja２模板直接计算 - table: course comment: '课程表' columns: id: comment: 数据主键id engine: faker.uuid stu_id: comment: 数据主键id engine: faker.eq('{{ stu.id }}') # 引用其他表中的值 course_name: comment: 课程名称 engine: faker.choice(['数学','语文','英语','化学','地理']) # 通过内置方法从列表中随机取一个值 course_time: comment: 上课时间 engine: faker.now(format="{{ env.time_format }}") # 通过内置方法获取当前时间，并按照指定格式返回 extraction: stu_name: value: '{{ stu.name }}' course_name: value: '{{ course.course_name }}' ３，创建ｓｑｌ PC:~/01 Work/07 MyProject/dbfaker$ source venv/bin/activate (venv) PC:~/01 Work/07 MyProject/dbfaker$ dbfaker -h usage: dbfaker [-h] [-n [NUMBER]] [-i] [-c [CONNECT]] [-o [OUTPUT]] [-p] [--project_name PROJECT_NAME] [meta_file] 通过ｙｍｌ格式的描述文件来生成数据 positional arguments: meta_file yml文件所在路径 optional arguments: -h, --help show this help message and exit -n [NUMBER], --number [NUMBER] 生成数据数量 -i, --insert 是否插入到数据库 -c [CONNECT], --connect [CONNECT] 数据库连接语法，例如：mysql+mysqldb://pdmsadmin:system001@cpcs.ho melabs.in/pdms_hospital -o [OUTPUT], --output [OUTPUT] 指定文件名，输出内容到文件 -p, --_print 是否打印到控制台 --project_name PROJECT_NAME 初始化项目时的项目名称 # 打印输出 dbfaker data/test.yml --number 10 -p # 保存到文件 dbfaker data/test.yml --number 10 -o out.sql # 插入到数据库 dbfaker data/test.yml --number 10 -i --connect mysql+mysqldb://pdmsadmin:system001@cpcs.homelabs.in/pdms_hospital 通过上述模板文件生成出sql: (venv) guolong@guolong-PC:~/01Work/07MyProject/dbfaker$ dbfaker data/test.yml -p -n 10 100%|██████████████████████████████████████████████████████████████████████████