mysql批量准备测试数据3种方式-python脚本(pymysql)

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

因目前在测试数据相关的内容，涉及到使用mysql作为数据源，同时又没有相关的业务数据可以落库，故需要自己手动造一些测试数据来满足当前的测试场景，本篇接上一篇mysql 数据的利用存储过程仅从数据创建测试数据。

---

提示：以下是本篇文章正文内容，下面案例可供参考

一、使用场景

该篇章主要介绍使用python脚本中的pymysql + faker 包进行测试数据准备，该方式适合大部分的场景的测试数据创造，简单易用，测试数据也比较丰富。

二、使用相关包简介

faker ：Faker是一个Python包，开源的GITHUB项目，主要用来创建伪数据，使用Faker包，无需再手动生成或者手写随机数来生成数据，只需要调用Faker提供的方法，即可完成数据的生成。
pymysql：python链接mysql的模块，主要还是聚焦于如何和数据库进行链接和进行基本的操作，主要的体现形式还是主要进行 SQL语句的执行，更多的Faker 可以提供的伪数据方式，可以参考一下链接，写的很详细
https://www.jb51.net/article/243120.htm#_lab2_2_0

二、使用步骤

1. 工具包的下载

使用python pip 工具下载pymysql ，和 faker包

# 下载第三方包
pip install pymysql
pip install faker
# 查看是否下载成功
pip list

如存在一下提示，即使下载成功

2. 利用pymysql 连接数据库

代码如下（示例）：

# coding=utf-8
import pymysql
import kafka
class QdasSql(object):
    # Mysqldb 提供了connect 方法来与数据库的连接，
    def __init__(self, host='10.0.0.0', port=36000, dbName='auto', user='username01', password='word001'):
        self.db = pymysql.connect(host=host, port=port,user=user, passwd=password, db=dbName, charset='utf8')
        # 创建一个游标对象，通过游标对象来进行数据的增删改查。
        self.cursor = self.db.cursor()


    # 调用该对象的close() 方法来关闭数据库。
    def __del__(self):
        self.db.close()
    # 对数据进行相关增删改查，建议做查询时，使用该方法
    def execute(self,sql):
        self.cursor.execute(sql)


    # 获取全部查询结果
    def fetchall(self):
        return self.cursor.fetchall()

    # 获取多行查询结果
    def fetchmany(self,size):
        return self.cursor.fetchmany(size)

    # 获取查询结果一行数据
    def fetchone(self):
        return self.cursor.fetchone()

    # 对于需要提交commit操作使用该方法：eg：插入，修改，删除
    def execute_commit(self,sql):
        try:
            # 执行sql语句
            self.cursor.execute(sql)
            # 提交到数据库执行
            self.db.commit()
        except:
            # 发生错误时回滚
            self.db.rollback()

    def commit(self):
        self.db.commit()

3. 利用faker 创建伪数据，并insert数据

# coding=utf-8
from mysql_dbconn import QdasSql
import faker
import time
def aa():
    for i in range(1000, 2000):
        # 使用faker 函数创造测试数据
        fake = faker.Faker("zh_CN")
        user_name = fake.name()
        click_url = fake.url()
        timestring = int(round(time.time() * 1000))
        # 构造insert into 语句，使用到了format 占位符
        sql = "insert into auto.bath_table07 values ('{}','{}','{}')".format(user_name, click_url, i)
        print("当前时间戳===", timestring)
        sq = QdasSql()
        # 调用封装类中的数据写入方法。
        sq.execute_commit(sql)

if __name__ =="__main__":
    aa()

3. 数据写入验证

---

总结

以上就是今天要分享的内容了，把对数据库的操作封装成一个类，可以更方便与我们后面的使用，同时该方法数据写入的速度还比较可观，1000 条数据写入数据大概需要2分钟，希望内容对大家有所帮助。