From e49417ee6eecf7f00bd77e01a6b4a1c7a420708b Mon Sep 17 00:00:00 2001
From: xuyonghao <898441624@qq.com>
Date: 星期三, 04 十二月 2024 15:08:03 +0800
Subject: [PATCH] 知识库列表获取

---
 app/service/knowledge.py |    7 +++++--
 1 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/app/service/knowledge.py b/app/service/knowledge.py
index c9c89b4..c0121fa 100644
--- a/app/service/knowledge.py
+++ b/app/service/knowledge.py
@@ -3,8 +3,11 @@
 from Log import logger
 
 
-async def get_knowledge_list(db, role, user_id):
-    if role == "admin":
+async def get_knowledge_list(db, user_id):
+    user = db.query(UserModel).filter(UserModel.id == user_id).first()
+    if user is None:
+        return {"rows": []}
+    if user.permission == "admin":
         kld_list = db.query(KnowledgeModel).all()
     else:
         user = db.query(UserModel).filter(UserModel.id == user_id).first()

--
Gitblit v1.8.0