from django.utils.translation import gettext_lazy as _
from django.db import models
from datetime import timedelta

from ..models.weekpref import WeekPreference
from ..models.aircrafts import AircraftTypes


class HourBuilding(models.Model):
    id = models.BigAutoField(
        primary_key=True
    )

    weekpref = models.ForeignKey(
        WeekPreference,
        null=False,
        on_delete=models.CASCADE
    )

    aircraft = models.CharField(
        null=False,
        choices=AircraftTypes
    )

    monday = models.BooleanField(
        default=True,
        null=False
    )

    tuesday = models.BooleanField(
        default=True,
        null=False
    )

    wednesday = models.BooleanField(
        default=True,
        null=False
    )

    thursday = models.BooleanField(
        default=True,
        null=False
    )

    friday = models.BooleanField(
        default=True,
        null=False
    )

    saturday = models.BooleanField(
        default=True,
        null=False
    )

    sunday = models.BooleanField(
        default=True,
        null=False
    )

    notes = models.TextField(
        max_length=140,
        null=True,
        blank=True
    )

    def __str__(self):
        return f"Hour Building: {self.aircraft}"

class HourBuildingLeg(models.Model):
    id = models.BigAutoField(
        primary_key=True
    )

    hb = models.ForeignKey(
        HourBuilding,
        on_delete=models.CASCADE
    )

    departure = models.CharField(
        null=False,
        blank=False,
        default="LILV",
        max_length=4
    )

    destination = models.CharField(
        null=False,
        blank=False,
        default="LILV",
        max_length=4
    )

    time = models.DurationField(
        null=False,
        default = timedelta(hours=1)
    )

    stop = models.BooleanField(
        default=False
    )

    def __str__(self):
        if self.stop:
            return "Refuelling Stop"
        else:
            return f"Flight Leg: {self.departure} -> {self.destination}"