python从基础到进阶（七）--操作数据库

操作数据库

使用Python操作数据库需要先安装适用于相应数据库的数据库驱动程序。下面是安装数据库驱动程序的详细步骤：

1. 确定数据库类型

确定你要连接的数据库类型，例如SQLite、MySQL、PostgreSQL等。

2. 安装数据库驱动程序

SQLite:

SQLite是一个嵌入式数据库，Python自带了对SQLite的支持，因此不需要额外安装驱动程序。

MySQL:

使用mysql-connector-python或pymysql模块来连接MySQL数据库。你可以使用以下命令之一来安装这些模块：

pip install mysql-connector-python

或

pip install pymysql

PostgreSQL:

使用psycopg2模块来连接PostgreSQL数据库。你可以使用以下命令来安装：

pip install psycopg2

3. 安装其他依赖项（如果需要）

根据你所选的数据库驱动程序，可能需要安装其他依赖项。例如，如果你使用MySQL，可能需要安装MySQL客户端库。具体的依赖项请参考相应的文档。

4. 验证安装及导入相应的数据库驱动程序模块

安装完成后，你可以在Python中导入相应的模块来验证安装是否成功。例如：

import sqlite3  # 如果使用SQLite
import mysql.connector  # 如果使用MySQL
import psycopg2  # 如果使用PostgreSQL

如果导入没有出现错误，则说明安装成功。

5. 建立与数据库的连接

使用数据库驱动程序提供的函数或方法来建立与数据库的连接。在连接时，你需要提供数据库的主机名、用户名、密码以及数据库名称等信息。

SQLite:

python复制代码
# 连接到SQLite数据库，如果不存在，则会创建一个新的数据库
conn = sqlite3.connect('example.db')

MySQL:

# 连接到MySQL数据库
conn = mysql.connector.connect(
    host='hostname',
    user='username',
    password='password',
    database='database_name'
)

PostgreSQL:

# 连接到PostgreSQL数据库
conn = psycopg2.connect(
    host='hostname',
    user='username',
    password='password',
    database='database_name'
)

6. 创建游标对象

一旦建立了连接，你需要创建一个游标对象。游标对象用于执行SQL语句并处理结果。

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

7. 执行SQL语句

使用游标对象执行SQL语句。你可以执行各种SQL操作，如创建表、插入数据、更新数据、删除数据或查询数据等。

# 创建表
cursor.execute('''CREATE TABLE IF NOT EXISTS users 
                  (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)''')

# 插入数据
cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Alice', 30))

# 查询数据
cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()
for row in rows:
    print(row)

8. 提交事务（对于需要的操作）

在执行了一系列操作后，如果你需要将这些操作永久保存到数据库中，你需要提交事务。

# 提交事务
conn.commit()

9. 关闭游标和连接

在完成所有数据库操作后，记得关闭游标和连接，以释放资源。

# 关闭游标
cursor.close()

# 关闭连接
conn.close()

常用的游标操作方法

1. execute()

execute() 方法用于执行SQL语句。可以执行各种SQL操作，如创建表、插入数据、更新数据、删除数据或查询数据等。

cursor.execute("SELECT * FROM users")

2. fetchall()

fetchall() 方法用于获取所有的查询结果。执行查询后，调用 fetchall() 方法将返回一个包含所有查询结果的列表。

rows = cursor.fetchall()

3. fetchone()

fetchone() 方法用于获取一条查询结果。执行查询后，调用 fetchone() 方法将返回查询结果的下一行，如果没有更多的行，则返回 None。

row = cursor.fetchone()

4. fetchmany(size)

fetchmany(size) 方法用于获取指定数量的查询结果。执行查询后，调用 fetchmany(size) 方法将返回一个包含指定数量行的列表。

rows = cursor.fetchmany(5)  # 获取5行查询结果

5. scroll()

scroll() 方法用于在结果集中移动游标的位置。它接受两个参数：offset（偏移量）和 mode（模式）。

• offset: 移动的偏移量，可以是正数（向后移动）或负数（向前移动）。
• mode: 模式，指定从哪个位置开始计算偏移量。默认为 relative（相对位置），也可以指定为 absolute（绝对位置）。

cursor.scroll(2, mode='relative')  # 向后移动2行

6. rowcount

rowcount 属性用于获取最近一次执行的SQL语句所影响的行数。

print(cursor.rowcount)