Python连接MySQL数据库,批量生成随机测试数据。

2022/2/22 19:23:43

本文主要是介绍Python连接MySQL数据库,批量生成随机测试数据。,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

Python连接MySQL数据库,批量生成测试数据。
效果如图所示:在这里插入图片描述

代码如下:

import os
import random
import time

import pymysql as ps
db=ps.connect(host='192.168.1.1',user='root',password='123456',database='test',charset='utf8') #连接数据库
cur=db.cursor()	#游标用来执行SQL语句
try:
    #intable=input('输入要创建的表:')  #手动输入表名
    #cur.execute('drop table {}'.format(intable))
    intable='user1'  #默认定义user1表
    cur.execute('drop table {}'.format(intable))
    create_table=cur.execute('create table if not exists {0}(id int primary key, name varchar(100),age int,salary int,dt datetime)'.format(intable))
    db.commit()
    print(intable,'表已创建')
except:
    print('创建',intable,'表失败')
#ID不重复,循环100次
for id in range(0,100):
    print(id)
    #生成随机姓名
    str=[chr(i) for i in range(97,123)]	#97-123为小写字母,65-91为大写
    for j in str:
        name=random.choice(str)+random.choice(str)+random.choice(str)+random.choice(str)  #随机组成4个字母名字

    #生成年龄
    age=random.randrange(20,40,1)  #年龄20-40,步长为1
    #生成工资
    salary=random.randrange(10000,20000,100)
    #生成日期
    date1=(2021,1,1,0,0,0,0,0,0)   #年月日分时秒
    date2=(2022,3,1,0,0,0,0,0,0)
    start=time.mktime(date1)
    end=time.mktime(date2)
    #生成日期
    tm=random.randrange(start,end)   #开始与结束时间
    date_tuple=time.localtime(tm)   #生成元组形式
    dt=time.strftime('%Y-%m-%d',date_tuple) #转换字符串形式

    #插入数据
    data=cur.execute("insert into user1 values({0},'{1}',{2},{3},'{4}')".format(id,name,age,salary,dt))
    data_all=cur.execute('select * from user1')
    db.commit()  #提交SQL变更
    list=cur.fetchall()  #提取所有数据
    print(list)



这篇关于Python连接MySQL数据库,批量生成随机测试数据。的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程