pymysql 执行多条语句

原创

mob649e81624618 2023-11-28 14:03:51 ©著作权

文章标签 SQL MySQL python 文章分类 MySQL 数据库

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

PyMySQL执行多条语句

PyMySQL是一个用于Python编程语言的纯Python MySQL客户端库，它实现了MySQL数据库连接协议。它可以被用来连接MySQL服务器，并与其进行交互。在实际应用中，我们经常需要执行多条SQL语句来完成一系列的数据库操作。本文将介绍如何使用PyMySQL执行多条SQL语句。

安装PyMySQL

在开始之前，我们需要先安装PyMySQL库。可以使用pip命令进行安装：

pip install PyMySQL

连接数据库

在执行多条SQL语句之前，我们需要先连接到MySQL数据库。首先，我们需要导入PyMySQL库：

import pymysql

然后，使用connect()函数连接到数据库：

conn = pymysql.connect(host='localhost', user='root', password='password', database='test')

在这个例子中，我们连接到本地的MySQL服务器，用户名为root，密码为password，并且使用名为test的数据库。你需要根据自己的实际情况修改这些参数。

在连接成功后，我们可以通过cursor()方法创建一个游标对象，用于执行SQL语句：

cursor = conn.cursor()

执行多条SQL语句

PyMySQL的execute()方法可以执行一条SQL语句，但是它不能执行多条SQL语句。如果我们要执行多条SQL语句，可以使用executescript()方法。

sql = '''
CREATE TABLE users (
    id INT PRIMARY KEY AUTO_INCREMENT,
    name VARCHAR(255),
    age INT
);

INSERT INTO users (name, age) VALUES ('Alice', 25);
INSERT INTO users (name, age) VALUES ('Bob', 30);
'''

cursor.executescript(sql)

在上面的例子中，我们首先创建了一个名为users的表，并定义了三个字段：id、name和age。然后，我们插入了两条记录。

需要注意的是，executescript()方法执行的是一组SQL语句，它会将这些语句作为一个整体发送给服务器执行。这意味着如果其中一条SQL语句出现错误，整个批量操作将会失败，所有的改动都将被回滚。

提交事务

在执行多条SQL语句之后，我们需要提交事务以使改动生效。可以使用commit()方法提交事务：

conn.commit()

如果不执行commit()方法，那么所有的改动将不会被保存到数据库中。

关闭连接

在完成数据库操作后，我们需要关闭数据库连接，释放资源。可以使用close()方法关闭连接：

conn.close()

完整示例

以下是一个完整的示例，演示了如何使用PyMySQL执行多条SQL语句：

import pymysql

# 连接到数据库
conn = pymysql.connect(host='localhost', user='root', password='password', database='test')

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

# 执行多条SQL语句
sql = '''
CREATE TABLE users (
    id INT PRIMARY KEY AUTO_INCREMENT,
    name VARCHAR(255),
    age INT
);

INSERT INTO users (name, age) VALUES ('Alice', 25);
INSERT INTO users (name, age) VALUES ('Bob', 30);
'''

cursor.executescript(sql)

# 提交事务
conn.commit()

# 关闭连接
conn.close()