File size: 6,435 Bytes
96cbd66
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
from datetime import datetime
from typing import Annotated

from fastapi import Path, Query, Request, Response
from pydantic_validation_decorator import ValidateFields
from sqlalchemy.ext.asyncio import AsyncSession

from common.annotation.log_annotation import Log
from common.aspect.db_seesion import DBSessionDependency
from common.aspect.interface_auth import UserInterfaceAuthDependency
from common.aspect.pre_auth import CurrentUserDependency, PreAuthDependency
from common.enums import BusinessType
from common.router import APIRouterPro
from common.vo import DataResponseModel, DynamicResponseModel, ResponseBaseModel
from module_admin.entity.vo.menu_vo import DeleteMenuModel, MenuModel, MenuQueryModel, MenuTreeModel
from module_admin.entity.vo.role_vo import RoleMenuQueryModel
from module_admin.entity.vo.user_vo import CurrentUserModel
from module_admin.service.menu_service import MenuService
from utils.log_util import logger
from utils.response_util import ResponseUtil

menu_controller = APIRouterPro(
    prefix='/system/menu', order_num=5, tags=['系统管理-菜单管理'], dependencies=[PreAuthDependency()]
)


@menu_controller.get(

    '/treeselect',

    summary='获取菜单树接口',

    description='用于获取当前用户可见的菜单树',

    response_model=DataResponseModel[list[MenuTreeModel]],

)
async def get_system_menu_tree(

    request: Request,

    query_db: Annotated[AsyncSession, DBSessionDependency()],

    current_user: Annotated[CurrentUserModel, CurrentUserDependency()],

) -> Response:
    menu_query_result = await MenuService.get_menu_tree_services(query_db, current_user)
    logger.info('获取成功')

    return ResponseUtil.success(data=menu_query_result)


@menu_controller.get(

    '/roleMenuTreeselect/{role_id}',

    summary='获取角色菜单树接口',

    description='用于获取指定角色可见的菜单树',

    response_model=DynamicResponseModel[RoleMenuQueryModel],

)
async def get_system_role_menu_tree(

    request: Request,

    role_id: Annotated[int, Path(description='角色ID')],

    query_db: Annotated[AsyncSession, DBSessionDependency()],

    current_user: Annotated[CurrentUserModel, CurrentUserDependency()],

) -> Response:
    role_menu_query_result = await MenuService.get_role_menu_tree_services(query_db, role_id, current_user)
    logger.info('获取成功')

    return ResponseUtil.success(model_content=role_menu_query_result)


@menu_controller.get(

    '/list',

    summary='获取菜单列表接口',

    description='用于获取当前用户可见的菜单列表',

    response_model=DataResponseModel[list[MenuModel]],

    dependencies=[UserInterfaceAuthDependency('system:menu:list')],

)
async def get_system_menu_list(

    request: Request,

    menu_query: Annotated[MenuQueryModel, Query()],

    query_db: Annotated[AsyncSession, DBSessionDependency()],

    current_user: Annotated[CurrentUserModel, CurrentUserDependency()],

) -> Response:
    menu_query_result = await MenuService.get_menu_list_services(query_db, menu_query, current_user)
    logger.info('获取成功')

    return ResponseUtil.success(data=menu_query_result)


@menu_controller.post(

    '',

    summary='新增菜单接口',

    description='用于新增菜单',

    response_model=ResponseBaseModel,

    dependencies=[UserInterfaceAuthDependency('system:menu:add')],

)
@ValidateFields(validate_model='add_menu')
@Log(title='菜单管理', business_type=BusinessType.INSERT)
async def add_system_menu(

    request: Request,

    add_menu: MenuModel,

    query_db: Annotated[AsyncSession, DBSessionDependency()],

    current_user: Annotated[CurrentUserModel, CurrentUserDependency()],

) -> Response:
    add_menu.create_by = current_user.user.user_name
    add_menu.create_time = datetime.now()
    add_menu.update_by = current_user.user.user_name
    add_menu.update_time = datetime.now()
    add_menu_result = await MenuService.add_menu_services(query_db, add_menu)
    logger.info(add_menu_result.message)

    return ResponseUtil.success(msg=add_menu_result.message)


@menu_controller.put(

    '',

    summary='编辑菜单接口',

    description='用于编辑菜单',

    response_model=ResponseBaseModel,

    dependencies=[UserInterfaceAuthDependency('system:menu:edit')],

)
@ValidateFields(validate_model='edit_menu')
@Log(title='菜单管理', business_type=BusinessType.UPDATE)
async def edit_system_menu(

    request: Request,

    edit_menu: MenuModel,

    query_db: Annotated[AsyncSession, DBSessionDependency()],

    current_user: Annotated[CurrentUserModel, CurrentUserDependency()],

) -> Response:
    edit_menu.update_by = current_user.user.user_name
    edit_menu.update_time = datetime.now()
    edit_menu_result = await MenuService.edit_menu_services(query_db, edit_menu)
    logger.info(edit_menu_result.message)

    return ResponseUtil.success(msg=edit_menu_result.message)


@menu_controller.delete(

    '/{menu_ids}',

    summary='删除菜单接口',

    description='用于删除菜单',

    response_model=ResponseBaseModel,

    dependencies=[UserInterfaceAuthDependency('system:menu:remove')],

)
@Log(title='菜单管理', business_type=BusinessType.DELETE)
async def delete_system_menu(

    request: Request,

    menu_ids: Annotated[str, Path(description='需要删除的菜单ID')],

    query_db: Annotated[AsyncSession, DBSessionDependency()],

) -> Response:
    delete_menu = DeleteMenuModel(menuIds=menu_ids)
    delete_menu_result = await MenuService.delete_menu_services(query_db, delete_menu)
    logger.info(delete_menu_result.message)

    return ResponseUtil.success(msg=delete_menu_result.message)


@menu_controller.get(

    '/{menu_id}',

    summary='获取菜单详情接口',

    description='用于获取指定菜单的详情信息',

    response_model=DataResponseModel[MenuModel],

    dependencies=[UserInterfaceAuthDependency('system:menu:query')],

)
async def query_detail_system_menu(

    request: Request,

    menu_id: Annotated[int, Path(description='菜单ID')],

    query_db: Annotated[AsyncSession, DBSessionDependency()],

) -> Response:
    menu_detail_result = await MenuService.menu_detail_services(query_db, menu_id)
    logger.info(f'获取menu_id为{menu_id}的信息成功')

    return ResponseUtil.success(data=menu_detail_result)