Monthly Reports

class ReporteMensual(models.Model):
    """
    Representa la 'Carpeta Mensual' de una OT.
    Agrupa todos los reportes diarios y de producción de un mes específico.
    """
    id_ot = models.ForeignKey(OTE, on_delete=models.CASCADE, related_name='reportes_mensuales', blank=True, null=True)
    mes = models.IntegerField(help_text="Mes numérico (1-12)")
    anio = models.IntegerField(help_text="Año (Ej. 2025)")
    archivo = models.URLField(blank=True, null=True, verbose_name="Link Evidencia (Drive)")
    id_estatus = models.ForeignKey(Estatus, on_delete=models.CASCADE, limit_choices_to={'nivel_afectacion': 5}, default=1, verbose_name="Estatus Cierre")
    fecha_creacion = models.DateTimeField(auto_now_add=True)
    fecha_actualizacion = models.DateTimeField(auto_now=True)

    class Meta:
        db_table = 'reporte_mensual_header'
        unique_together = ['id_ot', 'mes', 'anio']
        verbose_name = "Reporte Mensual"
        verbose_name_plural = "Reportes Mensuales"

    def __str__(self):
        return f"Reporte {self.id_ot.orden_trabajo} - {self.mes}/{self.anio}"

@login_required
@require_http_methods(["POST"])
def guardar_archivo_mensual(request):
    """
    Guarda el enlace de archivo en el ReporteMensual
    """
    try:
        data = json.loads(request.body)
        
        id_ot = data.get('id_ot')
        mes = data.get('mes')
        anio = data.get('anio')
        url = data.get('archivo')
        
        if not all([id_ot, mes, anio]):
            return JsonResponse({
                'exito': False,
                'mensaje': 'Faltan datos obligatorios (OT, Mes o Año).'
            })

        if not url:
            return JsonResponse({
                'exito': False,
                'mensaje': 'La URL del archivo es obligatoria.'
            })

        with transaction.atomic():
            reporte, created = ReporteMensual.objects.get_or_create(
                id_ot_id=id_ot,
                mes=int(mes),
                anio=int(anio),
                defaults={
                    'id_estatus_id': 1
                }
            )

            reporte.archivo = url
            reporte.save()

        return JsonResponse({
            'exito': True,
            'mensaje': 'Evidencia vinculada al Reporte Mensual correctamente.',
            'archivo': reporte.archivo
        })

    except Exception as e:
        return JsonResponse({
            'exito': False,
            'mensaje': f'Error interno al guardar evidencia: {str(e)}'
        }, status=500)

reporte_mensual, created = ReporteMensual.objects.get_or_create(
    id_ot_id=id_ot,
    mes=mes,
    anio=anio,
    defaults={'id_estatus_id': 1}
)

archivo_url = ''
dias_bloqueados = []

try:
    reporte = ReporteMensual.objects.filter(id_ot_id=id_ot, mes=mes, anio=anio).first()
    if reporte:
        dias_bloqueados = list(ReporteDiario.objects.filter(
            id_reporte_mensual=reporte,
            id_estatus_id=17,  # Cerrado status
            id_sitio=id_sitio_int
        ).values_list('fecha__day', flat=True))
        
        if reporte.archivo:
            archivo_url = reporte.archivo
except Exception:
    pass