import pymysql  # 无报错则安装成功

import pymysqlfrom pymysql.err import OperationalError, ProgrammingError, IntegrityErrorclass MySQLDB:
    """MySQL 数据库工具类：封装连接、CRUD 操作、资源关闭"""
    def __init__(self, host='localhost', port=3306, user='root', password='123456', database='python_db', charset='utf8mb4'):
        # 初始化连接参数
        self.host = host
        self.port = port
        self.user = user
        self.password = password
        self.database = database
        self.charset = charset
        self.conn = None  # 数据库连接对象
        self.cursor = None  # 游标对象（执行 SQL 用）

    def connect(self):
        """建立数据库连接"""
        try:
            self.conn = pymysql.connect(
                host=self.host,
                port=self.port,
                user=self.user,
                password=self.password,
                database=self.database,
                charset=self.charset,
                cursorclass=pymysql.cursors.DictCursor  # 游标返回字典格式（键为字段名）
            )
            self.cursor = self.conn.cursor()
            print("数据库连接成功！")
        except OperationalError as e:
            print(f"数据库连接失败：{e}")
            raise  # 抛出异常，由调用者处理

    def close(self):
        """关闭游标和连接（避免资源泄漏）"""
        try:
            if self.cursor:
                self.cursor.close()
            if self.conn and self.conn.open:
                self.conn.close()
                print("数据库连接已关闭！")
        except Exception as e:
            print(f"关闭资源失败：{e}")

    def execute_sql(self, sql, params=None):
        """
        执行 SQL 语句（insert/update/delete）
        :param sql: SQL 语句（使用 %s 作为占位符）
        :param params: SQL 参数（元组/列表，如 (name, age)）
        :return: 受影响的行数
        """
        try:
            self.connect()  # 建立连接
            affected_rows = self.cursor.execute(sql, params)  # 执行 SQL（params 避免 SQL 注入）
            self.conn.commit()  # 提交事务（增删改必须提交）
            print(f"SQL 执行成功，受影响行数：{affected_rows}")
            return affected_rows        except (ProgrammingError, IntegrityError) as e:
            self.conn.rollback()  # 执行失败，回滚事务
            print(f"SQL 执行失败：{e}")
            raise
        finally:
            self.close()  # 无论成功失败，都关闭资源

    def query_sql(self, sql, params=None, fetch_all=True):
        """
        执行查询 SQL（select）
        :param sql: 查询 SQL 语句
        :param params: SQL 参数
        :param fetch_all: 是否查询所有结果（True 返回全部，False 返回单条）
        :return: 查询结果（字典列表/单个字典）
        """
        try:
            self.connect()
            self.cursor.execute(sql, params)
            if fetch_all:
                result = self.cursor.fetchall()  # 获取所有结果
            else:
                result = self.cursor.fetchone()  # 获取单条结果
            print(f"查询成功，共 {len(result) if fetch_all else 1} 条数据" if result else "查询无结果")
            return result        except ProgrammingError as e:
            print(f"查询失败：{e}")
            raise
        finally:
            self.close()

if __name__ == "__main__":
    # 初始化数据库工具类（替换为你的 MySQL 账号密码）
    db = MySQLDB(user='root', password='123456')

    # 1. 新增用户
    try:
        sql = "INSERT INTO users (name, age, email) VALUES (%s, %s, %s)"
        params = ("张三", 24, "zhangsan@example.com")  # 对应 SQL 中的 %s 占位符
        affected_rows = db.execute_sql(sql, params)
        print(f"新增用户成功，受影响行数：{affected_rows}")
    except Exception as e:
        print(f"新增失败：{e}")

if __name__ == "__main__":
    db = MySQLDB(user='root', password='123456')

    # 2. 按 ID 查询单条用户
    try:
        sql = "SELECT id, name, age, email, create_time FROM users WHERE id = %s"
        params = (1,)  # 查询 ID=1 的用户
        user = db.query_sql(sql, params, fetch_all=False)  # fetch_all=False 返回单条
        if user:
            print("查询到的用户信息：")
            print(f"ID：{user['id']}")
            print(f"姓名：{user['name']}")
            print(f"年龄：{user['age']}")
            print(f"邮箱：{user['email']}")
            print(f"注册时间：{user['create_time']}")
        else:
            print("未查询到该用户")
    except Exception as e:
        print(f"查询失败：{e}")

if __name__ == "__main__":
    db = MySQLDB(user='root', password='123456')

    # 3. 查询所有用户（按注册时间倒序）
    try:
        sql = "SELECT id, name, age, email FROM users ORDER BY create_time DESC"
        users = db.query_sql(sql, fetch_all=True)  # fetch_all=True 返回全部
        if users:
            print("所有用户列表：")
            for idx, user in enumerate(users, 1):
                print(f"序号 {idx}：ID={user['id']}，姓名={user['name']}，邮箱={user['email']}")
        else:
            print("暂无用户数据")
    except Exception as e:
        print(f"查询失败：{e}")

if __name__ == "__main__":
    db = MySQLDB(user='root', password='123456')

    # 4. 修改用户信息（修改 ID=1 的用户）
    try:
        sql = "UPDATE users SET name = %s, age = %s WHERE id = %s"
        params = ("张三_更新", 25, 1)  # 新姓名、新年龄、目标 ID
        affected_rows = db.execute_sql(sql, params)
        if affected_rows > 0:
            print("用户信息修改成功！")
        else:
            print("修改失败：未找到用户或数据未变更")
    except Exception as e:
        print(f"修改失败：{e}")

if __name__ == "__main__":
    db = MySQLDB(user='root', password='123456')

    # 5. 删除用户（删除 ID=1 的用户）
    try:
        sql = "DELETE FROM users WHERE id = %s"
        params = (1,)
        affected_rows = db.execute_sql(sql, params)
        if affected_rows > 0:
            print(f"删除成功！已删除 ID=1 的用户")
        else:
            print("删除失败：未找到该用户")
    except Exception as e:
        print(f"删除失败：{e}")