""" 资产调拨业务服务层 """ from typing import List, Optional, Dict, Any from sqlalchemy.orm import Session, selectinload from app.crud.transfer import transfer_order, transfer_item from app.crud.asset import asset from app.schemas.transfer import ( AssetTransferOrderCreate, AssetTransferOrderUpdate ) from app.core.exceptions import NotFoundException, BusinessException class TransferService: """资产调拨服务类""" async def get_order( self, db: Session, order_id: int ) -> Dict[str, Any]: """获取调拨单详情""" # 使用selectinload预加载关联数据，避免N+1查询 from app.models.transfer import AssetTransferOrder from app.models.organization import Organization from app.models.user import User from app.models.transfer import AssetTransferItem obj = db.query( AssetTransferOrder ).options( selectinload(AssetTransferOrder.items), selectinload(AssetTransferOrder.source_org.of_type(Organization)), selectinload(AssetTransferOrder.target_org.of_type(Organization)), selectinload(AssetTransferOrder.applicant.of_type(User)), selectinload(AssetTransferOrder.approver.of_type(User)), selectinload(AssetTransferOrder.executor.of_type(User)) ).filter( AssetTransferOrder.id == order_id ).first() if not obj: raise NotFoundException("调拨单") # 加载关联信息 return self._load_order_relations(db, obj) def get_orders( self, db: Session, skip: int = 0, limit: int = 20, transfer_type: Optional[str] = None, approval_status: Optional[str] = None, execute_status: Optional[str] = None, source_org_id: Optional[int] = None, target_org_id: Optional[int] = None, keyword: Optional[str] = None ) -> tuple: """获取调拨单列表""" items, total = transfer_order.get_multi( db=db, skip=skip, limit=limit, transfer_type=transfer_type, approval_status=approval_status, execute_status=execute_status, source_org_id=source_org_id, target_org_id=target_org_id, keyword=keyword ) # 加载关联信息 items_with_relations = [self._load_order_relations(db, item) for item in items] return items_with_relations, total async def create_order( self, db: Session, obj_in: AssetTransferOrderCreate, apply_user_id: int ): """创建调拨单""" # 验证资产存在性和状态 assets = [] for asset_id in obj_in.asset_ids: asset_obj = asset.get(db, asset_id) if not asset_obj: raise NotFoundException(f"资产ID {asset_id}") assets.append(asset_obj) # 验证资产状态是否允许调拨 for asset_obj in assets: if asset_obj.status not in ["in_stock", "in_use"]: raise BusinessException( f"资产 {asset_obj.asset_code} 当前状态为 {asset_obj.status}，不允许调拨操作" ) # 验证资产所属机构是否为调出机构 for asset_obj in assets: if asset_obj.organization_id != obj_in.source_org_id: raise BusinessException( f"资产 {asset_obj.asset_code} 所属机构与调出机构不一致" ) # 生成调拨单号 order_code = await self._generate_order_code(db) # 创建调拨单 db_obj = transfer_order.create( db=db, obj_in=obj_in, order_code=order_code, apply_user_id=apply_user_id ) return self._load_order_relations(db, db_obj) def update_order( self, db: Session, order_id: int, obj_in: AssetTransferOrderUpdate ): """更新调拨单""" db_obj = transfer_order.get(db, order_id) if not db_obj: raise NotFoundException("调拨单") # 只有待审批状态可以更新 if db_obj.approval_status != "pending": raise BusinessException("只有待审批状态的调拨单可以更新") return transfer_order.update(db, db_obj, obj_in) def approve_order( self, db: Session, order_id: int, approval_status: str, approval_user_id: int, approval_remark: Optional[str] = None ): """审批调拨单""" db_obj = transfer_order.get(db, order_id) if not db_obj: raise NotFoundException("调拨单") # 检查状态 if db_obj.approval_status != "pending": raise BusinessException("该调拨单已审批，无法重复审批") # 审批 db_obj = transfer_order.approve( db=db, db_obj=db_obj, approval_status=approval_status, approval_user_id=approval_user_id, approval_remark=approval_remark ) return self._load_order_relations(db, db_obj) def start_order( self, db: Session, order_id: int, execute_user_id: int ): """开始调拨""" db_obj = transfer_order.get(db, order_id) if not db_obj: raise NotFoundException("调拨单") # 检查状态 if db_obj.approval_status != "approved": raise BusinessException("该调拨单未审批通过，无法开始执行") if db_obj.execute_status != "pending": raise BusinessException("该调拨单已开始或已完成") # 开始调拨 db_obj = transfer_order.start(db, db_obj, execute_user_id) # 更新明细状态为调拨中 transfer_item.batch_update_transfer_status(db, order_id, "transferring") return self._load_order_relations(db, db_obj) async def complete_order( self, db: Session, order_id: int, execute_user_id: int ): """完成调拨""" db_obj = transfer_order.get(db, order_id) if not db_obj: raise NotFoundException("调拨单") # 检查状态 if db_obj.execute_status not in ["pending", "executing"]: raise BusinessException("该调拨单状态不允许完成操作") # 完成调拨单 db_obj = transfer_order.complete(db, db_obj, execute_user_id) # 更新资产机构和状态 await self._execute_transfer_logic(db, db_obj) # 更新明细状态为完成 transfer_item.batch_update_transfer_status(db, order_id, "completed") return self._load_order_relations(db, db_obj) def cancel_order( self, db: Session, order_id: int ) -> bool: """取消调拨单""" db_obj = transfer_order.get(db, order_id) if not db_obj: raise NotFoundException("调拨单") # 检查状态 if db_obj.execute_status == "completed": raise BusinessException("已完成的调拨单无法取消") transfer_order.cancel(db, db_obj) return True def delete_order( self, db: Session, order_id: int ) -> bool: """删除调拨单""" db_obj = transfer_order.get(db, order_id) if not db_obj: raise NotFoundException("调拨单") # 只有已取消或已拒绝的可以删除 if db_obj.approval_status not in ["rejected", "cancelled"]: raise BusinessException("只能删除已拒绝或已取消的调拨单") return transfer_order.delete(db, order_id) def get_order_items( self, db: Session, order_id: int ) -> List: """获取调拨单明细""" # 验证调拨单存在 if not transfer_order.get(db, order_id): raise NotFoundException("调拨单") return transfer_item.get_by_order(db, order_id) def get_statistics( self, db: Session, source_org_id: Optional[int] = None, target_org_id: Optional[int] = None ) -> Dict[str, int]: """获取调拨单统计信息""" return transfer_order.get_statistics(db, source_org_id, target_org_id) async def _execute_transfer_logic( self, db: Session, order_obj ): """执行调拨逻辑（完成调拨时自动执行）""" # 获取明细 items = transfer_item.get_by_order(db, order_obj.id) # 更新资产机构和状态 from app.services.asset_service import asset_service from app.schemas.asset import AssetStatusTransition, AssetUpdate for item in items: try: # 变更资产状态 await asset_service.change_asset_status( db=db, asset_id=item.asset_id, status_transition=AssetStatusTransition( new_status="transferring", remark=f"调拨单: {order_obj.order_code}，从{item.source_organization_id}到{item.target_organization_id}" ), operator_id=order_obj.execute_user_id ) # 更新资产所属机构 asset_obj = asset.get(db, item.asset_id) if asset_obj: asset.update( db=db, db_obj=asset_obj, obj_in=AssetUpdate( organization_id=item.target_organization_id ), updater_id=order_obj.execute_user_id ) # 最终状态变更 target_status = "in_stock" await asset_service.change_asset_status( db=db, asset_id=item.asset_id, status_transition=AssetStatusTransition( new_status=target_status, remark=f"调拨完成: {order_obj.order_code}" ), operator_id=order_obj.execute_user_id ) except Exception as e: # 记录失败日志 print(f"调拨资产 {item.asset_code} 失败: {str(e)}") raise def _load_order_relations( self, db: Session, obj ) -> Dict[str, Any]: """加载调拨单关联信息""" from app.models.user import User from app.models.organization import Organization result = { "id": obj.id, "order_code": obj.order_code, "source_org_id": obj.source_org_id, "target_org_id": obj.target_org_id, "transfer_type": obj.transfer_type, "title": obj.title, "asset_count": obj.asset_count, "apply_user_id": obj.apply_user_id, "apply_time": obj.apply_time, "approval_status": obj.approval_status, "approval_user_id": obj.approval_user_id, "approval_time": obj.approval_time, "approval_remark": obj.approval_remark, "execute_status": obj.execute_status, "execute_user_id": obj.execute_user_id, "execute_time": obj.execute_time, "remark": obj.remark, "created_at": obj.created_at, "updated_at": obj.updated_at } # 加载调出机构 if obj.source_org_id: source_org = db.query(Organization).filter( Organization.id == obj.source_org_id ).first() if source_org: result["source_organization"] = { "id": source_org.id, "org_name": source_org.org_name, "org_type": source_org.org_type } # 加载调入机构 if obj.target_org_id: target_org = db.query(Organization).filter( Organization.id == obj.target_org_id ).first() if target_org: result["target_organization"] = { "id": target_org.id, "org_name": target_org.org_name, "org_type": target_org.org_type } # 加载申请人 if obj.apply_user_id: apply_user = db.query(User).filter(User.id == obj.apply_user_id).first() if apply_user: result["apply_user"] = { "id": apply_user.id, "real_name": apply_user.real_name, "username": apply_user.username } # 加载审批人 if obj.approval_user_id: approval_user = db.query(User).filter(User.id == obj.approval_user_id).first() if approval_user: result["approval_user"] = { "id": approval_user.id, "real_name": approval_user.real_name, "username": approval_user.username } # 加载执行人 if obj.execute_user_id: execute_user = db.query(User).filter(User.id == obj.execute_user_id).first() if execute_user: result["execute_user"] = { "id": execute_user.id, "real_name": execute_user.real_name, "username": execute_user.username } # 加载明细 items = transfer_item.get_by_order(db, obj.id) result["items"] = [ { "id": item.id, "asset_id": item.asset_id, "asset_code": item.asset_code, "source_organization_id": item.source_organization_id, "target_organization_id": item.target_organization_id, "transfer_status": item.transfer_status } for item in items ] # Frontend-friendly aliases result["transfer_no"] = obj.order_code result["status"] = obj.approval_status result["reason"] = obj.title result["applicant"] = result.get("apply_user") if "source_organization" in result: result["source_org"] = result["source_organization"] if "target_organization" in result: result["target_org"] = result["target_organization"] # Asset details and total value (best effort) asset_ids = [item.asset_id for item in items] if asset_ids: from app.models.asset import Asset assets = db.query(Asset).filter(Asset.id.in_(asset_ids)).all() assets_detail = [] total_value = 0 for asset_obj in assets: price = asset_obj.purchase_price or 0 total_value += float(price) assets_detail.append({ "id": asset_obj.id, "asset_code": asset_obj.asset_code, "asset_name": asset_obj.asset_name, "model": asset_obj.model, "serial_number": asset_obj.serial_number, "purchase_price": float(price) if asset_obj.purchase_price is not None else None, "device_type": { "type_name": asset_obj.device_type.type_name } if asset_obj.device_type else None, "brand": { "brand_name": asset_obj.brand.brand_name } if asset_obj.brand else None, }) result["assets"] = assets_detail result["total_value"] = total_value return result async def _generate_order_code(self, db: Session) -> str: """生成调拨单号""" from datetime import datetime import random import string # 日期部分 date_str = datetime.now().strftime("%Y%m%d") # 序号部分（5位随机数） sequence = "".join(random.choices(string.digits, k=5)) # 组合单号: TO-20250124-00001 order_code = f"TO-{date_str}-{sequence}" # 检查是否重复，如果重复则重新生成 while transfer_order.get_by_code(db, order_code): sequence = "".join(random.choices(string.digits, k=5)) order_code = f"TO-{date_str}-{sequence}" return order_code # 创建全局实例 transfer_service = TransferService()