Nested polymorphic hour building legs and stops added, needs restyling

cntmanage/flightslot/admins/hourbuilding_adm.py

@@ -7,32 +7,33 @@ from django.http import HttpRequest
 
 from durationwidget.widgets import TimeDurationWidget 
 
-from ..models.hourbuildings import HourBuilding, HourBuildingLeg
+from ..models.hourbuildings import HourBuilding, HourBuildingLegBase, HourBuildingLegFlight, HourBuildingLegStop
 from ..models.weekpref import WeekPreference
 
 from datetime import date
 
 class HourBuildingLegForm(forms.ModelForm):
     class Meta:
-        model = HourBuildingLeg
+        model = HourBuildingLegFlight
         fields = '__all__'
         widgets = {
             'time': TimeDurationWidget(show_days=False, 
-                                       show_seconds=False
-            )
+                                       show_seconds=False)
         }
 
 # Register your models here.
-class HourBuildingLegInline(nested_admin.NestedTabularInline):
-    model = HourBuildingLeg
-    form = HourBuildingLegForm
-    extra = 0
+class HourBuildingLegBaseInLine(nested_admin.NestedStackedPolymorphicInline):
+    model = HourBuildingLegBase
     fk_name = 'hb'
-    max_num = 5
-    formfield_overrides = {
-        models.CharField: {'widget': TextInput(attrs={'size':'20'})},
-        models.TextField: {'widget': Textarea(attrs={'rows':4, 'cols':35})},
-    }
+    verbose_name_plural = "Hour Building Legs"
+    
+    class HourBuildingLegFlightInLine(nested_admin.NestedStackedPolymorphicInline.Child):
+        model = HourBuildingLegFlight
+        form = HourBuildingLegForm
+    class HourBuildingLegStopInLine(nested_admin.NestedStackedPolymorphicInline.Child):
+        model = HourBuildingLegStop
+
+    child_inlines = (HourBuildingLegFlightInLine, HourBuildingLegStopInLine, )
 
     # If user is a student deny edit permission for week past the current one
     def has_change_permission(self, request: HttpRequest, obj: HourBuilding | None = None):
@@ -45,26 +46,16 @@ class HourBuildingLegInline(nested_admin.NestedTabularInline):
     def has_delete_permission(self, request: HttpRequest, obj: HourBuilding | None = None):
         return self.has_change_permission(request=request, obj=obj)
 
+
 class HourBuildingInLine(nested_admin.NestedTabularInline):
     model = HourBuilding
-    extra = 0
-    inlines = [HourBuildingLegInline]
+    extra = 1
+    inlines = (HourBuildingLegBaseInLine,)
     fk_name = 'weekpref'
-    verbose_name_plural = "Hour Building"
+    verbose_name_plural = "Hour Buildings"
     max_num = 7
     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):
-        if hasattr(request.user, 'student') and obj:
-            current_week = date.today().isocalendar().week
-            if current_week > obj.week:
-                return False
-        return True
-    
-    def has_delete_permission(self, request: HttpRequest, obj: WeekPreference | None = None):
-        return self.has_change_permission(request=request, obj=obj)