From e5e2be29cfc033c976558a4d0b5fa73426818b90 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 26 二月 2025 11:30:48 +0800
Subject: [PATCH] 增加系统信息修改

---
 app/service/knowledge.py |   24 +++++++++++++++++++++---
 1 files changed, 21 insertions(+), 3 deletions(-)

diff --git a/app/service/knowledge.py b/app/service/knowledge.py
index 7131186..4fb834f 100644
--- a/app/service/knowledge.py
+++ b/app/service/knowledge.py
@@ -1,3 +1,5 @@
+from datetime import datetime
+
 from sqlalchemy import or_
 
 from app.models import KnowledgeModel, group_knowledge_table
@@ -13,9 +15,11 @@
     query = db.query(KnowledgeModel).filter(KnowledgeModel.knowledge_type==knowledge)
     if user.permission != "admin":
         klg_list = [j.id for i in user.groups for j in i.knowledges]
-        query = query.filter(or_(KnowledgeModel.tenant_id == user_id, KnowledgeModel.id.in_(klg_list)))
+        query = query.filter(or_(KnowledgeModel.id.in_(klg_list), KnowledgeModel.tenant_id == str(user_id)))
+
     if location:
-        query = query.filter(KnowledgeModel.permission == "team")
+        query = query.filter(or_(KnowledgeModel.permission == "team", KnowledgeModel.tenant_id == str(user_id)))
+
     if keyword:
         query = query.filter(KnowledgeModel.name.like('%{}%'.format(keyword)))
     total = query.count()
@@ -26,7 +30,7 @@
 
 async def create_knowledge_service(db, klg_id, name, description, icon, klg_type, user_id):
     try:
-        dialog_model = KnowledgeModel(id=klg_id,name=name, description=description,icon=icon, knowledge_type=klg_type, tenant_id=user_id)
+        dialog_model = KnowledgeModel(id=klg_id,name=name, description=description,icon=icon, knowledge_type=klg_type, tenant_id=user_id,update_date=datetime.now(),create_date=datetime.now())
         db.add(dialog_model)
         db.commit()
         db.refresh(dialog_model)
@@ -48,6 +52,8 @@
                 knowledge.permission = app_knowledge["permission"]
                 knowledge.documents = app_knowledge["doc_num"]
                 knowledge.status = app_knowledge["status"]
+                knowledge.embd_id = app_knowledge["embd_id"]
+                knowledge.update_date = datetime.now()
                 db.add(knowledge)
                 db.commit()
                 db.refresh(knowledge)
@@ -69,3 +75,15 @@
         return False
     return True
 
+
+
+async def update_knowledge_icon_service(db, klg_id, icon):
+    try:
+        db.query(KnowledgeModel).filter_by(id=klg_id).update({"icon":icon, "update_date": datetime.now()})
+        db.commit()
+    except Exception as e:
+        logger.error(e)
+        db.rollback()
+        return False
+    return True
+

--
Gitblit v1.8.0