187 lines
7.3 KiB
Python
187 lines
7.3 KiB
Python
from datetime import datetime
|
|
from typing import Annotated
|
|
|
|
from fastapi import Path, Query, Request, Response
|
|
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, PageResponseModel, ResponseBaseModel
|
|
from module_admin.entity.vo.user_vo import CurrentUserModel
|
|
from module_business.dependency.test_batch_dependency import TestBatchScopeDependency
|
|
from module_business.domain.test_batch_scope import TestBatchScope
|
|
from module_business.entity.vo.master_vo import (
|
|
BusinessUserOptionModel,
|
|
DeleteWarehouseModel,
|
|
UserAssignModel,
|
|
WarehouseModel,
|
|
WarehousePageQueryModel,
|
|
)
|
|
from module_business.service.master_service import MasterService
|
|
from module_business.service.pallet_service import PalletService
|
|
from utils.response_util import ResponseUtil
|
|
|
|
ASSIGN_LOG = Log
|
|
ASSIGN_CURRENT_USER = CurrentUserDependency
|
|
|
|
warehouse_controller = APIRouterPro(
|
|
prefix='/master/warehouse', tags=['基础资料-仓库管理'], dependencies=[PreAuthDependency()]
|
|
)
|
|
warehouse_user_controller = APIRouterPro(
|
|
prefix='/master/warehouse', tags=['基础资料-仓库用户分配'], dependencies=[PreAuthDependency()]
|
|
)
|
|
|
|
|
|
@warehouse_user_controller.get(
|
|
'/{warehouse_id}/users',
|
|
summary='获取仓库可分配用户接口',
|
|
description='获取启用且未删除的用户及仓库分配状态',
|
|
response_model=DataResponseModel[list[BusinessUserOptionModel]],
|
|
dependencies=[UserInterfaceAuthDependency('master:warehouse:assign')],
|
|
)
|
|
async def get_master_warehouse_users(
|
|
request: Request,
|
|
warehouse_id: Annotated[int, Path(gt=0, description='仓库ID')],
|
|
query_db: Annotated[AsyncSession, DBSessionDependency()],
|
|
scope: Annotated[TestBatchScope, TestBatchScopeDependency()],
|
|
) -> Response:
|
|
users = await PalletService.list_warehouse_users(query_db, warehouse_id, scope)
|
|
return ResponseUtil.success(data=users)
|
|
|
|
|
|
@warehouse_user_controller.put(
|
|
'/{warehouse_id}/users',
|
|
summary='分配仓库用户接口',
|
|
description='完整替换仓库的用户关联',
|
|
response_model=ResponseBaseModel,
|
|
dependencies=[UserInterfaceAuthDependency('master:warehouse:assign')],
|
|
)
|
|
@ASSIGN_LOG(title='仓库用户分配', business_type=BusinessType.GRANT)
|
|
async def assign_master_warehouse_users(
|
|
request: Request,
|
|
warehouse_id: Annotated[int, Path(gt=0, description='仓库ID')],
|
|
assignment: UserAssignModel,
|
|
query_db: Annotated[AsyncSession, DBSessionDependency()],
|
|
current_user: Annotated[CurrentUserModel, ASSIGN_CURRENT_USER()],
|
|
scope: Annotated[TestBatchScope, TestBatchScopeDependency(write=True)],
|
|
) -> Response:
|
|
result = await PalletService.assign_warehouse_users(query_db, warehouse_id, assignment, current_user, scope)
|
|
return ResponseUtil.success(msg=result.message)
|
|
|
|
|
|
@warehouse_controller.get(
|
|
'/list',
|
|
summary='获取仓库分页列表接口',
|
|
description='用于获取仓库分页列表',
|
|
response_model=PageResponseModel[WarehouseModel],
|
|
dependencies=[UserInterfaceAuthDependency('master:warehouse:list')],
|
|
)
|
|
async def get_master_warehouse_list(
|
|
request: Request,
|
|
warehouse_page_query: Annotated[WarehousePageQueryModel, Query()],
|
|
query_db: Annotated[AsyncSession, DBSessionDependency()],
|
|
scope: Annotated[TestBatchScope, TestBatchScopeDependency()],
|
|
) -> Response:
|
|
warehouse_page_query_result = await MasterService.list_warehouse(
|
|
query_db,
|
|
warehouse_page_query,
|
|
scope,
|
|
is_page=True,
|
|
)
|
|
|
|
return ResponseUtil.success(model_content=warehouse_page_query_result)
|
|
|
|
|
|
@warehouse_controller.get(
|
|
'/{warehouse_id}',
|
|
summary='获取仓库详情接口',
|
|
description='用于获取指定仓库的详细信息',
|
|
response_model=DataResponseModel[WarehouseModel],
|
|
dependencies=[UserInterfaceAuthDependency('master:warehouse:query')],
|
|
)
|
|
async def query_detail_master_warehouse(
|
|
request: Request,
|
|
warehouse_id: Annotated[int, Path(gt=0, description='仓库ID')],
|
|
query_db: Annotated[AsyncSession, DBSessionDependency()],
|
|
scope: Annotated[TestBatchScope, TestBatchScopeDependency()],
|
|
) -> Response:
|
|
warehouse_detail_result = await MasterService.detail_warehouse(query_db, warehouse_id, scope)
|
|
|
|
return ResponseUtil.success(data=warehouse_detail_result)
|
|
|
|
|
|
@warehouse_controller.post(
|
|
'',
|
|
summary='新增仓库接口',
|
|
description='用于新增仓库',
|
|
response_model=ResponseBaseModel,
|
|
dependencies=[UserInterfaceAuthDependency('master:warehouse:add')],
|
|
)
|
|
@Log(title='仓库管理', business_type=BusinessType.INSERT)
|
|
async def add_master_warehouse(
|
|
request: Request,
|
|
add_warehouse: WarehouseModel,
|
|
query_db: Annotated[AsyncSession, DBSessionDependency()],
|
|
current_user: Annotated[CurrentUserModel, CurrentUserDependency()],
|
|
scope: Annotated[TestBatchScope, TestBatchScopeDependency(write=True)],
|
|
) -> Response:
|
|
current_time = datetime.now()
|
|
add_warehouse.create_by = current_user.user.user_name
|
|
add_warehouse.create_time = current_time
|
|
add_warehouse.update_by = current_user.user.user_name
|
|
add_warehouse.update_time = current_time
|
|
add_warehouse_result = await MasterService.add_warehouse(query_db, add_warehouse, current_user, scope)
|
|
|
|
return ResponseUtil.success(msg=add_warehouse_result.message)
|
|
|
|
|
|
@warehouse_controller.put(
|
|
'',
|
|
summary='编辑仓库接口',
|
|
description='用于编辑仓库',
|
|
response_model=ResponseBaseModel,
|
|
dependencies=[UserInterfaceAuthDependency('master:warehouse:edit')],
|
|
)
|
|
@Log(title='仓库管理', business_type=BusinessType.UPDATE)
|
|
async def edit_master_warehouse(
|
|
request: Request,
|
|
edit_warehouse: WarehouseModel,
|
|
query_db: Annotated[AsyncSession, DBSessionDependency()],
|
|
current_user: Annotated[CurrentUserModel, CurrentUserDependency()],
|
|
scope: Annotated[TestBatchScope, TestBatchScopeDependency(write=True)],
|
|
) -> Response:
|
|
edit_warehouse.update_by = current_user.user.user_name
|
|
edit_warehouse.update_time = datetime.now()
|
|
edit_warehouse_result = await MasterService.edit_warehouse(query_db, edit_warehouse, current_user, scope)
|
|
|
|
return ResponseUtil.success(msg=edit_warehouse_result.message)
|
|
|
|
|
|
@warehouse_controller.delete(
|
|
'/{warehouse_ids}',
|
|
summary='删除仓库接口',
|
|
description='用于删除仓库',
|
|
response_model=ResponseBaseModel,
|
|
dependencies=[UserInterfaceAuthDependency('master:warehouse:remove')],
|
|
)
|
|
@Log(title='仓库管理', business_type=BusinessType.DELETE)
|
|
async def delete_master_warehouse(
|
|
request: Request,
|
|
warehouse_ids: Annotated[
|
|
str,
|
|
Path(pattern=r'^[1-9]\d*(,[1-9]\d*)*$', description='需要删除的仓库ID'),
|
|
],
|
|
query_db: Annotated[AsyncSession, DBSessionDependency()],
|
|
current_user: Annotated[CurrentUserModel, CurrentUserDependency()],
|
|
) -> Response:
|
|
delete_warehouse = DeleteWarehouseModel(
|
|
warehouseIds=[int(warehouse_id) for warehouse_id in warehouse_ids.split(',')]
|
|
)
|
|
delete_warehouse_result = await MasterService.remove_warehouse(query_db, delete_warehouse)
|
|
|
|
return ResponseUtil.success(msg=delete_warehouse_result.message)
|