From 7a9f9cb3c34ebc0836720d0f5738bbe70070bbb3 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期四, 05 十二月 2024 14:22:24 +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