增加项目的各个功能

app/api/service/rbacService.py

@@ -0,0 +1,103 @@
+# encoding: UTF-8
+from ..dao.rbacDao import RbacDao
+
+
+def has_permission(permission_code, permission_codes):
+    if not permission_code:
+        return True
+    if not permission_codes:
+        return False
+    if permission_code in permission_codes:
+        return True
+    if '*:*' in permission_codes:
+        return True
+    if ':' in permission_code:
+        module_code = permission_code.split(':', 1)[0]
+        if f'{module_code}:*' in permission_codes:
+            return True
+        if '_' in module_code:
+            parent_module_code = module_code.split('_', 1)[0]
+            if f'{parent_module_code}:*' in permission_codes:
+                return True
+    return False
+
+
+class RbacService(object):
+    @staticmethod
+    def create(session, model_cls, add_info):
+        return RbacDao.create(session, model_cls, add_info)
+
+    @staticmethod
+    def update_by_id(session, model_cls, obj_id, update_info, soft_delete=True):
+        return RbacDao.update_by_id(session, model_cls, obj_id, update_info, soft_delete)
+
+    @staticmethod
+    def get_by_id(session, model_cls, obj_id, soft_delete=True):
+        return RbacDao.get_by_id(session, model_cls, obj_id, soft_delete)
+
+    @staticmethod
+    def list_by_filters(session, model_cls, filter_list, page_num=1, page_size=20, order_column=None):
+        return RbacDao.list_by_filters(session, model_cls, filter_list, int(page_num), int(page_size), order_column)
+
+    @staticmethod
+    def delete_by_id(session, model_cls, obj_id):
+        return RbacDao.delete_by_id(session, model_cls, obj_id)
+
+    @staticmethod
+    def assign_permissions(session, role_ids, permission_id):
+        return RbacDao.assign_permissions_to_roles(session, role_ids, permission_id)
+
+    @staticmethod
+    def assign_menus(session, role_id, menu_ids):
+        return RbacDao.replace_role_menus(session, role_id, menu_ids)
+
+    @staticmethod
+    def get_role_permission_ids(session, role_id):
+        return RbacDao.get_role_permission_ids(session, role_id)
+
+    @staticmethod
+    def get_role_menu_ids(session, role_id):
+        return RbacDao.get_role_menu_ids(session, role_id)
+
+    @staticmethod
+    def build_menu_tree(session, filters, role_ids=None, menu_ids=None):
+        items = RbacDao.get_menu_tree_items(session, filters)
+        visible_ids = set()
+        if not role_ids and not menu_ids:
+            visible_ids = {item.id for item in items}
+        else:
+            role_menu_ids = set(menu_ids or [])
+            if role_ids:
+                for role_id in role_ids:
+                    role_menu_ids.update(RbacDao.get_role_menu_ids(session, role_id))
+            visible_ids = set(role_menu_ids)
+            item_by_id = {item.id: item for item in items}
+            for item_id in list(visible_ids):
+                if item_id not in item_by_id:
+                    continue
+                parent_id = item_by_id[item_id].parent_id
+                while parent_id and parent_id in item_by_id:
+                    if parent_id in visible_ids:
+                        break
+                    visible_ids.add(parent_id)
+                    parent_id = item_by_id[parent_id].parent_id
+        item_map = {}
+        roots = []
+        for item in items:
+            if item.id not in visible_ids:
+                continue
+            item_dict = item.to_dict()
+            item_dict['children'] = []
+            item_map[item.id] = item_dict
+        for item in items:
+            if item.id not in item_map:
+                continue
+            if item.parent_id and item.parent_id in item_map:
+                item_map[item.parent_id]['children'].append(item_map[item.id])
+            else:
+                roots.append(item_map[item.id])
+        return roots
+
+    @staticmethod
+    def get_role_permission_codes(session, role_ids):
+        return RbacDao.get_role_permission_codes(session, role_ids)