pymysql执行多条语句|极客笔记

pymysql执行多条语句

在开发过程中，有时候我们需要一次性执行多条SQL语句，比如初始化数据库表结构或者批量插入数据等。在Python中使用pymysql库可以很方便地实现这个功能。本文将详细介绍如何使用pymysql执行多条语句。

安装pymysql

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

pip install pymysql

安装完成后，我们可以开始使用pymysql来执行多条SQL语句。

连接数据库

在执行多条SQL语句之前，我们首先需要连接数据库。下面是一个简单的连接数据库的示例代码：

import pymysql

# 连接数据库
connection = pymysql.connect(host='localhost',
                             user='root',
                             password='password',
                             database='test_db',
                             charset='utf8mb4',
                             cursorclass=pymysql.cursors.DictCursor)

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

在这段代码中，我们使用pymysql.connect()方法连接到数据库，并创建了一个游标对象。

执行多条SQL语句

接下来，我们将演示如何使用pymysql执行多条SQL语句。我们可以使用execute()方法执行多条SQL语句，只需将这些SQL语句放在一个字符串中，使用分号分隔开即可。

下面是一个示例代码，演示了如何执行多条SQL语句：

# 定义多条SQL语句
sql = """
CREATE TABLE IF NOT EXISTS users (
    id INT AUTO_INCREMENT PRIMARY KEY,
    name VARCHAR(255) NOT NULL,
    email VARCHAR(255) NOT NULL
);

INSERT INTO users (name, email) VALUES ('Alice', 'alice@example.com');
INSERT INTO users (name, email) VALUES ('Bob', 'bob@example.com');
"""

# 执行多条SQL语句
cursor.execute(sql)

# 提交事务
connection.commit()

# 关闭游标和连接
cursor.close()
connection.close()

在这段代码中，我们定义了一个包含多条SQL语句的字符串变量sql，然后使用execute()方法执行这些SQL语句。最后使用commit()方法提交事务，并关闭游标和连接。