python faker批量生成数据写入MySQL_python faker 生成sql

import pymysql
from faker import Faker
import random
 
conn = pymysql.connect(
    host='主机名或IP地址',
    user='用户名',
    password='密码',
    database='数据库名',
    port=3306 # 端口
)
 
# 创建游标对象
cur = conn.cursor()
 
# 创建faker对象，用于生成数据
faker = Faker('zh_CN')
 
# 定义要插入的表名
table_name = 'table_name'
 
def generate_and_insert_data(table_name,conn,cur,num_rows):
 
    job_id1 = ["job001","job002","job003"]
    # 生成列名
    columns = '(user_id,user_name,user_age,job_id,job_age,job_enable,city,work_date)'
    # 生成值
    values = [(faker.pystr(min_chars=8,max_chars=8).upper()+faker.numerify(),
               faker.name(),
               random.randint(23,55),
               random.choice(job_id1),
               faker.random_int(min=1,max=10),
               random.choice(['true','false']),
               faker.city(),
               faker.date_between(start_date='-10y',end_date='-1m')) for i in range(1,num_rows+1)]
    # 生成批量插入的SQL语句
    insert_query = 'INSERT INTO {} {} VALUES %s'.format(table_name,columns)
    # 创建批量插入的数据元组列表
    execute_values = [(v,) for v in values]
    # 执行批量插入
    cur.executemany(insert_query,execute_values)
    # 提交事务
    conn.commit()
 
# 调用函数，生成并插入数据
num = 10
generate_and_insert_data(table_name,conn,cur,num)
 
# 关闭游标和连接
cur.close()
conn.close()
 
 
print(f'{num}条数据已插入数据库中...')