from django.db import models
from django.contrib.auth.models import User

# MODELOS

class Aerolinea(models.Model):
    nombre = models.CharField(max_length=100,verbose_name="nombre")
    
    def __str__(self):
        return self.nombre


class Ciudades(models.Model):
    postal = models.IntegerField(primary_key=True)
    nombre = models.CharField(max_length = 30,null=False)
    foto = models.ImageField(upload_to='vuelos/ciudades/',null = True, blank = True)

    def __str__(self):
        return self.nombre

class Vuelo(models.Model):
    id_vuelo = models.CharField(max_length=30,primary_key=True, verbose_name="id_vuelo")
    aerolinea = models.ForeignKey(Aerolinea,on_delete=models.CASCADE, verbose_name="aerolinea")
    destino = models.ForeignKey(Ciudades,on_delete=models.CASCADE)
    costo = models.FloatField(blank=False,null=False, verbose_name="costo")
    fecha = models.DateTimeField(blank=False,null=False)
    requisitos = models.TextField(null=False,blank=False)
    def __str__(self):
        return str(self.id_vuelo)


class Escala(models.Model):
    vuelo = models.ForeignKey(Vuelo,on_delete=models.CASCADE)
    ciudad_escala = models.ForeignKey(Ciudades,on_delete=models.CASCADE)
    fecha_escala = models.DateTimeField(null=False,blank=False)

    def __str__(self):
        return f'{self.vuelo},{self.ciudad_escala}' 

