使用本地mysql做简单测试时需要创建的示例数据,于是用python编写了mysql的连接与批量新增数据的过程,其中测试表的设计如下:
源码
import random
import pymysql
# 连接数据库
try:
conn = pymysql.connect(host="localhost", user="root", password="1234", database="test")
except RuntimeError as e:
print('connect failed', e)
exit(0)
print('connect success')
cursor = conn.cursor()
# 创建示例数据
datas = []
for idx in range(97, 123):
name = chr(idx) * 2
age = random.randint(20, 30)
sex = ['male', 'female'][random.randint(0, 1)]
datas.append((name, age, sex))
# 插入示例数据
sql = 'insert into t_test (name, age, sex) values (%s, %s, %s);'
ret = cursor.executemany(sql, datas)
conn.commit()
# 查询数据表数据
cursor.execute("select * from t_test;")
print(ret, cursor.fetchall())
# 关闭
cursor.close()
conn.close()