From 536a8bbbe42c527ef9eaee9a8a8d2ee3615f5a11 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 14 一月 2025 16:19:04 +0800
Subject: [PATCH] 文档智能过程信息显示

---
 app/service/dialog.py |   40 +++++++++++++++++++++-------------------
 1 files changed, 21 insertions(+), 19 deletions(-)

diff --git a/app/service/dialog.py b/app/service/dialog.py
index bfc1fc0..c302d10 100644
--- a/app/service/dialog.py
+++ b/app/service/dialog.py
@@ -13,23 +13,26 @@
     user = db.query(UserModel).filter(UserModel.id == user_id).first()
     if user is None:
         return {"rows": []}
-    query = db.query(DialogModel).filter(DialogModel.status != Dialog_STATSU_DELETE)
-    id_list = []
-    if label:
-        id_list = [i.object_id for i in db.query(LabelWorkerModel).filter(LabelWorkerModel.label_id==label).all()]
-    if user.permission != "admin":
-        dia_list = [j.id for i in user.groups for j in i.dialogs if not label or j.id in id_list]
-        query = query.filter(or_(DialogModel.tenant_id == user_id, DialogModel.id.in_(dia_list)))
+    query = db.query(DialogModel)
+    if status:
+        query = query.filter(DialogModel.status == status)
     else:
-        if label:
-            query = query.filter(or_(DialogModel.id.in_(id_list)))
+        query = query.filter(DialogModel.status != Dialog_STATSU_DELETE)
+    id_list = []
+    # if label:
+    #     id_list = [i.object_id for i in db.query(LabelWorkerModel).filter(LabelWorkerModel.label_id==label).all()]
+    if user.permission != "admin":
+        dia_list = [j.id for i in user.groups for j in i.dialogs]
+        query = query.filter(or_(DialogModel.tenant_id == user_id, DialogModel.id.in_(dia_list)))
+    # else:
+    if label:
+        id_list = [i.object_id for i in db.query(LabelWorkerModel).filter(LabelWorkerModel.label_id == label).all()]
+        query = query.filter(DialogModel.id.in_(id_list))
 
     if keyword:
         query = query.filter(DialogModel.name.like('%{}%'.format(keyword)))
 
-    if status:
-        # print(status)
-        query = query.filter(DialogModel.status == status)
+
     query = query.order_by(DialogModel.update_date.desc())
     total = query.count()
     if page_size:
@@ -142,15 +145,14 @@
     if user is None:
         return {"rows": []}
     query = db.query(DialogModel).filter(DialogModel.status != Dialog_STATSU_DELETE)
-    id_list = []
-    if label:
-        id_list = set([i.object_id for i in db.query(LabelWorkerModel).filter(LabelWorkerModel.label_id.in_(label)).all()])
     if user.permission != "admin":
-        dia_list = [j.id for i in user.groups for j in i.dialogs if not label or j.id in id_list]
+        dia_list = [j.id for i in user.groups for j in i.dialogs]
         query = query.filter(or_(DialogModel.tenant_id == user_id, DialogModel.id.in_(dia_list)))
-    else:
-        if label:
-            query = query.filter(or_(DialogModel.id.in_(id_list)))
+
+    if label:
+        id_list = set(
+            [i.object_id for i in db.query(LabelWorkerModel).filter(LabelWorkerModel.label_id.in_(label)).all()])
+        query = query.filter(DialogModel.id.in_(id_list))
 
     if keyword:
         query = query.filter(DialogModel.name.like('%{}%'.format(keyword)))

--
Gitblit v1.8.0