pymysql批量执行sql

原创

mob649e816209c2 2024-06-26 06:14:40 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e816209c2的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用pymysql批量执行SQL语句

在开发中，我们经常需要执行批量的SQL语句，比如批量插入数据、批量更新数据等。而使用pymysql这个Python库，可以很方便地实现批量执行SQL语句的功能。

安装pymysql

首先，我们需要安装pymysql库。可以使用pip进行安装：

pip install pymysql

连接数据库

在使用pymysql之前，我们需要先连接到数据库。以下是一个连接到MySQL数据库的示例代码：

import pymysql

# 连接数据库
conn = pymysql.connect(host='localhost', user='root', password='123456', database='testdb', charset='utf8')

# 创建游标对象
cursor = conn.cursor()

批量执行SQL语句

接下来，我们可以通过使用executemany()方法来批量执行SQL语句。这个方法接受两个参数，第一个参数是SQL语句，第二个参数是一个元组的列表，每个元组对应一条SQL语句的参数。

# 批量插入数据
sql = "INSERT INTO users (name, age) VALUES (%s, %s)"
data = [('Alice', 25), ('Bob', 30), ('Cathy', 28)]

cursor.executemany(sql, data)

提交事务

在执行完所有的SQL语句后，我们需要提交事务才能使数据变更生效。可以使用commit()方法来提交事务：

# 提交事务
conn.commit()

关闭连接

最后，在使用完pymysql后，记得关闭连接，释放资源：

# 关闭游标
cursor.close()

# 关闭连接
conn.close()

完整示例代码

下面是一个完整的示例代码，包括连接数据库、批量执行SQL语句、提交事务和关闭连接：

import pymysql

# 连接数据库
conn = pymysql.connect(host='localhost', user='root', password='123456', database='testdb', charset='utf8')

# 创建游标对象
cursor = conn.cursor()

# 批量插入数据
sql = "INSERT INTO users (name, age) VALUES (%s, %s)"
data = [('Alice', 25), ('Bob', 30), ('Cathy', 28)]

cursor.executemany(sql, data)

# 提交事务
conn.commit()

# 关闭游标
cursor.close()

# 关闭连接
conn.close()