Python中的游标用法介绍|极客教程

Python中的游标用法介绍

1. 概述

在开发使用数据库的应用程序时，我们经常需要使用游标（cursor）来执行SQL语句，并操作数据库的数据。游标相当于是一个指向数据库中某一行数据的指针，通过游标我们可以对数据库进行查询、插入、更新和删除等操作。

本文将介绍Python中使用游标的一些常见操作和用法。我们将学习如何连接数据库、创建游标、执行SQL语句、获取查询结果以及处理事务等。

2. 连接数据库

在使用游标之前，我们首先需要连接到数据库。Python提供了多个第三方库来连接各种数据库，包括MySQL、SQLite、PostgreSQL等。这里我们以MySQL数据库为例。

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

pip install pymysql

接下来，我们可以使用以下代码来连接到MySQL数据库：

import pymysql

# 连接数据库，设置连接参数
conn = pymysql.connect(
    host='localhost',
    port=3306,
    user='root',
    password='password',
    db='test_db'
)

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

# 执行SQL语句
cursor.execute('SELECT * FROM users')

# 获取查询结果
result = cursor.fetchall()

# 打印查询结果
for row in result:
    print(row)

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

通过以上代码，我们可以连接到MySQL数据库，并执行一条查询语句，然后获取查询结果并打印出来。注意，需要根据实际情况修改连接参数，包括主机名、端口号、用户名、密码和数据库名等。

3. 执行SQL语句

在使用游标进行数据库操作时，我们需要执行SQL语句。Python的游标对象提供了多个方法来执行SQL语句，并返回结果。

3.1 execute方法

execute方法可以执行任意的SQL语句，包括查询、插入、更新和删除等。它接受一个SQL语句作为参数，并返回受影响的行数。

# 执行插入语句
cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 25)")

# 执行更新语句
cursor.execute("UPDATE users SET age = 30 WHERE name = 'Bob'")

# 执行删除语句
cursor.execute("DELETE FROM users WHERE name = 'Bob'")

3.2 executemany方法

executemany方法可以一次执行多条相同结构的SQL语句，比如批量插入多条记录。它接受一个SQL语句和一个参数列表作为参数，并返回受影响的行数。

# 批量插入多条记录
data = [
    ('Alice', 20),
    ('Mike', 22),
    ('Jack', 24)
]
cursor.executemany("INSERT INTO users (name, age) VALUES (%s, %s)", data)

3.3 fetchone方法

fetchone方法可以获取查询结果的下一行数据。如果结果集为空，则返回None。

# 执行查询语句
cursor.execute('SELECT * FROM users')

# 获取下一行数据
row = cursor.fetchone()

# 打印查询结果
while row:
    print(row)
    row = cursor.fetchone()

3.4 fetchall方法

fetchall方法可以获取查询结果的所有行数据。返回一个元组的列表，每个元组表示一行数据。

# 执行查询语句
cursor.execute('SELECT * FROM users')

# 获取所有行数据
result = cursor.fetchall()

# 打印查询结果
for row in result:
    print(row)

4. 处理事务

在开发应用程序时，有时候需要保证多个数据库操作的原子性，即要么全部执行成功，要么全部执行失败。这时我们可以使用事务（transaction）来实现。

在Python中，我们可以使用游标对象的commit方法来提交事务，使用rollback方法来回滚事务。

# 开始事务
conn.begin()

try:
    # 执行多个数据库操作
    cursor.execute("INSERT INTO users (name, age) VALUES ('Bob', 25)")
    cursor.execute("UPDATE users SET age = 30 WHERE name = 'Bob'")

    # 提交事务
    conn.commit()
except:
    # 出现异常，回滚事务
    conn.rollback()