From c941b948f1fa8eff615985a83110a401ae426ab3 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期三, 18 十二月 2024 11:03:49 +0800
Subject: [PATCH] 知识库和智能体同步

---
 app/models/role_model.py |   32 ++++++++++++++++++++++++--------
 1 files changed, 24 insertions(+), 8 deletions(-)

diff --git a/app/models/role_model.py b/app/models/role_model.py
index de18a04..42895df 100644
--- a/app/models/role_model.py
+++ b/app/models/role_model.py
@@ -25,8 +25,9 @@
     seq = Column(Integer)
     roleKey = Column(String(100))
     dataScope = Column(Integer)
-    status = Column(String(10), default="0")
+    status = Column(String(10), default="1")
     creator = Column(Integer)
+    role_type = Column(Integer, default=1)
 
     # 鍖呭惈璧勬簮
     resources = relationship('ResourceModel',
@@ -57,32 +58,47 @@
             'dataScope': self.dataScope
         }
 
-        if len(self.resources) > 0:
-            json['resources'] = [resource.to_json() for resource in self.resources]
+        # if len(self.resources) > 0:
+        json['resources'] = [resource.to_json() for resource in self.resources]
 
-        if hasattr(self, 'flag'):
-            json['flag'] = self.flag
+        json['dept'] = [dept.to_base_json() for dept in self.organizations]
 
         return json
 
+
+    def to_dict(self):
+        json = {
+            'roleId': self.id,
+            'createTime': self.created_at.strftime('%Y-%m-%d %H:%M:%S'),
+            'updateTime': self.updated_at.strftime('%Y-%m-%d %H:%M:%S'),
+            'roleName': self.name,
+            'remark': self.description,
+            'iconCls': self.iconCls,
+            'roleSort': self.seq,
+            'status': self.status,
+            'roleKey': self.roleKey,
+            'dataScope': self.dataScope
+        }
+        return json
 
 class RoleData(BaseModel):
     remark: Optional[str] = ""
     roleName: str
     roleKey: Optional[str] = ""
-    dataScope: Optional[str] = ""
+    dataScope: Optional[int] = 0
 
 
 class RoleStatus(BaseModel):
-    role_id: str
+    roleId: str
     status: constr(min_length=1, max_length=1, pattern='^(0|1)$')
 
 
 class RoleEdit(BaseModel):
-    role_id: str
+    roleId: str
     remark: Optional[str] = ""
     roleName: Optional[str] = ""
     resources: Optional[list] = []
     roleKey: Optional[str] = ""
     dataScope: Optional[int] = 0
     editType: int
+    depts: Optional[list] = []

--
Gitblit v1.8.0