from django.contrib import admin
from django.http import HttpRequest

from .models.aircrafts import Aircraft
from .models.courses import Course
from .models.students import Student
from .models.missions import MissionProfile
from .models.weekpref import WeekPreference

from .admins.aircraft_adm import AircraftAdmin
from .admins.course_adm import CourseAdmin
from .admins.student_adm import StudentAdmin
from .admins.mission_adm import MissionProfileAdmin
from .admins.weekpref_adm import WeekPreferenceAdmin

from django.contrib.admin import AdminSite

class FlightSlotUserSite(AdminSite):
    site_header = "Flight Scheduler 🛫"
    site_title = "Flight Scheduler 🛫"
    index_title = "Welcome to CantorAir Flight Scheduler Portal"

    def get_app_list(self, request: HttpRequest, *args, **kwargs):
        app_list = super().get_app_list(request)
        
        if not request.user.is_superuser:
            self.enable_nav_sidebar = False
        
        return app_list

flightslot_user = FlightSlotUserSite(name="user_site")

# registra SOLO i modelli autorizzati
flightslot_user.register(WeekPreference, WeekPreferenceAdmin)

admin.site.site_header = "Flight Scheduler Admin 🛫"
admin.site.site_title = "Flight Scheduler Admin 🛫"
admin.site.index_title = "Welcome to CantorAir Flight Scheduler Administrator Portal"

admin.site.register(Aircraft, AircraftAdmin)
admin.site.register(Course, CourseAdmin)
admin.site.register(MissionProfile, MissionProfileAdmin)
admin.site.register(Student, StudentAdmin)
admin.site.register(WeekPreference, WeekPreferenceAdmin)