45 lines
1,022 B
Python
45 lines
1,022 B
Python
from django import forms
|
|
from django.contrib import admin
|
|
|
|
from .models import Event, EventRegistration, TimeSlot
|
|
|
|
|
|
class TimeSlotFormSet(forms.BaseInlineFormSet):
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs["initial"] = [
|
|
{"time": TimeSlot.EVENING}
|
|
]
|
|
super().__init__(*args, **kwargs)
|
|
|
|
|
|
class RegistrationFormSet(forms.BaseInlineFormSet):
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs["initial"] = [
|
|
{"state": EventRegistration.ADMITTED}
|
|
]
|
|
super().__init__(*args, **kwargs)
|
|
|
|
|
|
class TimeSlotInline(admin.StackedInline):
|
|
model = TimeSlot
|
|
extra = 1
|
|
formset = TimeSlotFormSet
|
|
|
|
|
|
class RegistrationInline(admin.TabularInline):
|
|
model = EventRegistration
|
|
extra = 1
|
|
formset = RegistrationFormSet
|
|
#
|
|
|
|
class EventAdmin(admin.ModelAdmin):
|
|
inlines = [TimeSlotInline]
|
|
|
|
|
|
class TimeSlotAdmin(admin.ModelAdmin):
|
|
inlines = [RegistrationInline]
|
|
|
|
|
|
admin.site.register(Event, EventAdmin)
|
|
admin.site.register(TimeSlot, TimeSlotAdmin)
|