Python连接MySQL：PyMySQL入门与实践指南

1. PyMySQL基础入门与安装指南PyMySQL是Python3中用于连接MySQL数据库的纯Python驱动库它实现了Python数据库API规范2.0版本。与Python2时代广泛使用的MySQLdb不同PyMySQL完全用Python编写不需要编译任何C扩展模块这使得它在跨平台部署时更加方便。1.1 PyMySQL的核心特性PyMySQL具有以下几个显著特点纯Python实现无需编译安装支持Python3Python2可使用MySQLdb实现了Python DB API 2.0规范支持SSL安全连接包含连接池支持兼容大多数MySQL特性包括存储过程、事务等在实际项目中我发现PyMySQL相比其他MySQL连接器有几个优势首先它的安装过程非常简单特别是对于没有编译环境的Windows用户其次它的错误信息更加友好调试起来更方便最后由于是纯Python实现可以方便地查看源码理解内部机制。1.2 安装PyMySQL的多种方式安装PyMySQL最常用的方式是通过pippip install pymysql如果你需要安装特定版本可以指定版本号pip install pymysql1.0.2在某些特殊环境下可能需要从源码安装。这时可以克隆GitHub仓库git clone https://github.com/PyMySQL/PyMySQL cd PyMySQL python setup.py install注意从源码安装需要确保系统已安装setuptools。如果遇到ImportError: No module named setuptools错误需要先安装setuptools。在Linux系统上可以通过以下命令安装setuptoolswget https://bootstrap.pypa.io/ez_setup.py python ez_setup.py2. 数据库连接与基本操作2.1 建立数据库连接使用PyMySQL连接MySQL数据库的基本代码如下import pymysql # 创建数据库连接 connection pymysql.connect( hostlocalhost, # 数据库服务器地址 userusername, # 数据库用户名 passwordpassword, # 数据库密码 databasedbname, # 数据库名 port3306, # 端口默认为3306 charsetutf8mb4, # 字符集 cursorclasspymysql.cursors.DictCursor # 使用字典游标 ) try: with connection.cursor() as cursor: # 执行SQL查询 sql SELECT * FROM users WHERE id %s cursor.execute(sql, (1,)) # 获取查询结果 result cursor.fetchone() print(result) finally: # 关闭连接 connection.close()在实际项目中我建议将数据库连接信息存储在配置文件中而不是硬编码在代码里。这样可以方便地在不同环境开发、测试、生产之间切换。2.2 使用上下文管理器简化连接管理Python的with语句可以简化连接管理确保连接在使用后被正确关闭。我们可以封装一个数据库上下文管理器import pymysql from contextlib import contextmanager contextmanager def get_db_connection(): conn pymysql.connect( hostlocalhost, useruser, passwordpass, databasedb, charsetutf8mb4, cursorclasspymysql.cursors.DictCursor ) try: yield conn finally: conn.close() # 使用示例 with get_db_connection() as conn: with conn.cursor() as cursor: cursor.execute(SELECT * FROM products) results cursor.fetchall() for row in results: print(row)这种写法不仅代码更简洁而且能确保在任何情况下包括发生异常时数据库连接都会被正确关闭避免资源泄漏。3. 执行SQL查询与事务处理3.1 基本CRUD操作PyMySQL支持所有标准的SQL操作包括增删改查CRUD。下面是一些常见操作的示例查询数据with connection.cursor() as cursor: # 查询单条记录 cursor.execute(SELECT * FROM users WHERE id %s, (1,)) user cursor.fetchone() # 查询多条记录 cursor.execute(SELECT * FROM users WHERE status %s, (active,)) active_users cursor.fetchall() # 获取记录数量 cursor.execute(SELECT COUNT(*) FROM users) user_count cursor.fetchone()[COUNT(*)]插入数据try: with connection.cursor() as cursor: sql INSERT INTO users (name, email) VALUES (%s, %s) cursor.execute(sql, (John Doe, johnexample.com)) # 提交事务 connection.commit() except: # 发生错误时回滚 connection.rollback() raise更新数据try: with connection.cursor() as cursor: sql UPDATE users SET name %s WHERE id %s cursor.execute(sql, (John Smith, 1)) print(f更新了{cursor.rowcount}条记录) connection.commit() except: connection.rollback() raise删除数据try: with connection.cursor() as cursor: sql DELETE FROM users WHERE id %s cursor.execute(sql, (1,)) connection.commit() except: connection.rollback() raise3.2 事务处理MySQL支持事务PyMySQL也提供了完整的事务支持。事务的四个特性ACID是原子性Atomicity事务中的所有操作要么全部完成要么全部不完成一致性Consistency事务执行前后数据库从一个一致状态变为另一个一致状态隔离性Isolation多个事务并发执行时一个事务的执行不应影响其他事务持久性Durability事务一旦提交其结果就是永久性的在PyMySQL中默认情况下每个SQL语句都会自动提交。要使用事务需要先关闭自动提交connection pymysql.connect( hostlocalhost, useruser, passwordpass, databasedb, autocommitFalse # 关闭自动提交 ) try: with connection.cursor() as cursor: # 执行多个SQL语句 cursor.execute(UPDATE accounts SET balance balance - 100 WHERE id 1) cursor.execute(UPDATE accounts SET balance balance 100 WHERE id 2) # 手动提交事务 connection.commit() except: # 发生错误时回滚 connection.rollback() raise finally: connection.close()在实际项目中我发现事务的正确使用非常重要。特别是在处理金融交易、库存管理等场景时一定要确保事务的完整性和一致性。4. 高级特性与最佳实践4.1 批量操作与参数化查询PyMySQL支持批量插入和更新操作这可以显著提高性能# 批量插入 data [ (John, johnexample.com), (Jane, janeexample.com), (Bob, bobexample.com) ] with connection.cursor() as cursor: sql INSERT INTO users (name, email) VALUES (%s, %s) cursor.executemany(sql, data) connection.commit()参数化查询不仅可以防止SQL注入还能提高性能因为MySQL可以缓存编译后的语句# 不安全的写法容易受到SQL注入攻击 cursor.execute(SELECT * FROM users WHERE id user_input) # 安全的参数化查询 cursor.execute(SELECT * FROM users WHERE id %s, (user_input,))4.2 错误处理与异常管理PyMySQL定义了多种数据库操作相关的异常合理的错误处理可以使程序更加健壮try: with connection.cursor() as cursor: cursor.execute(SELECT * FROM non_existent_table) except pymysql.err.ProgrammingError as e: print(fSQL语法错误: {e}) except pymysql.err.OperationalError as e: print(f数据库操作错误: {e}) except Exception as e: print(f其他错误: {e})常见的PyMySQL异常包括ProgrammingErrorSQL语法错误OperationalError数据库操作错误如连接失败IntegrityError完整性约束错误如重复键DataError数据处理错误如数据超出范围4.3 连接池与性能优化对于高并发应用频繁创建和关闭数据库连接会影响性能。可以使用连接池来管理数据库连接from pymysql import pools # 创建连接池 pool pools.Pool( hostlocalhost, useruser, passwordpass, databasedb, min2, # 最小连接数 max10, # 最大连接数 autocommitFalse ) # 从连接池获取连接 connection pool.connection() try: with connection.cursor() as cursor: cursor.execute(SELECT * FROM users) results cursor.fetchall() finally: # 将连接返回连接池 connection.close()其他性能优化建议合理使用索引加速查询避免在循环中执行SQL语句只查询需要的列避免SELECT *对于大量数据考虑使用分页查询4.4 使用ORM与PyMySQL结合虽然PyMySQL提供了直接的SQL接口但在大型项目中使用ORM对象关系映射框架可以使代码更加清晰from sqlalchemy import create_engine from sqlalchemy.orm import sessionmaker # 创建SQLAlchemy引擎底层使用PyMySQL engine create_engine(mysqlpymysql://user:passlocalhost/db) # 创建Session工厂 Session sessionmaker(bindengine) # 使用Session session Session() try: # 执行查询 users session.execute(SELECT * FROM users) for user in users: print(user) finally: session.close()在实际项目中我发现对于简单的查询直接使用PyMySQL更高效而对于复杂的业务逻辑使用ORM可以提高开发效率和代码可维护性。