踩坑pymysql

def select(user_id):
    """
    根据用户 ID 查询用户数据，若不存在则创建新用户并返回用户数据
    """
    cursor = config.db.cursor()
    try:
        # 查询用户数据
        sql = f"SELECT * FROM `{table}` WHERE `pd_id` = {user_id}"
        cursor.execute(sql)
        results = cursor.fetchall()

        # 判断用户是否存在
        if len(results) == 1:
            # 存在则返回用户数据
            user_data = results
            print("存在")
        else:
            print("不存在")
            # 关闭游标
            cursor.close()
            cursor = config.db.cursor()
            # 不存在则创建新用户
            sql = f"INSERT IGNORE INTO {table}(pd_id) values(%s)"
            data = [user_id]
            cursor.executemany(sql, data)
            # 写入
            config.db.commit()
            # 再次查询用户数据
            sql = f"SELECT * FROM `{table}` WHERE `pd_id` = {user_id}"
            cursor.execute(sql)
            results = cursor.fetchall()
            if len(results) == 1:
                user_data = results
            else:
                raise Exception(f"Failed to create user with ID {user_id}")

        # 构造用户对象并返回
        user = User(user_data)
        cursor.close()
        return user

    except Exception as e:
        cursor.close()
        raise e