import nested_admin

from django import forms
from django.db import models
from django.forms import TextInput, Textarea
from django.http import HttpRequest

from durationwidget.widgets import TimeDurationWidget 

from ..models.hourbuildings import HourBuilding, HourBuildingLegBase, HourBuildingLegFlight, HourBuildingLegStop
from ..models.weekpref import WeekPreference
from ..models.students import Student

from datetime import date

class HourBuildingLegFlightForm(forms.ModelForm):
    class Meta:
        model = HourBuildingLegFlight
        fields = "__all__"
        widgets = {
            "time": TimeDurationWidget(show_days=False, 
                                       show_seconds=False,
                                       attrs={
            "style": (
                "margin-right:5px; margin-left:5px; width:40px; min:0; max:5"
            )
        })
        }

class HourBuildingLegStopForm(forms.ModelForm):
    class Meta:
        model = HourBuildingLegStop
        fields = "__all__"
        widgets = {
            "time": TimeDurationWidget(show_days=False, 
                                       show_seconds=False,
                                       attrs={
            "style": (
                "margin-right:5px; margin-left:5px; width:40px;"
            )
        })
        }

# Register your models here.
class HourBuildingLegBaseInLine(nested_admin.NestedStackedPolymorphicInline):
    model = HourBuildingLegBase
    fk_name = "hb"
    verbose_name_plural = "Hour Building Legs"
    
    class HourBuildingLegFlightInLine(nested_admin.NestedStackedPolymorphicInline.Child):
        model = HourBuildingLegFlight
        form = HourBuildingLegFlightForm
        fk_name = "hourbuildinglegbase_ptr"
        fields = ("departure", "time", "destination", "pax", )
            
    class HourBuildingLegStopInLine(nested_admin.NestedStackedPolymorphicInline.Child):
        model = HourBuildingLegStop
        form = HourBuildingLegFlightForm
        fk_name = "hourbuildinglegbase_ptr"
        fields = ("time", "refuel", )

    child_inlines = (HourBuildingLegFlightInLine, HourBuildingLegStopInLine, )

class HourBuildingInLine(nested_admin.NestedTabularInline):
    model = HourBuilding
    inlines = (HourBuildingLegBaseInLine,)
    extra = 0
    max_num = 7
    fk_name = "weekpref"
    verbose_name_plural = "Hour Buildings"
    formfield_overrides = {
        models.CharField: {"widget": TextInput(attrs={"size":"20"})},
        models.TextField: {"widget": Textarea(attrs={"rows":4, "cols":35})},
    }

    # If user is a student deny edit permission for week past the current one
    def has_change_permission(self, request: HttpRequest, obj: WeekPreference | None = None) -> bool:
        if hasattr(request.user, 'student') and obj:
            student: Student = request.user.student
            current_week: int = date.today().isocalendar().week
            if current_week > obj.week or not student.active:
                return False
        return True
    
    def has_delete_permission(self, request: HttpRequest, obj: WeekPreference | None = None) -> bool:
        return self.has_change_permission(request=request, obj=obj)
    
    def has_add_permission(self, request: HttpRequest, obj: WeekPreference | None = None) -> bool:
        return self.has_change_permission(request=request, obj=obj)