from Text2Sql.Util.PostgreSQLUtil import PostgreSQLUtil


# 删除数据
def delete_question(db, question_id: str):
    # 删除 t_bi_question 表中的数据
    delete_sql = """
                DELETE FROM t_bi_question WHERE id = %s
                """
    db.execute_query(delete_sql, (question_id,))


# 插入数据
def insert_question(db, question_id: str, question: str):
    # 向 t_bi_question 表插入数据
    insert_sql = """
               INSERT INTO t_bi_question (id,question, state_id, is_system, is_collect)
               VALUES (%s,%s, %s, %s, %s)
           """
    db.execute_query(insert_sql, (question_id, question, 0, 0, 0))


# 修改数据
'''
示例:
# 更新 question 和 state_id 字段
update_question_by_id(db, question_id=1, question="新的问题描述", state_id=1)

# 只更新 excel_file_name 字段
update_question_by_id(db, question_id=1, excel_file_name="new_excel.xlsx")

# 只更新 is_collect 字段
update_question_by_id(db, question_id=1, is_collect=1)

# 不更新任何字段（因为所有参数都是 None）
update_question_by_id(db, question_id=1, question=None, state_id=None)
'''


def update_question_by_id(db: PostgreSQLUtil, question_id: str, **kwargs):
    """
    根据主键更新 t_bi_question 表，只更新非 None 的字段
    :param db: PostgreSQLUtil 实例
    :param question_id: 主键 id
    :param kwargs: 需要更新的字段和值
    :return: 更新是否成功
    """
    # 过滤掉值为 None 的字段
    update_fields = {k: v for k, v in kwargs.items() if v is not None}
    if not update_fields:
        return False  # 没有需要更新的字段

    # 动态构建 SET 子句
    set_clause = ", ".join([f"{field} = %s" for field in update_fields.keys()])
    # 构建完整 SQL
    sql = f"""
        UPDATE t_bi_question
        SET {set_clause}
        WHERE id = %s
    """
    # 参数列表
    params = list(update_fields.values()) + [question_id]

    # 执行更新
    try:
        db.execute_query(sql, params)
        return True
    except Exception as e:
        print(f"更新失败: {e}")
        return False


# 根据 问题id 查询 sql
def get_question_by_id(db, question_id: str):
    select_sql = """
                        select * from t_bi_question where id=%s
                        """
    _data = db.execute_query(select_sql, (question_id,))
    return _data


# 保存系统推荐
def set_system_recommend_questions(db, question_id: str, flag: str):
    sql = f"""
            UPDATE t_bi_question
            SET is_system =%s  WHERE id = %s
        """
    # 执行更新
    try:
        db.execute_query(sql, int(flag), question_id)
        return True
    except Exception as e:
        print(f"更新失败: {e}")
        return False

# 设置用户收藏
def set_user_collect_questions(db, question_id: str, flag: str):
    sql = f"""
            UPDATE t_bi_question
            SET is_collect =%s  WHERE id = %s
        """
    # 执行更新
    try:
        db.execute_query(sql, int(flag), question_id)
        return True
    except Exception as e:
        print(f"更新失败: {e}")
        return False

# 查询有哪些系统推荐问题
def get_system_recommend_questions(db):
    sql = """
    SELECT * FROM t_bi_question WHERE is_system = 1
    """
    _data = db.execute_query(sql)
    return _data

# 查询有哪些用户收藏问题
def get_user_collect_questions(db):
    # 从t_bi_question表中获取所有is_collect=1的数据
    sql="""
    SELECT * FROM t_bi_question WHERE is_collect = 1
    """
    _data = db.execute_query(sql)
    return _data