import pymysql
from pymysql import Error

def connect_mysql():
    """用PyMySQL连接MySQL的通用函数"""
    connection = None  # 初始化连接对象，避免后续报错
    try:
        # 1. 建立连接：传入MySQL配置参数
        connection = pymysql.connect(
            host='localhost',        # MySQL服务地址（本地默认localhost）
            database='test_db',      # 目标数据库名
            user='root',             # MySQL用户名（默认管理员账号为root）
            password='your_password' # 替换为你的MySQL密码（如无密码可传空字符串）
        )
        
        # 2. 验证连接是否成功
        if connection.open:  # connection.open判断连接是否存活
            db_version = connection.get_server_info()  # 获取MySQL服务版本
            print(f"✅ 成功连接MySQL，服务版本：{db_version}")
            
            # 查询当前连接的数据库名称，验证是否正确指向test_db
            cursor = connection.cursor()
            cursor.execute("SELECT DATABASE();")  # 执行SQL查询
            current_db = cursor.fetchone()        # 获取查询结果（元组类型）
            print(f"🔗 当前连接的数据库：{current_db[0]}")

    except Error as e:
        # 捕获连接异常（如密码错误、数据库不存在、服务未启动）
        print(f"❌ 连接MySQL失败！错误信息：{e}")
    finally:
        # 3. 关闭连接与游标，释放资源
        if connection and connection.open:
            cursor.close()    # 先关游标
            connection.close()# 再关连接
            print("\n🔌 MySQL连接已关闭")

# 调用函数测试连接
connect_mysql()

import pymysql
from pymysql import Error

def create_user_table():
    """在test_db数据库中创建user表"""
    connection = None
    try:
        connection = pymysql.connect(
            host='localhost',
            database='test_db',
            user='root',
            password='your_password'
        )
        
        if connection.open:
            cursor = connection.cursor()
            # 定义创建表的SQL：IF NOT EXISTS避免重复创建报错
            create_table_sql = """
            CREATE TABLE IF NOT EXISTS user (
                id INT AUTO_INCREMENT PRIMARY KEY,  # 主键（唯一标识），自增
                name VARCHAR(50) NOT NULL,          # 姓名，非空（不允许空值）
                age INT DEFAULT 0,                  # 年龄，默认值0（未填时自动补0）
                email VARCHAR(100) UNIQUE,          # 邮箱，唯一（避免重复注册）
                create_time DATETIME DEFAULT CURRENT_TIMESTAMP  # 创建时间，默认当前时间
            ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;  # InnoDB支持事务，utf8mb4支持emoji
            """
            cursor.execute(create_table_sql)  # 执行创建表的SQL
            print("✅ user表创建成功（或已存在）")

    except Error as e:
        print(f"❌ 创建表失败！错误信息：{e}")
    finally:
        if connection and connection.open:
            cursor.close()
            connection.close()

create_user_table()

import pymysql
from pymysql import Error

def insert_user_data():
    """向user表插入数据（单条+批量）"""
    connection = None
    try:
        connection = pymysql.connect(
            host='localhost',
            database='test_db',
            user='root',
            password='your_password'
        )
        
        if connection.open:
            cursor = connection.cursor()
            
            # 1. 单条插入：用%s作为占位符（关键！避免SQL注入）
            single_insert_sql = "INSERT INTO user (name, age, email) VALUES (%s, %s, %s);"
            single_user = ("张三", 25, "zhangsan@example.com")  # 数据顺序与占位符对应
            cursor.execute(single_insert_sql, single_user)
            print(f"✅ 单条插入成功，新增用户ID：{cursor.lastrowid}")  # lastrowid获取自增主键
            
            # 2. 批量插入：executemany()接收“SQL语句+数据列表”
            batch_insert_sql = "INSERT INTO user (name, age, email) VALUES (%s, %s, %s);"
            batch_users = [
                ("李四", 28, "lisi@example.com"),
                ("王五", 22, "wangwu@example.com"),
                ("赵六", 30, "zhaoliu@example.com")
            ]
            cursor.executemany(batch_insert_sql, batch_users)
            print(f"✅ 批量插入成功，共插入{cursor.rowcount}条数据")  # rowcount获取影响行数
            
            # 关键：写操作必须提交事务！否则数据仅在当前连接可见，未真正写入数据库
            connection.commit()
            print("✅ 事务提交成功")

    except Error as e:
        # 出错时回滚事务，避免数据不一致（如部分插入成功、部分失败）
        if connection:
            connection.rollback()
        print(f"❌ 插入数据失败！已回滚，错误信息：{e}")
    finally:
        if connection and connection.open:
            cursor.close()
            connection.close()

insert_user_data()

import pymysql
from pymysql import Error

def update_and_delete_user():
    """更新和删除user表数据"""
    connection = None
    try:
        connection = pymysql.connect(
            host='localhost',
            database='test_db',
            user='root',
            password='your_password'
        )
        
        if connection.open:
            cursor = connection.cursor()
            
            # 1. 更新数据：将“张三”的年龄改为26
            update_sql = "UPDATE user SET age = %s WHERE name = %s;"
            cursor.execute(update_sql, (26, "张三"))
            print(f"✅ 更新完成，影响行数：{cursor.rowcount}")  # 若“张三”不存在，rowcount为0
            
            # 2. 删除数据：删除邮箱为“zhaoliu@example.com”的用户
            delete_sql = "DELETE FROM user WHERE email = %s;"
            cursor.execute(delete_sql, ("zhaoliu@example.com",))
            print(f"✅ 删除完成，影响行数：{cursor.rowcount}")
            
            # 提交事务（写操作必须提交，否则修改不生效）
            connection.commit()
            print("✅ 事务提交成功")

    except Error as e:
        if connection:
            connection.rollback()  # 出错回滚，恢复到操作前状态
        print(f"❌ 更新/删除失败！已回滚，错误信息：{e}")
    finally:
        if connection and connection.open:
            cursor.close()
            connection.close()

update_and_delete_user()