Python实现生成多条数据并插入数据库

1.创建TXT文件与数据:add_data

代码示例:

import os
from datetime import datetime
##设置文档路径与文档名
file_path='D:\\自动化测试\\apitest_pycharm\\data\\user.txt'
def build_data(file_path,data_count):
    if os.path.exists(file_path):
        os.remove(file_path)
    file=open(file_path,'w')#打开文档并写入数据
    start=datetime.now()#数据开始生成时间
    print('数据生成中...')
    for var in range(data_count):
        #设置数据内容
        file.writelines(str(var)+" 用户"+str(var)+" email"+str(var)+"@email.com"+" 150"+str(var))
        file.write('\n')
    end=datetime.now()#数据生成结束时间
    print('耗时:'+str((end-start).seconds)+"s")#打印数据生成耗时
build_data(file_path,50)#设置生成数据数量

生成的数据,如下:
在这里插入图片描述

2.封装数据库相关操

Python中可以使用批量插入的方式一次性插入多条语句到数据库。 首先,需要安装Python数据库操作模块,比如MySQLdb、psycopg2、pymysql等,以便能够连接数据库并操作数据。 其次,可以使用循环的方式生成多个插入语句的参数列表。例如,创建一个包含多个元组的列表,每个元组代表一条语句的参数。 然后,可以使用数据库操作模块提供的executemany()方法执行批量插入。这个方法接收两个参数,第一个参数是插入语句,第二个参数是参数列表。 最后,需要进行事务的提交。可以使用数据库连接对象提供的commit()方法。 下面是一个简单的示例,演示了如何使用Python进行批量插入多条语句到MySQL数据库: ```python import MySQLdb # 连接数据库 conn = MySQLdb.connect(host='localhost', user='root', passwd='password', db='test') cursor = conn.cursor() # 创建插入语句 insert_sql = "INSERT INTO table_name (column1, column2) VALUES (%s, %s)" # 创建参数列表 values = [ ('value1', 'value2'), ('value3', 'value4'), ('value5', 'value6') ] try: # 执行批量插入 cursor.executemany(insert_sql, values) # 提交事务 conn.commit() print("插入成功") except Exception as e: # 发生错误时回滚 conn.rollback() print(str(e)) finally: # 关闭数据库连接 cursor.close() conn.close() ``` 以上示例中,首先连接到数据库,然后创建插入语句,然后创建参数列表。接下来执行批量插入,如果成功则提交事务,失败则回滚。最后关闭数据库连接。 这样就可以一次性插入多条语句到数据库。注意要根据实际情况修改表名、列名、参数等。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值