from django.db import models
from bound.models import BoundBatchModel,BoundDetailModel,OutBatchModel,BoundListModel,BoundListModel


# Create your models here.
# 主表：托盘数据
class ContainerListModel(models.Model):
    CONTAINER_STATUS = (
        (0, '空置'),
        (1, '入库中'),
        (2, '在库'),
        (3, '出库中'),
        (4, '已出库'),
        (5, '空托盘组')
    )
   
    container_code = models.IntegerField( verbose_name='托盘编号',unique=True)
    current_location = models.CharField(max_length=50, verbose_name='当前库位', default='N/A')
    target_location = models.CharField(max_length=50, verbose_name='目标库位', default='N/A')
    available= models.BooleanField(default=True, verbose_name='可用')
    status = models.IntegerField(choices=CONTAINER_STATUS, default=0, verbose_name='托盘状态')
    last_operation = models.DateTimeField(auto_now=True, verbose_name='最后操作时间')

    class Meta:
        db_table = 'container_list'
        verbose_name = 'ContainerList'
        verbose_name_plural = "ContainerList"
        ordering = ['-container_code']

# 明细表：托盘详细数据记录当前组盘的 批次 数量 使用托盘码和状态来获取、托盘上的物料信息，
class ContainerDetailModel(models.Model):
    BATCH_STATUS=(
        (0, '空盘'),
        (1, '组盘'),
        (2, '在库'),
        (3, '已出库')
    )
    BATCH_CLASS = (
        (1, '成品'),
        (2, '空盘'),
        (3, '散盘'),
    )
    month = models.IntegerField(verbose_name='月份')
    container = models.ForeignKey(ContainerListModel, on_delete=models.CASCADE, related_name='details')
    batch = models.ForeignKey(BoundBatchModel, on_delete=models.CASCADE, verbose_name='批次',blank=True, null=True)
    goods_class = models.IntegerField(verbose_name='货品类别',choices=BATCH_CLASS, default=1)
    goods_code = models.CharField(max_length=50, verbose_name='货品编码')
    goods_desc = models.CharField(max_length=100, verbose_name='货品描述')
    goods_qty = models.IntegerField(verbose_name='数量')
    goods_out_qty = models.IntegerField(verbose_name='出库数量', default=0)
    goods_weight = models.DecimalField(max_digits=10, decimal_places=3, verbose_name='重量')
    status = models.IntegerField(choices=BATCH_STATUS,default=0, verbose_name='状态')          
    creater = models.CharField(max_length=50, verbose_name='创建人')
    create_time = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
    update_time = models.DateTimeField(auto_now=True, verbose_name='更新时间')
    is_delete = models.BooleanField(default=False, verbose_name='是否删除')

    class Meta:
        db_table = 'container_detail'
        verbose_name = 'ContainerDetail'
        verbose_name_plural = "ContainerDetail"
        ordering = ['-id']

    def __str__(self):
        return f"{self.container_code} - {self.batch.bound_number} - {self.goods_code} - {self.goods_desc} - {self.goods_qty}"
    
    def save(self, *args, **kwargs):
        """
        更新托盘上的物料数量,更新批次上的
        goods_in_qty(组盘数目)
        ,goods_in_location_qty(在库数目)
        ,goods_out_qty(出库数目)
    
        """
        if self.batch:
          
            super().save(*args, **kwargs)

            if self.goods_qty - self.goods_out_qty <= 0:
                self.status = 3
                super().save(*args, **kwargs)
            elif self.goods_qty - self.goods_out_qty > 0 and self.goods_out_qty > 0:
                self.status = 2
                super().save(*args, **kwargs)
            if self.status == 3 and self.goods_qty - self.goods_out_qty > 0 :
                self.status = 2
                super().save(*args, **kwargs)

            container_all_qty = ContainerDetailModel.objects.filter(batch=self.batch,is_delete=False).aggregate(total_qty=models.Sum('goods_qty'))['total_qty']
            container_all_out_qty = ContainerDetailModel.objects.filter(batch=self.batch,is_delete=False).aggregate(total_out_qty=models.Sum('goods_out_qty'))['total_out_qty']
            self.batch.goods_in_qty = container_all_qty
            self.batch.goods_in_location_qty = container_all_qty - container_all_out_qty
            self.batch.goods_out_qty = container_all_out_qty
            self.batch.save()


        

    


# 明细表：操作记录 记录每次出入库的记录，使用goods来进行盘点，使用托盘码来进行托盘的操作记录
class ContainerOperationModel(models.Model):
    OPERATION_TYPES = (
        ('container','组盘'),
        ('inbound', '入库'),
        ('outbound', '出库'), 
        ('adjust', '调整'),
    )
    month = models.IntegerField(verbose_name='月份')
    container = models.ForeignKey(ContainerListModel, on_delete=models.CASCADE, related_name='operations')
    operation_type = models.CharField(max_length=20, choices=OPERATION_TYPES, verbose_name='操作类型')
    bound_id = models.IntegerField(verbose_name='出库申请', null=True, blank=True)

    batch = models.ForeignKey(BoundBatchModel, on_delete=models.CASCADE, verbose_name='批次',null=True, blank=True)
    goods_code = models.CharField(max_length=50, verbose_name='货品编码')
    goods_desc = models.CharField(max_length=100, verbose_name='货品描述')
    goods_qty = models.IntegerField(verbose_name='数量')
    goods_weight = models.DecimalField(max_digits=10, decimal_places=3, verbose_name='重量')

    operator = models.CharField(max_length=50, verbose_name='操作人')
    timestamp = models.DateTimeField(auto_now_add=True, verbose_name='操作时间')    
    from_location = models.CharField(max_length=50, null=True, verbose_name='原库位')
    to_location = models.CharField(max_length=50, null=True, verbose_name='目标库位')
    memo = models.TextField(null=True, verbose_name='备注')
    is_delete = models.BooleanField(default=False, verbose_name='是否删除')


    class Meta:
        db_table = 'container_operation'    
        verbose_name = 'ContainerOperation'
        verbose_name_plural = "ContainerOperation"
        ordering = ['-timestamp']

class ContainerWCSModel(models.Model):
    TASK_STATUS = (
        (100, '等待中'),
        (101, '处理中'),
        (102, '已暂停'),
        (103, '入库中'),
        (200, '已发送'),
        (300, '已完成'),
        (400, '失败'),
    )
    taskid = models.CharField(max_length=50, verbose_name='任务ID')
    batch = models.ForeignKey(BoundBatchModel, on_delete=models.CASCADE, verbose_name='关联批次',null=True, blank=True )
    batch_out = models.ForeignKey(OutBatchModel, on_delete=models.CASCADE, verbose_name='出库批次' , null=True, blank=True )
    bound_list = models.ForeignKey(BoundListModel, on_delete=models.CASCADE, verbose_name='关联出库单',null=True, blank=True )
    batch_number = models.CharField(max_length=50, verbose_name='批次号',null=True, blank=True )
    sequence = models.BigIntegerField(verbose_name='任务顺序')
    priority = models.IntegerField(default=100, verbose_name='优先级')
    month = models.IntegerField(verbose_name='月份')
    tasktype = models.CharField(max_length=50, verbose_name='任务类型')
    tasknumber = models.IntegerField(verbose_name='任务号',unique=True)
    order_number = models.IntegerField(verbose_name='c_number')
    container = models.CharField(max_length=50, verbose_name='托盘号')
    current_location = models.CharField(max_length=50, verbose_name='当前库位')
    target_location = models.CharField(max_length=50, verbose_name='目标库位')
    message = models.TextField(verbose_name='消息')
    working = models.IntegerField(default = 1,verbose_name='工作状态')
    status = models.IntegerField(choices=TASK_STATUS, default=100, verbose_name='状态')

    create_time = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
    is_delete = models.BooleanField(default=False, verbose_name='是否删除')

    class Meta:
        db_table = 'container_wcs'
        verbose_name = 'ContainerWCS'
        verbose_name_plural = "ContainerWCS"
        ordering = ['-create_time']
    
    def to_dict(self):
        return {
            'container': self.container,
            'current_location': self.current_location,
            'month' : self.month,
            'target_location': self.target_location,
            'tasktype': self.tasktype,
            'taskid': self.taskid,
            'taskNumber': self.tasknumber-20000000000,
            'message': self.message,
            'container': self.container,
            'status': self.status
        }
    def __str__(self):
        return f"{self.taskid} - {self.get_status_display()}"
# 这里的批次详情是主入库申请单下的子批次
class TaskModel(models.Model):
    
    task_wcs = models.ForeignKey(ContainerWCSModel, on_delete=models.CASCADE, related_name='tasks')
    batch_detail = models.ForeignKey(BoundDetailModel, on_delete=models.CASCADE, verbose_name='批次详情')
    container_detail = models.ForeignKey(ContainerDetailModel, on_delete=models.CASCADE, verbose_name='托盘明细')

    class Meta:
        db_table = 'task'
        verbose_name = 'Task'
        verbose_name_plural = "Task"
        ordering = ['-id']

class out_batch_detail(models.Model):
    out_bound = models.ForeignKey(BoundListModel, on_delete=models.CASCADE, related_name='out_batch_details')
    container = models.ForeignKey(ContainerListModel, on_delete=models.CASCADE, related_name='out_batch_details')
    container_detail = models.ForeignKey(ContainerDetailModel, on_delete=models.CASCADE, verbose_name='托盘明细')
    out_goods_qty = models.IntegerField(verbose_name='数量')
    last_out_goods_qty = models.IntegerField(verbose_name='上次出库数量', default=0)
    working = models.IntegerField(default = 1,verbose_name='工作状态')
    is_delete = models.BooleanField(default=False, verbose_name='是否删除')
    class Meta:
        db_table = 'out_batch_detail'
        verbose_name = 'OutBatchDetail'
        verbose_name_plural = "OutBatchDetail"
        ordering = ['container']