From 2669d0e0a93e06983dade724146bd0e1b6da6ba7 Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期二, 12 十一月 2024 18:42:05 +0800
Subject: [PATCH] 增加接口

---
 app/models/user_model.py |   29 +++++++++++++++--------------
 1 files changed, 15 insertions(+), 14 deletions(-)

diff --git a/app/models/user_model.py b/app/models/user_model.py
index 81715df..aea4cbd 100644
--- a/app/models/user_model.py
+++ b/app/models/user_model.py
@@ -6,16 +6,16 @@
 from app.models.base_model import Base
 
 user_organization_table = Table('user_organization',Base.metadata
-                                   , Column('user_id', Integer, ForeignKey('user.id'))
-                                   , Column('organization_id', String(36), ForeignKey('organization.id')))
+                                   , Column('user_id', Integer, ForeignKey('user.id', ondelete='CASCADE'))
+                                   , Column('organization_id', String(36), ForeignKey('organization.id', ondelete='CASCADE')))
 
 user_role_table = Table('user_role', Base.metadata
-                           , Column('user_id', Integer, ForeignKey('user.id'))
-                           , Column('role_id', String(36), ForeignKey('role.id')))
+                           , Column('user_id', Integer, ForeignKey('user.id', ondelete='CASCADE'))
+                           , Column('role_id', String(36), ForeignKey('role.id', ondelete='CASCADE')))
 
 user_group_table = Table('user_group', Base.metadata
-                           , Column('user_id', Integer, ForeignKey('user.id'))
-                           , Column('group_id', Integer, ForeignKey('group.id')))
+                           , Column('user_id', Integer, ForeignKey('user.id', ondelete='CASCADE'))
+                           , Column('group_id', Integer, ForeignKey('group.id', ondelete='CASCADE')))
 
 
 
@@ -89,7 +89,7 @@
             'sex': self.sex,
             'age': self.age,
             "status": self.status,
-            'photo': self.phone,
+            'phone': self.phone,
             'email': self.email,
             # 'phoneNumber': self.phone_number
         }
@@ -101,16 +101,17 @@
         json['groups'] = [group.to_dict() for group in self.groups]
 
         roles = []
-        # if len(self.roles.all()) > 0:
-        #     roles = [role.to_json() for role in self.roles]
-        # organization_roles = [role.to_json() for organization in self.organizations for role in
-        #                       organization.role_list]
-        # for role in organization_roles:
-        #     if role not in roles:
-        #         roles.append(role)
+        # if len(self.roles) > 0:
+        roles = [role.to_json() for role in self.roles]
+        organization_roles = [role.to_json() for organization in self.organizations for role in
+                              organization.roles]
+        for role in organization_roles:
+            if role not in roles:
+                roles.append(role)
         json['roles'] = roles
         return json
 
+
     def get_children(self):
         """閫掑綊鑾峰彇鎸囧畾鐢ㄦ埛鐨勫瓙鐢ㄦ埛ID鍒楄〃"""
         users = UserModel.query.filter_by(CREATOR=self.ID).all()

--
Gitblit v1.8.0