【python】批量向数据库插入数据

前言

python脚本实现批量向数据库插入特定不重复的数据，如需使用可根据自身需要修改代码

---

一、不多废话，直接上代码

import mysql.connector
import random


# 向指定表中插入不重复sn的数据

def insert_data():
    connection = mysql.connector.connect(
        host='',  # MySQL数据库主机名
        user='',  # MySQL数据库用户名
        password='',  # MySQL数据库密码
        database=''  # MySQL数据库名称
    )

    cursor = connection.cursor()  # 创建游标对象，用于执行SQL语句

    prefix = 'YA7CAT237PCB01C'  # 定义序列号前缀
    start_number = 1  # 定义起始序号
    for i in range(2000):  # 循环插入条数
        sn = f'{prefix}{str(start_number + i).zfill(4)}'  # 生成带有前缀和补零的SN
        product_type = random.choice([1, 2, 3])  # 从列表[1, 2, 3]中随机选择business_type值
        query = """
        INSERT INTO best_after_sale.sa_gyy_order
        (platform_code, express_code, express_dmmc, addr, customer_name, customer_phone, order_time, create_time, create_type, product_type, is_sxs, sn, business_tag)
        VALUES
        (NULL, '暂无', '暂无', NULL, NULL, NULL, NULL, NOW(), 1, %s, 0, %s, 3)
        """  # SQL插入语句，使用占位符%s表示待插入的值
        values = (product_type, sn)  # 定义待插入的值，将序列号和business_tag赋值给values变量
        cursor.execute(query, values)  # 执行SQL插入语句，将values中的值插入到对应的占位符中

    connection.commit()  # 提交事务，将数据保存到数据库

    cursor.close()  # 关闭游标对象
    connection.close()  # 关闭数据库连接


if __name__ == '__main__':
    insert_data()

相关链接

更多相关知识请前往：Leo‘s Blog