greater_wms/container/container_operate.py

from django.db import transaction
from rest_framework.response import Response
from rest_framework.views import APIView
from django.utils import timezone
from .models import ContainerListModel,ContainerDetailModel,ContainerOperationModel,TaskModel,ContainerWCSModel
from bound.models import BoundBatchModel,BoundDetailModel,BoundListModel
from .serializers import *

class ContainerService:
    """
    托盘管理服务
        function:
            create_container_operation: 创建托盘操作记录

    """
    @staticmethod
    @transaction.atomic
    def create_container_operation(data, logger):
     
        # 初始化基础数据
        data['month'] = timezone.now().strftime('%Y%m')
        
        # 处理容器
        container = ContainerManager.get_or_create_container(
            container_code=data.get('container'),
            logger=logger
        )
        data['container_code'] = container.container_code

        data['container'] = container.id

        # 处理批次数据
        if data.get('goods_class') == 2:
            return BatchProcessor.handle_special_batches(data, logger)
        return BatchProcessor.handle_normal_batches(data, logger)

class ContainerManager:
    @staticmethod
    def get_or_create_container(container_code, logger):
        """统一管理容器对象"""
        if container := ContainerListModel.objects.filter(container_code=container_code).first():
            logger.info(f"托盘 {container_code} 已存在")
            return container
            
        logger.info(f"创建新托盘 {container_code}")
        serializer = ContainerListPostSerializer(data={'container_code': container_code})
        serializer.status = 5
        serializer.is_valid(raise_exception=True)
        serializer.save()
        return ContainerListModel.objects.filter(container_code=container_code).first()

class BatchProcessor:
    @staticmethod
    def handle_normal_batches(data, logger):
        """标准批次处理流程"""
        for batch in data.get('batches', []):
            try:
                BatchOperator.process_single_batch(data, batch, logger)
            except Exception as e:
                logger.error(f"批次处理失败: {str(e)}")
                continue
        return Response({"status": "success"})

    @staticmethod
    def handle_special_batches(data, logger):
        """托盘组处理流程"""
        for batch in data.get('batches', []):
            try:
                BatchOperator.process_single_container(data, batch, logger)
            except Exception as e:
                logger.error(f"托盘组处理失败: {str(e)}")
                continue
        return Response({"status": "success"})

class BatchOperator:
    @staticmethod
    def process_single_container(data, batch, logger):
        """单个批次处理托盘"""
        # 获取批次对象
        container_obj = ContainerManager.get_or_create_container(
            batch.get('goods_code'), 
            logger
        )

        # 更新批次状态
        update_result = BatchStatusUpdater.update_container_status(
            container_obj, 
            5,
            data.get('container_code'),
        )
        print(update_result)
        # 创建托盘关联记录

        RecordCreator.create_container_related_records(
            data=data,
            container_obj=container_obj
        )

    @staticmethod
    def process_single_batch(data, batch, logger):
        """单个批次处理批次"""
        # 获取批次对象
        bound_obj = BatchValidator.get_valid_batch(
            batch.get('goods_code'), 
            logger
        )
        
        # 更新批次状态（新增数目）
        update_result = BatchStatusUpdater.update_status(
            bound_obj, 
            batch['goods_qty'] * batch['goods_group'],
            logger
        )
        
        # 创建关联记录
        RecordCreator.create_related_records(
            data=data,
            bound_obj=bound_obj,
            batch_data= batch,
            qty_diff=update_result['added_qty'],
            logger=logger
        )

class BatchValidator:
    @staticmethod
    def get_valid_batch(bound_number, logger):
        """批次校验模块"""
        if not (bound_obj := BoundBatchModel.objects.filter(bound_number=bound_number).first()):
            logger.error(f"批次 {bound_number} 不存在")
            raise ValueError("无效批次号")
        return bound_obj

class BatchStatusUpdater:
    @staticmethod
    def update_status(bound_obj, added_qty, logger):
        """批次状态管理"""
        last_qty = bound_obj.goods_in_qty
        bound_obj.goods_in_qty += added_qty
        
        if bound_obj.goods_in_qty >= bound_obj.goods_qty:
            FullBatchHandler.mark_as_completed(bound_obj, logger)
        else:
            bound_obj.status = 0
        
        bound_obj.save()
        return {
            'last_qty': last_qty,
            'added_qty': added_qty
        }
    @staticmethod
    def update_container_status(container_obj, status, container_code):
        """批次状态管理""" 
    
        container_obj.status = status
        container_obj.available = False
        container_obj.current_location = '托盘'+str(container_code)
        container_obj.save()
        return {
            'last_qty': 1,
            'added_qty': 1
        }
  

class FullBatchHandler:
    @staticmethod
    def mark_as_completed(bound_obj, logger):
        """满批次处理"""
        # bound_obj.goods_in_qty = bound_obj.goods_qty
        bound_obj.status = 1
        BatchRelationUpdater.update_related_records(bound_obj, logger)

class BatchRelationUpdater:
    @staticmethod
    def update_related_records(bound_obj, logger):
        """更新关联单据"""
        if detail := BoundDetailModel.objects.filter(bound_batch=bound_obj.id).first():
            detail.status = 1
            detail.save()
            CompletionChecker.check_list_completion(detail.bound_list_id, logger)

class CompletionChecker:
    @staticmethod
    def check_list_completion(list_id, logger):
        """完成度检查"""
        completed = BoundDetailModel.objects.filter(bound_list=list_id, status=1).count()
        total = BoundDetailModel.objects.filter(bound_list=list_id).count()
        
        if completed == total:
            ListStatusUpdater.update_list_status(list_id, 102, logger)

class ListStatusUpdater:
    @staticmethod
    def update_list_status(list_id, status, logger):
        """更新单据状态"""
        if list_obj := BoundListModel.objects.filter(id=list_id).first():
            list_obj.bound_status = status
            list_obj.save()
            logger.info("入库申请全部完成")

class RecordCreator:
    @staticmethod
    def create_container_related_records(data, container_obj, qty_diff=1):
        """记录创建批次"""
        # 生成明细记录
        detail_data = RecordBuilder.build_detail_data_container_group(
            data=data,
            container_obj=container_obj,
            qty_diff=qty_diff
        )

        DetailRecordCreator.create_container_related_records(detail_data)
        
        # 生成操作记录
        operation_data = RecordBuilder.build_operation_data_container_group(
            detail_data=detail_data,
            qty_diff=qty_diff
        )

        OperationRecordCreator.create_container_related_operation_records(operation_data)

    @staticmethod
    def create_related_records(data, bound_obj, batch_data, qty_diff, logger):
        """记录创建批次"""
        # 生成明细记录

        goods_group = batch_data.get('goods_group')
        goods_qty = batch_data.get('goods_qty')
        for i in range(goods_group):
            detail_data = RecordBuilder.build_detail_data(
                data=data,
                bound_obj=bound_obj,
                qty_diff=goods_qty
            )
            DetailRecordCreator.create_record(detail_data)
        
        # 生成操作记录
        operation_data = RecordBuilder.build_operation_data(
            detail_data=detail_data,
            qty_diff=qty_diff
        )
        OperationRecordCreator.create_operation_record(operation_data)



class RecordBuilder:
    @staticmethod
    def build_detail_data(data, bound_obj, qty_diff):
        """构建明细数据结构"""
        return {
            "container": data['container'],
            "batch": bound_obj.id,
            "goods_code": bound_obj.goods_code,
            "goods_desc": bound_obj.goods_desc,
            "goods_qty": qty_diff,
            "goods_weight": bound_obj.goods_weight,
            "status": 1,
            "month": data['month'],
            "creater": data.get('creater', 'wms')
        }
    
    @staticmethod
    def build_detail_data_container_group(data, container_obj, qty_diff=1):
        """构建明细数据结构"""
        return {
            "container": data['container'],
            "goods_code": container_obj.container_code,
            "goods_desc": '托盘组',
            "goods_qty": qty_diff,
            "goods_weight": 0,
            "status": 1,
            "month": data['month'],
            "creater": data.get('creater', 'wms')
        }

    @staticmethod
    def build_operation_data(detail_data, qty_diff):
        """构建操作记录数据"""

        return {
            "month": detail_data['month'],
            "container": detail_data['container'],
            "operation_type": 'container',
            "batch": detail_data['batch'],
            "goods_code": detail_data['goods_code'],
            "goods_desc": detail_data['goods_desc'],
            "goods_qty": qty_diff,
            "goods_weight": qty_diff,
            "operator": detail_data['creater'],
            "timestamp": timezone.now(),
            "from_location": "container",
            "to_location": "container",
            "memo": f"每组{detail_data['goods_code']}数目{detail_data['goods_qty']},PDA组盘共计组盘{qty_diff}"
        }
    
    def build_operation_data_container_group(detail_data, qty_diff):
        """构建操作记录数据"""

        return {
            "month": detail_data['month'],
            "container": detail_data['container'],
            "operation_type": 'container',
            "goods_code": detail_data['goods_code'],
            "goods_desc": detail_data['goods_desc'],
            "goods_qty": qty_diff,
            "goods_weight": detail_data['goods_weight'],
            "operator": detail_data['creater'],
            "timestamp": timezone.now(),
            "from_location": "container",
            "to_location": "container",
            "memo": f"托盘组下的：{detail_data['goods_code']}合盘"
        }

class DetailRecordCreator:
    @staticmethod
    def create_record(data):
        """统一创建明细记录"""
        serializer = ContainerDetailPostSerializer(data=data)
        serializer.is_valid(raise_exception=True)
        serializer.save()

    def create_container_related_records(data):
        """记录创建批次"""
        # 生成明细记录
        container_obj = ContainerListModel.objects.filter(id=data['container']).first()
        container_obj.status = 5
        container_obj.save()
        if ContainerDetailModel.objects.filter(container=container_obj,goods_code=data['goods_code'],status=1,is_delete=False).first():
            return 
        ContainerDetailModel.objects.create(
            container=container_obj,
            goods_code=data['goods_code'],
            goods_desc=data['goods_desc'],
            goods_qty=data['goods_qty'],
            goods_weight=data['goods_weight'],
            goods_class=2,
            status=data['status'],
            month=data['month'],
            creater=data['creater']
        )

class OperationRecordCreator:
    @staticmethod
    def create_operation_record(data):
        """统一创建操作记录"""
        serializer = ContainerOperationPostSerializer(data=data)
        serializer.is_valid(raise_exception=True)
        serializer.save()
    @staticmethod

    def create_container_related_operation_records(data):
        container_obj = ContainerListModel.objects.filter(id=data['container']).first()

        if ContainerOperationModel.objects.filter(container_id=container_obj.id,goods_code=data['goods_code'],is_delete=False).first():
            return 
        ContainerOperationModel.objects.create(      
            month=data['month'],
            container_id=container_obj.id,
            operation_type=data['operation_type'],
            bound_id=None,
            goods_code=data['goods_code'],
            goods_desc=data['goods_desc'],
            goods_qty=data['goods_qty'],
            goods_weight=data['goods_weight'],
            operator=data['operator'],
            timestamp=data['timestamp'],
            from_location=data['from_location'],
            to_location=data['to_location'],
            memo=data['memo']
        )

# 最终视图类
class OperationView(APIView):
    def create(self, request, *args, **kwargs):
        try:
            return ContainerService.create_container_operation(
                data=request.data,
                logger=self.get_logger()  # 假设有日志方法
            )
        except Exception as e:
            return Response({"error": str(e)}, status=400)