zcglxt/backend_new/app/services/maintenance_service.py

"""
维修管理业务服务层
"""
from typing import List, Optional, Dict, Any
from sqlalchemy.orm import Session, selectinload
from app.crud.maintenance import maintenance_record
from app.crud.asset import asset
from app.schemas.maintenance import (
    MaintenanceRecordCreate,
    MaintenanceRecordUpdate,
    MaintenanceRecordStart,
    MaintenanceRecordComplete
)
from app.core.exceptions import NotFoundException, BusinessException


class MaintenanceService:
    """维修管理服务类"""

    async def get_record(
        self,
        db: Session,
        record_id: int
    ) -> Dict[str, Any]:
        """获取维修记录详情"""
        # 使用selectinload预加载关联数据，避免N+1查询
        from app.models.maintenance import MaintenanceRecord
        from app.models.asset import Asset
        from app.models.user import User
        from app.models.brand_supplier import Supplier

        obj = db.query(
            MaintenanceRecord
        ).options(
            selectinload(MaintenanceRecord.asset.of_type(Asset)),
            selectinload(MaintenanceRecord.report_user.of_type(User)),
            selectinload(MaintenanceRecord.maintenance_user.of_type(User)),
            selectinload(MaintenanceRecord.vendor.of_type(Supplier))
        ).filter(
            MaintenanceRecord.id == record_id
        ).first()

        if not obj:
            raise NotFoundException("维修记录")

        return self._load_relations(db, obj)

    def get_records(
        self,
        db: Session,
        skip: int = 0,
        limit: int = 20,
        asset_id: Optional[int] = None,
        status: Optional[str] = None,
        fault_type: Optional[str] = None,
        priority: Optional[str] = None,
        maintenance_type: Optional[str] = None,
        keyword: Optional[str] = None
    ) -> tuple:
        """获取维修记录列表"""
        items, total = maintenance_record.get_multi(
            db=db,
            skip=skip,
            limit=limit,
            asset_id=asset_id,
            status=status,
            fault_type=fault_type,
            priority=priority,
            maintenance_type=maintenance_type,
            keyword=keyword
        )

        # 加载关联信息
        items_with_relations = [self._load_relations(db, item) for item in items]

        return items_with_relations, total

    async def create_record(
        self,
        db: Session,
        obj_in: MaintenanceRecordCreate,
        report_user_id: int,
        creator_id: int
    ):
        """创建维修记录"""
        # 验证资产存在
        asset_obj = asset.get(db, obj_in.asset_id)
        if not asset_obj:
            raise NotFoundException("资产")

        # 生成维修单号
        record_code = await self._generate_record_code(db)

        # 创建维修记录
        db_obj = maintenance_record.create(
            db=db,
            obj_in=obj_in,
            record_code=record_code,
            asset_code=asset_obj.asset_code,
            report_user_id=report_user_id,
            creator_id=creator_id
        )

        # 如果资产状态不是维修中，则更新状态
        if asset_obj.status != "maintenance":
            from app.services.asset_service import asset_service
            from app.schemas.asset import AssetStatusTransition

            try:
                await asset_service.change_asset_status(
                    db=db,
                    asset_id=asset_obj.id,
                    status_transition=AssetStatusTransition(
                        new_status="maintenance",
                        remark=f"报修: {record_code}"
                    ),
                    operator_id=report_user_id
                )
            except Exception as e:
                # 状态更新失败不影响维修记录创建
                pass

        return self._load_relations(db, db_obj)

    def update_record(
        self,
        db: Session,
        record_id: int,
        obj_in: MaintenanceRecordUpdate,
        updater_id: int
    ):
        """更新维修记录"""
        db_obj = maintenance_record.get(db, record_id)
        if not db_obj:
            raise NotFoundException("维修记录")

        # 已完成的维修记录不能更新
        if db_obj.status == "completed":
            raise BusinessException("已完成的维修记录不能更新")

        return maintenance_record.update(db, db_obj, obj_in, updater_id)

    async def start_maintenance(
        self,
        db: Session,
        record_id: int,
        start_in: MaintenanceRecordStart,
        maintenance_user_id: int
    ):
        """开始维修"""
        db_obj = maintenance_record.get(db, record_id)
        if not db_obj:
            raise NotFoundException("维修记录")

        # 检查状态
        if db_obj.status != "pending":
            raise BusinessException("只有待处理状态的维修记录可以开始维修")

        # 验证维修类型
        if start_in.maintenance_type == "vendor_repair" and not start_in.vendor_id:
            raise BusinessException("外部维修必须指定维修供应商")

        # 开始维修
        db_obj = maintenance_record.start_maintenance(
            db=db,
            db_obj=db_obj,
            maintenance_type=start_in.maintenance_type,
            maintenance_user_id=maintenance_user_id,
            vendor_id=start_in.vendor_id
        )

        return self._load_relations(db, db_obj)

    async def complete_maintenance(
        self,
        db: Session,
        record_id: int,
        complete_in: MaintenanceRecordComplete,
        maintenance_user_id: int
    ):
        """完成维修"""
        db_obj = maintenance_record.get(db, record_id)
        if not db_obj:
            raise NotFoundException("维修记录")

        # 检查状态
        if db_obj.status != "in_progress":
            raise BusinessException("只有维修中的记录可以完成")

        # 完成维修
        db_obj = maintenance_record.complete_maintenance(
            db=db,
            db_obj=db_obj,
            maintenance_result=complete_in.maintenance_result,
            maintenance_cost=complete_in.maintenance_cost,
            replaced_parts=complete_in.replaced_parts,
            images=complete_in.images
        )

        # 恢复资产状态
        from app.services.asset_service import asset_service
        from app.schemas.asset import AssetStatusTransition

        try:
            await asset_service.change_asset_status(
                db=db,
                asset_id=db_obj.asset_id,
                status_transition=AssetStatusTransition(
                    new_status=complete_in.asset_status,
                    remark=f"维修完成: {db_obj.record_code}"
                ),
                operator_id=maintenance_user_id
            )
        except Exception as e:
            # 状态更新失败不影响维修记录完成
            pass

        return self._load_relations(db, db_obj)

    def cancel_maintenance(
        self,
        db: Session,
        record_id: int
    ):
        """取消维修"""
        db_obj = maintenance_record.get(db, record_id)
        if not db_obj:
            raise NotFoundException("维修记录")

        # 检查状态
        if db_obj.status == "completed":
            raise BusinessException("已完成的维修记录不能取消")

        # 取消维修
        db_obj = maintenance_record.cancel_maintenance(db, db_obj)

        # 恢复资产状态
        asset_obj = asset.get(db, db_obj.asset_id)
        if asset_obj and asset_obj.status == "maintenance":
            from app.services.asset_service import asset_service
            from app.schemas.asset import AssetStatusTransition

            try:
                # 根据维修前的状态恢复
                target_status = "in_stock"  # 默认恢复为库存中
                asset_service.change_asset_status(
                    db=db,
                    asset_id=asset_obj.id,
                    status_transition=AssetStatusTransition(
                        new_status=target_status,
                        remark=f"取消维修: {db_obj.record_code}"
                    ),
                    operator_id=db_obj.report_user_id or 0
                )
            except Exception as e:
                # 状态更新失败不影响维修记录取消
                pass

        return self._load_relations(db, db_obj)

    def delete_record(
        self,
        db: Session,
        record_id: int
    ) -> bool:
        """删除维修记录"""
        db_obj = maintenance_record.get(db, record_id)
        if not db_obj:
            raise NotFoundException("维修记录")

        # 只能删除待处理或已取消的记录
        if db_obj.status not in ["pending", "cancelled"]:
            raise BusinessException("只能删除待处理或已取消的维修记录")

        return maintenance_record.delete(db, record_id)

    def get_asset_records(
        self,
        db: Session,
        asset_id: int,
        skip: int = 0,
        limit: int = 50
    ) -> List:
        """获取资产的维修记录"""
        # 验证资产存在
        if not asset.get(db, asset_id):
            raise NotFoundException("资产")

        records = maintenance_record.get_by_asset(db, asset_id, skip, limit)
        return [self._load_relations(db, record) for record in records]

    def get_statistics(
        self,
        db: Session,
        asset_id: Optional[int] = None
    ) -> Dict[str, Any]:
        """获取维修统计信息"""
        return maintenance_record.get_statistics(db, asset_id)

    def _load_relations(
        self,
        db: Session,
        obj
    ) -> Dict[str, Any]:
        """加载维修记录关联信息"""
        from app.models.asset import Asset
        from app.models.user import User
        from app.models.brand_supplier import Supplier

        result = {
            "id": obj.id,
            "record_code": obj.record_code,
            "asset_id": obj.asset_id,
            "asset_code": obj.asset_code,
            "fault_description": obj.fault_description,
            "fault_type": obj.fault_type,
            "report_user_id": obj.report_user_id,
            "report_time": obj.report_time,
            "priority": obj.priority,
            "maintenance_type": obj.maintenance_type,
            "vendor_id": obj.vendor_id,
            "maintenance_cost": float(obj.maintenance_cost) if obj.maintenance_cost else None,
            "start_time": obj.start_time,
            "complete_time": obj.complete_time,
            "maintenance_user_id": obj.maintenance_user_id,
            "maintenance_result": obj.maintenance_result,
            "replaced_parts": obj.replaced_parts,
            "status": obj.status,
            "images": obj.images,
            "remark": obj.remark,
            "created_at": obj.created_at,
            "updated_at": obj.updated_at
        }

        # 加载资产信息
        if obj.asset_id:
            asset_obj = db.query(Asset).filter(Asset.id == obj.asset_id).first()
            if asset_obj:
                result["asset"] = {
                    "id": asset_obj.id,
                    "asset_code": asset_obj.asset_code,
                    "asset_name": asset_obj.asset_name,
                    "status": asset_obj.status
                }

        # 加载报修人信息
        if obj.report_user_id:
            report_user = db.query(User).filter(User.id == obj.report_user_id).first()
            if report_user:
                result["report_user"] = {
                    "id": report_user.id,
                    "real_name": report_user.real_name,
                    "username": report_user.username
                }

        # 加载维修人员信息
        if obj.maintenance_user_id:
            maintenance_user = db.query(User).filter(User.id == obj.maintenance_user_id).first()
            if maintenance_user:
                result["maintenance_user"] = {
                    "id": maintenance_user.id,
                    "real_name": maintenance_user.real_name,
                    "username": maintenance_user.username
                }

        # 加载供应商信息
        if obj.vendor_id:
            vendor = db.query(Supplier).filter(Supplier.id == obj.vendor_id).first()
            if vendor:
                result["vendor"] = {
                    "id": vendor.id,
                    "supplier_name": vendor.supplier_name,
                    "contact_person": vendor.contact_person,
                    "contact_phone": vendor.contact_phone
                }

        return result

    async def _generate_record_code(self, db: Session) -> str:
        """生成维修单号"""
        from datetime import datetime
        import random
        import string

        # 日期部分
        date_str = datetime.now().strftime("%Y%m%d")

        # 序号部分（4位随机数）
        sequence = "".join(random.choices(string.digits, k=4))

        # 组合单号: MT202501240001
        record_code = f"MT{date_str}{sequence}"

        # 检查是否重复，如果重复则重新生成
        while maintenance_record.get_by_code(db, record_code):
            sequence = "".join(random.choices(string.digits, k=4))
            record_code = f"MT{date_str}{sequence}"

        return record_code


# 创建全局实例
maintenance_service = MaintenanceService()