python多线程写入数据库

import sqlite3
import threading
import time

# 定义数据库名称和表名称
db_name = 'test.db'
table_name = 'data'

# 定义数据列表
data_list = [(i,) for i in range(100)]

# 定义线程数量
num_threads = 4

# 创建数据库连接和游标对象
conn = sqlite3.connect(db_name)
cursor = conn.cursor()

# 创建数据表
cursor.execute('CREATE TABLE IF NOT EXISTS {} (id INTEGER PRIMARY KEY)'.format(table_name))

# 定义线程函数
def write_thread(data):
    cursor.executemany('INSERT INTO {} (id) VALUES (?)'.format(table_name), data)
    conn.commit()

# 创建线程列表并启动
threads = []
for i in range(num_threads):
    start = int(len(data_list) / num_threads * i)
    end = int(len(data_list) / num_threads * (i+1))
    thread = threading.Thread(target=write_thread, args=(data_list[start:end],))
    thread.start()
    threads.append(thread)

# 等待线程执行完成
for thread in threads:
    thread.join()

# 关闭数据库连接
conn.close()

print('写入完成')