Project Management

class Proyecto(models.Model):
    ESTADOS = [
        ('PLANEACION', 'Planeación'),
        ('EN_CURSO', 'En Curso'),
        ('SUSPENDIDO', 'Suspendido'),
        ('COMPLETADO', 'Completado'),
        ('CANCELADO', 'Cancelado'),
    ]
    
    # Auto-generated code: PROY-YYYY-NNNN
    codigo = models.CharField(max_length=50, unique=True)
    nombre = models.CharField(max_length=255)
    descripcion = models.TextField()
    
    # Dates
    fecha_inicio = models.DateField()
    fecha_fin_planificada = models.DateField()
    fecha_fin_real = models.DateField(blank=True, null=True)
    
    # Management
    responsable = models.ForeignKey(User, on_delete=models.SET_NULL)
    estado = models.CharField(max_length=20, choices=ESTADOS)
    
    # Budget
    presupuesto = models.DecimalField(max_digits=12, decimal_places=2)
    
    # Progress
    porcentaje_avance = models.DecimalField(max_digits=5, decimal_places=2, default=0)

class Actividad(models.Model):
    PRIORIDADES = [
        ('BAJA', 'Baja'),
        ('MEDIA', 'Media'),
        ('ALTA', 'Alta'),
        ('CRITICA', 'Crítica'),
    ]
    
    proyecto = models.ForeignKey('Proyecto', on_delete=models.CASCADE)
    codigo = models.CharField(max_length=50)
    nombre = models.CharField(max_length=255)
    descripcion = models.TextField()
    
    # Schedule
    fecha_inicio = models.DateField()
    fecha_fin = models.DateField()
    duracion_dias = models.IntegerField()
    
    # Dependencies
    predecesoras = models.ManyToManyField('self', 
                                         symmetrical=False,
                                         related_name='sucesoras',
                                         blank=True)
    
    # Assignment
    responsable = models.ForeignKey(User, on_delete=models.SET_NULL)
    ubicacion = models.ForeignKey('activos.Ubicacion', on_delete=models.SET_NULL)
    
    # Status
    estado = models.CharField(max_length=20)
    porcentaje_completado = models.IntegerField(default=0)
    prioridad = models.CharField(max_length=10, choices=PRIORIDADES)
    
    # Location on floor plan
    coordenada_x = models.FloatField(blank=True, null=True)
    coordenada_y = models.FloatField(blank=True, null=True)

class DocumentoProyecto(models.Model):
    proyecto = models.ForeignKey('Proyecto', on_delete=models.CASCADE)
    documento = models.ForeignKey('documentos.Documento', on_delete=models.CASCADE)
    fecha_vinculacion = models.DateField(auto_now_add=True)

def gantt_view(request, proyecto_id):
    proyecto = Proyecto.objects.get(id=proyecto_id)
    actividades = proyecto.actividad_set.all()
    
    # Prepare data for Gantt chart
    gantt_data = []
    for actividad in actividades:
        gantt_data.append({
            'id': actividad.id,
            'text': actividad.nombre,
            'start_date': actividad.fecha_inicio.isoformat(),
            'duration': actividad.duracion_dias,
            'progress': actividad.porcentaje_completado / 100,
            'parent': actividad.proyecto.id,
            'dependencies': [p.id for p in actividad.predecesoras.all()]
        })
    
    return render(request, 'gantt.html', {
        'proyecto': proyecto,
        'gantt_data': json.dumps(gantt_data)
    })

def calcular_ruta_critica(proyecto):
    """Calculate critical path using forward/backward pass"""
    actividades = list(proyecto.actividad_set.all())
    
    # Forward pass: Calculate earliest start/finish
    for actividad in actividades:
        if not actividad.predecesoras.exists():
            actividad.early_start = proyecto.fecha_inicio
        else:
            max_finish = max([p.early_finish for p in actividad.predecesoras.all()])
            actividad.early_start = max_finish
        
        actividad.early_finish = actividad.early_start + timedelta(days=actividad.duracion_dias)
    
    # Backward pass: Calculate latest start/finish
    for actividad in reversed(actividades):
        if not actividad.sucesoras.exists():
            actividad.late_finish = proyecto.fecha_fin_planificada
        else:
            min_start = min([s.late_start for s in actividad.sucesoras.all()])
            actividad.late_finish = min_start
        
        actividad.late_start = actividad.late_finish - timedelta(days=actividad.duracion_dias)
    
    # Identify critical activities (slack = 0)
    critical_activities = [
        a for a in actividades 
        if (a.late_start - a.early_start).days == 0
    ]
    
    return critical_activities

def ai_chat(request, proyecto_id):
    proyecto = Proyecto.objects.get(id=proyecto_id)
    question = request.POST.get('question')
    
    # Build context from project data
    context = f"""
    Project: {proyecto.nombre}
    Status: {proyecto.estado}
    Progress: {proyecto.porcentaje_avance}%
    Activities: {proyecto.actividad_set.count()}
    Budget: ${proyecto.presupuesto}
    """
    
    # Query Gemini API
    import google.generativeai as genai
    genai.configure(api_key=settings.GEMINI_API_KEY)
    
    model = genai.GenerativeModel('gemini-pro')
    response = model.generate_content(
        f"Context: {context}\n\nQuestion: {question}"
    )
    
    return JsonResponse({'answer': response.text})

def vincular_documento(request, proyecto_id):
    proyecto = Proyecto.objects.get(id=proyecto_id)
    documento_id = request.POST.get('documento_id')
    
    DocumentoProyecto.objects.create(
        proyecto=proyecto,
        documento_id=documento_id
    )
    
    return JsonResponse({'status': 'ok'})