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_display",
        "status_display",
    )

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

    @admin.display(description="Arrival", ordering="departure_time")
    def arrival_time_display(self, obj):
        return obj.arrival_time

    @admin.display(description="Status")
    def status_display(self, obj):
        return obj.status