import asyncpg

# 删除数据
async def delete_question(db: asyncpg.Connection, question_id: str):
    delete_sql = """
        DELETE FROM t_bi_question WHERE id = $1
    """
    await db.execute(delete_sql, question_id)

# 插入数据
async def insert_question(db: asyncpg.Connection, question_id: str, question: str):
    insert_sql = """
        INSERT INTO t_bi_question (id, question, state_id, is_system, is_collect)
        VALUES ($1, $2, $3, $4, $5)
    """
    await db.execute(insert_sql, question_id, question, 0, 0, 0)

# 修改数据
async def update_question_by_id(db: asyncpg.Connection, question_id: str, **kwargs):
    update_fields = {k: v for k, v in kwargs.items() if v is not None}
    if not update_fields:
        return False

    set_clause = ", ".join([f"{field} = ${i+1}" for i, field in enumerate(update_fields.keys())])
    sql = f"""
        UPDATE t_bi_question
        SET {set_clause}
        WHERE id = ${len(update_fields) + 1}
    """
    params = list(update_fields.values()) + [question_id]

    try:
        await db.execute(sql, *params)
        return True
    except Exception as e:
        print(f"更新失败: {e}")
        return False

# 根据问题 ID 查询 SQL
async def get_question_by_id(db: asyncpg.Connection, question_id: str):
    select_sql = """
        SELECT * FROM t_bi_question WHERE id = $1
    """
    _data = await db.fetch(select_sql, question_id)
    return _data

# 保存系统推荐
async def set_system_recommend_questions(db: asyncpg.Connection, question_id: str, flag: str):
    sql = """
        UPDATE t_bi_question
        SET is_system = $1 WHERE id = $2
    """
    try:
        await db.execute(sql, int(flag), question_id)
        return True
    except Exception as e:
        print(f"更新失败: {e}")
        return False

# 设置用户收藏
async def set_user_collect_questions(db: asyncpg.Connection, question_id: str, flag: str):
    sql = """
        UPDATE t_bi_question
        SET is_collect = $1 WHERE id = $2
    """
    try:
        await db.execute(sql, int(flag), question_id)
        return True
    except Exception as e:
        print(f"更新失败: {e}")
        return False

# 查询系统推荐问题
async def get_system_recommend_questions(db: asyncpg.Connection):
    sql = """
        SELECT * FROM t_bi_question WHERE is_system = 1
    """
    _data = await db.fetch(sql)
    return _data

# 查询用户收藏问题
async def get_user_collect_questions(db: asyncpg.Connection):
    sql = """
        SELECT * FROM t_bi_question WHERE is_collect = 1
    """
    _data = await db.fetch(sql)
    return _data