feat: 初始化项目架构

ruoyi-fastapi-backend/module_admin/dao/menu_dao.py

@@ -0,0 +1,142 @@
+from sqlalchemy import and_
+from sqlalchemy.orm import Session
+from module_admin.entity.do.menu_do import SysMenu
+from module_admin.entity.do.user_do import SysUser, SysUserRole
+from module_admin.entity.do.role_do import SysRole, SysRoleMenu
+from module_admin.entity.vo.menu_vo import *
+
+
+class MenuDao:
+    """
+    菜单管理模块数据库操作层
+    """
+
+    @classmethod
+    def get_menu_detail_by_id(cls, db: Session, menu_id: int):
+        """
+        根据菜单id获取菜单详细信息
+        :param db: orm对象
+        :param menu_id: 菜单id
+        :return: 菜单信息对象
+        """
+        menu_info = db.query(SysMenu) \
+            .filter(SysMenu.menu_id == menu_id) \
+            .first()
+
+        return menu_info
+
+    @classmethod
+    def get_menu_detail_by_info(cls, db: Session, menu: MenuModel):
+        """
+        根据菜单参数获取菜单信息
+        :param db: orm对象
+        :param menu: 菜单参数对象
+        :return: 菜单信息对象
+        """
+        menu_info = db.query(SysMenu) \
+            .filter(SysMenu.parent_id == menu.parent_id if menu.parent_id else True,
+                    SysMenu.menu_name == menu.menu_name if menu.menu_name else True,
+                    SysMenu.menu_type == menu.menu_type if menu.menu_type else True) \
+            .first()
+
+        return menu_info
+
+    @classmethod
+    def get_menu_list_for_tree(cls, db: Session, user_id: int, role: list):
+        """
+        根据角色信息获取所有在用菜单列表信息
+        :param db: orm对象
+        :param user_id: 用户id
+        :param role: 用户角色列表信息
+        :return: 菜单列表信息
+        """
+        role_id_list = [item.role_id for item in role]
+        if 1 in role_id_list:
+            menu_query_all = db.query(SysMenu) \
+                .filter(SysMenu.status == 0) \
+                .order_by(SysMenu.order_num) \
+                .distinct().all()
+        else:
+            menu_query_all = db.query(SysMenu).select_from(SysUser) \
+                .filter(SysUser.status == 0, SysUser.del_flag == 0, SysUser.user_id == user_id) \
+                .outerjoin(SysUserRole, SysUser.user_id == SysUserRole.user_id) \
+                .outerjoin(SysRole,
+                           and_(SysUserRole.role_id == SysRole.role_id, SysRole.status == 0, SysRole.del_flag == 0)) \
+                .outerjoin(SysRoleMenu, SysRole.role_id == SysRoleMenu.role_id) \
+                .join(SysMenu, and_(SysRoleMenu.menu_id == SysMenu.menu_id, SysMenu.status == 0,)) \
+                .order_by(SysMenu.order_num) \
+                .distinct().all()
+
+        return menu_query_all
+
+    @classmethod
+    def get_menu_list(cls, db: Session, page_object: MenuQueryModel, user_id: int, role: list):
+        """
+        根据查询参数获取菜单列表信息
+        :param db: orm对象
+        :param page_object: 不分页查询参数对象
+        :param user_id: 用户id
+        :param role: 用户角色列表
+        :return: 菜单列表信息对象
+        """
+        role_id_list = [item.role_id for item in role]
+        if 1 in role_id_list:
+            menu_query_all = db.query(SysMenu) \
+                .filter(SysMenu.status == page_object.status if page_object.status else True,
+                        SysMenu.menu_name.like(
+                            f'%{page_object.menu_name}%') if page_object.menu_name else True) \
+                .order_by(SysMenu.order_num) \
+                .distinct().all()
+        else:
+            menu_query_all = db.query(SysMenu).select_from(SysUser) \
+                .filter(SysUser.status == 0, SysUser.del_flag == 0, SysUser.user_id == user_id) \
+                .outerjoin(SysUserRole, SysUser.user_id == SysUserRole.user_id) \
+                .outerjoin(SysRole,
+                           and_(SysUserRole.role_id == SysRole.role_id, SysRole.status == 0, SysRole.del_flag == 0)) \
+                .outerjoin(SysRoleMenu, SysRole.role_id == SysRoleMenu.role_id) \
+                .join(SysMenu, and_(SysRoleMenu.menu_id == SysMenu.menu_id,
+                                    SysMenu.status == page_object.status if page_object.status else True,
+                                    SysMenu.menu_name.like(
+                                        f'%{page_object.menu_name}%') if page_object.menu_name else True)) \
+                .order_by(SysMenu.order_num) \
+                .distinct().all()
+
+        return menu_query_all
+
+    @classmethod
+    def add_menu_dao(cls, db: Session, menu: MenuModel):
+        """
+        新增菜单数据库操作
+        :param db: orm对象
+        :param menu: 菜单对象
+        :return:
+        """
+        db_menu = SysMenu(**menu.model_dump())
+        db.add(db_menu)
+        db.flush()
+
+        return db_menu
+
+    @classmethod
+    def edit_menu_dao(cls, db: Session, menu: dict):
+        """
+        编辑菜单数据库操作
+        :param db: orm对象
+        :param menu: 需要更新的菜单字典
+        :return:
+        """
+        db.query(SysMenu) \
+            .filter(SysMenu.menu_id == menu.get('menu_id')) \
+            .update(menu)
+
+    @classmethod
+    def delete_menu_dao(cls, db: Session, menu: MenuModel):
+        """
+        删除菜单数据库操作
+        :param db: orm对象
+        :param menu: 菜单对象
+        :return:
+        """
+        db.query(SysMenu) \
+            .filter(SysMenu.menu_id == menu.menu_id) \
+            .delete()