from django.db import models


class AircraftBase(models.Model):
    name = models.CharField(max_length=20)
    manufacturer = models.CharField(max_length=50)
    range_nm = models.IntegerField()
    capacity = models.IntegerField()
    cruise_speed_kt = models.IntegerField()

    def __str__(self):
        return f"{self.manufacturer} {self.name}"


class Equipment(models.Model):
    registration = models.CharField(max_length=10, unique=True)
    model = models.ForeignKey(
        "AircraftBase", on_delete=models.CASCADE, related_name="aircraft"
    )
    owner = models.ForeignKey("Carrier", on_delete=models.CASCADE, related_name="fleet")

    def __str__(self):
        return f"{self.registration} ({self.model})"