'commit'

AI/Text2Sql/Model/biModel.py

@@ -80,7 +80,7 @@ def get_question_by_id(db, question_id: str):
 
 
 # 保存系统推荐
-def model_set_system_recommend(db, question_id: str, flag: str):
+def set_system_recommend_questions(db, question_id: str, flag: str):
     sql = f"""
             UPDATE t_bi_question
             SET is_system =%s  WHERE id = %s
@@ -94,7 +94,7 @@ def model_set_system_recommend(db, question_id: str, flag: str):
         return False
 
 # 设置用户收藏
-def model_set_user_collect(db, question_id: str, flag: str):
+def set_user_collect_questions(db, question_id: str, flag: str):
     sql = f"""
             UPDATE t_bi_question
             SET is_collect =%s  WHERE id = %s
@@ -105,4 +105,21 @@ def model_set_user_collect(db, question_id: str, flag: str):
         return True
     except Exception as e:
         print(f"更新失败: {e}")
-        return False
+        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

AI/Text2Sql/app.py

@@ -188,17 +188,33 @@ async def get_docx_file(
 # 设置问题为系统推荐问题 ,0:取消，1：设置
 @app.post("/questions/set_system_recommend")
 def set_system_recommend(question_id: str = Form(...), flag: str = Form(...), db: PostgreSQLUtil = Depends(get_db)):
-    model_set_system_recommend(db, question_id, flag)
+    set_system_recommend_questions(db, question_id, flag)
     # 提示保存成功
     return {"success": True, "message": "保存成功"}
 
 # 设置问题为用户收藏问题 ,0:取消，1：设置
 @app.post("/questions/set_user_collect")
 def set_user_collect(question_id: str = Form(...), flag: str = Form(...), db: PostgreSQLUtil = Depends(get_db)):
-    model_set_user_collect(db, question_id, flag)
+    set_user_collect_questions(db, question_id, flag)
     # 提示保存成功
     return {"success": True, "message": "保存成功"}
 
+# 查询有哪些系统推荐问题
+@app.get("/questions/get_system_recommend")
+def get_system_recommend(db: PostgreSQLUtil = Depends(get_db)):
+    # 查询所有系统推荐问题
+    system_recommend_questions = get_system_recommend_questions(db)
+    # 返回查询结果
+    return {"success": True, "data": system_recommend_questions}
+
+# 查询有哪些用户收藏问题
+@app.get("/questions/get_user_collect")
+def get_user_collect(db: PostgreSQLUtil = Depends(get_db)):
+    # 查询所有用户收藏问题
+    user_collect_questions = get_user_collect_questions(db)
+    # 返回查询结果
+    return {"success": True, "data": user_collect_questions}
+
 
 # 确保直接运行脚本时启动 FastAPI 应用
 if __name__ == "__main__":