数据库连接前的准备工作
在使用Python连接数据库之前,需要确保已安装相应的数据库驱动程序,连接MySQL数据库需安装mysql-connector-python,连接PostgreSQL需安装psycopg2,连接SQLite则无需额外安装(Python内置支持),还需准备好数据库的连接信息,包括主机名、端口、用户名、密码以及数据库名称,这些信息通常由数据库管理员提供,或存储在配置文件中以确保安全性,建议使用环境变量或配置文件管理敏感信息,避免硬编码在脚本中。

安装必要的Python库
通过包管理器pip安装对应的数据库驱动,以MySQL为例,可在终端运行命令:pip install mysql-connector-python,安装完成后,可通过import语句验证库是否正确安装,例如import mysql.connector,如果遇到版本兼容问题,可尝试指定版本号,如pip install mysql-connector-python==8.0.33,安装过程中若遇到依赖缺失,需根据错误提示安装相应的编译工具或依赖库。
建立数据库连接
连接数据库的核心是调用驱动提供的连接函数,并传入必要的参数,以MySQL为例,代码示例如下:
import mysql.connector
conn = mysql.connector.connect(
host="localhost",
user="your_username",
password="your_password",
database="your_database"
)
成功连接后,conn对象将用于后续的数据库操作,若连接失败,需检查参数是否正确、数据库服务是否运行,以及网络是否可达,部分驱动支持连接池(如mysql.connector.pooling),可提高频繁连接场景下的性能。
执行SQL查询语句
连接建立后,需创建游标对象(cursor)来执行SQL语句,游标提供了执行查询、获取结果等功能。
cursor = conn.cursor()
cursor.execute("SELECT * FROM users WHERE age > 25")
results = cursor.fetchall()
fetchall()方法获取所有查询结果,也可使用fetchone()逐行获取,执行查询后,需关闭游标以释放资源:cursor.close()。

处理查询结果
查询结果通常以元组或列表形式返回,可通过遍历提取数据。
for row in results:
print(f"ID: {row[0]}, Name: {row[1]}")
若结果量较大,建议分批处理或使用生成器表达式减少内存占用,对于复杂查询,可使用字典游标(cursor(dictionary=True))将结果映射为字典,提高可读性。
关闭数据库连接
操作完成后,需关闭游标和连接对象,以释放数据库资源:
cursor.close() conn.close()
为确保连接正确关闭,建议使用try-finally或上下文管理器(with语句)。
with conn.cursor() as cursor:
cursor.execute("SELECT * FROM products")
results = cursor.fetchall()
异常处理与最佳实践
数据库操作可能因网络问题、权限不足或SQL语法错误而异常,需使用try-except捕获异常。

try:
conn = mysql.connector.connect(...)
cursor = conn.cursor()
cursor.execute("INSERT INTO orders (item) VALUES ('laptop')")
conn.commit()
except mysql.connector.Error as err:
print(f"Error: {err}")
conn.rollback()
finally:
cursor.close()
conn.close()
最佳实践包括:使用参数化查询防止SQL注入(如cursor.execute("SELECT * FROM users WHERE name = %s", (username,))),合理设置连接超时时间,以及定期备份数据库。
FAQs
Q1: 如何处理数据库连接超时问题?
A1: 连接超时通常由网络延迟或数据库服务器负载过高导致,可通过设置连接超时参数(如connect_timeout=30)调整等待时间,或检查数据库服务器状态,若频繁超时,建议优化网络环境或使用连接池管理连接。
Q2: Python如何实现数据库事务管理?
A2: 事务通过commit()提交或rollback()回滚,在执行多条SQL语句时,需确保所有操作成功后调用commit(),否则调用rollback()撤销更改。
try:
cursor.execute("UPDATE accounts SET balance = balance - 100 WHERE id = 1")
cursor.execute("UPDATE accounts SET balance = balance + 100 WHERE id = 2")
conn.commit()
except:
conn.rollback()