功能：新增托盘二维码与用户分配

shierfeng-fastapi-backend/module_business/controller/warehouse_controller.py

@@ -12,13 +12,61 @@ 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.entity.vo.master_vo import DeleteWarehouseModel, WarehouseModel, WarehousePageQueryModel
+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()],
+) -> Response:
+    users = await PalletService.list_warehouse_users(query_db, warehouse_id)
+    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()],
+) -> Response:
+    result = await PalletService.assign_warehouse_users(query_db, warehouse_id, assignment)
+    return ResponseUtil.success(msg=result.message)
 
 
 @warehouse_controller.get(