from django.contrib import admin
from django.db.models.query import QuerySet
from django.http import HttpRequest

from ..models.aircrafts import Aircraft
     
class AircraftAdmin(admin.ModelAdmin):
    model = Aircraft
    list_display = ("type", "markings", "avail_hours", "complex", )
    list_filter = ("type", )
    actions = ("reset_maint")

    def get_queryset(self, request: HttpRequest) -> QuerySet[Aircraft]:
        qs: QuerySet[Aircraft] = super().get_queryset(request)
        qs.order_by("type", "markings")
        return qs