python pymysql封装函数

Python pymysql 封装函数

基础概念

pymysql 是一个纯 Python 实现的 MySQL 客户端库，它允许 Python 程序与 MySQL 数据库进行交互。封装函数可以帮助我们简化数据库操作，提高代码的可读性和可维护性。

类型

封装函数通常包括以下几种类型：

连接函数：用于创建数据库连接。
执行查询函数：用于执行 SQL 查询并返回结果。
事务处理函数：用于管理数据库事务。
关闭连接函数：用于关闭数据库连接。

应用场景

封装函数广泛应用于各种需要与 MySQL 数据库交互的场景，如 Web 应用、数据分析、自动化脚本等。

示例代码

以下是一个简单的 pymysql 封装函数的示例：

import pymysql

class MySQLDB:
    def __init__(self, host, user, password, db, port=3306):
        self.host = host
        self.user = user
        self.password = password
        self.db = db
        self.port = port
        self.conn = None

    def connect(self):
        try:
            self.conn = pymysql.connect(
                host=self.host,
                user=self.user,
                password=self.password,
                db=self.db,
                port=self.port,
                charset='utf8mb4',
                cursorclass=pymysql.cursors.DictCursor
            )
            print("数据库连接成功")
        except pymysql.MySQLError as e:
            print(f"数据库连接失败: {e}")

    def execute_query(self, sql, params=None):
        if not self.conn:
            print("数据库未连接")
            return None
        try:
            with self.conn.cursor() as cursor:
                cursor.execute(sql, params)
                result = cursor.fetchall()
            self.conn.commit()
            return result
        except pymysql.MySQLError as e:
            print(f"查询执行失败: {e}")
            self.conn.rollback()
            return None

    def close(self):
        if self.conn:
            self.conn.close()
            print("数据库连接已关闭")

# 使用示例
if __name__ == "__main__":
    db = MySQLDB(host='localhost', user='root', password='password', db='testdb')
    db.connect()
    result = db.execute_query("SELECT * FROM users")
    print(result)
    db.close()

参考链接

pymysql 官方文档

常见问题及解决方法

连接失败：
- 原因：可能是数据库服务器未启动、用户名或密码错误、网络问题等。
- 解决方法：检查数据库服务器状态，确认用户名和密码正确，检查网络连接。

查询执行失败：
- 原因：可能是 SQL 语句错误、权限不足、数据库表不存在等。
- 解决方法：检查 SQL 语句语法，确认用户权限，确认数据库表存在。
事务处理问题：
- 原因：可能是事务未正确提交或回滚。
- 解决方法：确保在适当的位置调用 commit 或 rollback 方法。

通过封装函数，我们可以更方便地管理和操作数据库，减少重复代码，提高开发效率。