Python向数据库中批量插入数据

测试过程中可能会需要造大量的测试数据，例如用户数据，订单数据等等，此时可以使用python批量插入数据，提高测试效率。

# 导包
import pymysql

# 建立连接
conn = pymysql.Connect(host='192.168.*.*', #数据库地址
                       port=3306,
                       user='root', #用户名
                       password='******', #密码
                       database='数据库名', #数据库名
                       charset='utf8')

# 获取游标
cursor = conn.cursor()

# 执行sql语句

sql1 = "INSERT INTO `库名`.`user` (`id`, `name`, `字段名3`, `字段名4`,) VALUES ('{}', '{}', '{}',6)" #直接赋值可直接填入字符
sql2 = "INSERT INTO `库名`.`user_wallet` (`uid`, `assets_name`, `balance`) VALUES ('{}', 'dollar', 0)"

user_start = 1
for i in range(100000): #遍历十万次，即添加十万条数据
    id = user_start + i
    name= "2b7d" + str(id)  # 转换成字符串
    字段名3= 111000 + i
    uid = id

    print("插入第{}条数据，用户id为{}".format(i + 1, id))

    cursor.execute(sql1.format(id, name, 字段名3)) #已直接赋值的字段不必再填写
    cursor.execute(sql2.format(uid))

# 连接提交
conn.commit()

# 关闭游标
cursor.close()

# 关闭连接
conn.close()