Schedule Tracking

class CronogramaVersion(models.Model):
    """
    La 'foto' del archivo .mpp importado.
    """
    id_ot = models.ForeignKey(OTE, on_delete=models.CASCADE, related_name='cronogramas')
    nombre_version = models.CharField(max_length=150)
    archivo_mpp = models.FileField(upload_to='operaciones/mpps/')
    fecha_carga = models.DateTimeField(auto_now_add=True)
    es_activo = models.BooleanField(default=True)
    fecha_inicio_proyecto = models.DateField(null=True, blank=True)
    fecha_fin_proyecto = models.DateField(null=True, blank=True)

    class Meta:
        db_table = 'importacion_cronograma'

class TareaCronograma(models.Model):
    """
    Desglose de tareas del Project.
    """
    version = models.ForeignKey(CronogramaVersion, on_delete=models.CASCADE, related_name='tareas')
    uid_project = models.IntegerField()  # Unique ID from MS Project
    id_project = models.IntegerField()   # Sequential ID from MS Project
    wbs = models.CharField(max_length=50)  # Work Breakdown Structure code
    nombre = models.CharField(max_length=500)
    nivel_esquema = models.IntegerField(default=0)  # Indentation level
    es_resumen = models.BooleanField(default=False)  # Summary/rollup task
    padre_uid = models.IntegerField(null=True, blank=True)  # Parent task UID
    fecha_inicio = models.DateField(null=True)
    fecha_fin = models.DateField(null=True)
    duracion_dias = models.DecimalField(max_digits=10, decimal_places=2, default=0)
    porcentaje_mpp = models.DecimalField(max_digits=5, decimal_places=2, default=0)  # % from MPP
    porcentaje_completado = models.DecimalField(max_digits=5, decimal_places=2, default=0)
    recursos = models.TextField(blank=True, default='')  # Comma-separated resource names

    class Meta:
        db_table = 'importacion_cronograma_tarea'
        indexes = [models.Index(fields=['version', 'uid_project'])]

class AvanceCronograma(models.Model):
    """
    La doble verdad: Real vs Cliente.
    Se separa de la tarea para no sobreescribir al re-importar versiones.
    """
    tarea = models.OneToOneField(TareaCronograma, on_delete=models.CASCADE, related_name='avance')
    porcentaje_real = models.DecimalField(max_digits=5, decimal_places=2, default=0)
    porcentaje_cliente = models.DecimalField(max_digits=5, decimal_places=2, default=0)
    comentario = models.TextField(blank=True)
    fecha_actualizacion = models.DateTimeField(auto_now=True)

    class Meta:
        db_table = 'importacion_cronograma_avance'

class DependenciaTarea(models.Model):
    """
    Relaciones de precedencia entre tareas del cronograma.
    """
    version = models.ForeignKey(CronogramaVersion, on_delete=models.CASCADE, related_name='dependencias')
    tarea_predecesora_uid = models.IntegerField()
    tarea_sucesora_uid = models.IntegerField()
    tipo = models.CharField(max_length=2, default='FS', choices=[
        ('FS', 'Fin a Inicio'),      # Finish-to-Start
        ('SS', 'Inicio a Inicio'),   # Start-to-Start
        ('FF', 'Fin a Fin'),         # Finish-to-Finish
        ('SF', 'Inicio a Fin'),      # Start-to-Finish
    ])
    lag_dias = models.DecimalField(max_digits=8, decimal_places=2, default=0)

    class Meta:
        db_table = 'importacion_cronograma_dependencia'

@login_required
@require_http_methods(["GET"])
def obtener_arbol_mpp(request):
    """
    Obtiene las tareas del cronograma vigente de una OT y las formatea
    en una estructura jerárquica (_children) para TOAST UI Tree Grid.
    """
    try:
        ot_id = request.GET.get('ot_id')
        version_activa = CronogramaVersion.objects.get(id_ot_id=ot_id, es_activo=True)
        
        tareas = TareaCronograma.objects.filter(version=version_activa).order_by('id_project')

        # Build task dictionary
        tareas_dict = {}
        for t in tareas:
            tareas_dict[t.uid_project] = {
                "wbs": t.wbs,
                "nombre": t.nombre,
                "fecha_inicio": t.fecha_inicio.strftime("%d/%m/%Y") if t.fecha_inicio else "",
                "fecha_fin": t.fecha_fin.strftime("%d/%m/%Y") if t.fecha_fin else "",
                "duracion_dias": float(t.duracion_dias),
                "porcentaje_mpp": float(t.porcentaje_mpp),
                "recursos": t.recursos,
                "uid_project": t.uid_project,
                "padre_uid": t.padre_uid,
                "_attributes": {"expanded": True}  # Auto-expand in tree view
            }

        # Build hierarchical tree
        tree_data = []
        for uid, tarea in tareas_dict.items():
            padre_uid = tarea["padre_uid"]
            
            if padre_uid is not None and padre_uid in tareas_dict:
                # Add as child to parent
                padre = tareas_dict[padre_uid]
                if "_children" not in padre:
                    padre["_children"] = []
                padre["_children"].append(tarea)
            else:
                # Top-level task
                tree_data.append(tarea)

        return JsonResponse({"estatus": "ok", "data": tree_data})

    except CronogramaVersion.DoesNotExist:
        return JsonResponse({"estatus": "error", "mensaje": "No hay cronograma activo"})
    except Exception as e:
        import traceback
        traceback.print_exc()
        return JsonResponse({"estatus": "error", "mensaje": str(e)}, status=500)