from django.contrib import admin
from .models import Carrier, AircraftBase, Equipment, Aerodrome, Flight
# Register your models here.

admin.site.register(Carrier)
admin.site.register(AircraftBase)
admin.site.register(Equipment)
admin.site.register(Aerodrome)


@admin.register(Flight)
class FlightAdmin(admin.ModelAdmin):  # type: ignore[type-arg]
    list_display = (
        "carrier",
        "flight_number_display",
        "origin",
        "destination",
        "departure_time",
        "arrival_time",
        "status",
    )

    @admin.display(description="Flight")
    def flight_number_display(self, obj):
        return f"{obj.carrier.icao}{obj.flight_number}"