from django.db import models
from .aircraft import Equipment


class Carrier(models.Model):
    icao = models.CharField(max_length=3, unique=True)
    iata = models.CharField(max_length=2, unique=True)
    name = models.CharField(max_length=100, unique=True)
    country = models.CharField(max_length=3)

    def __str__(self) -> str:
        return f"{self.name} ({self.icao}/{self.iata})"

    def fleet_size(self) -> int:
        return Equipment.objects.filter(owner=self).count()