from django.contrib import admin
from nested_inline.admin import NestedTabularInline, NestedModelAdmin
from .models import *

# Register your models here.
admin.site.register(Student, StudentAdmin)
admin.site.register(MissionProfile)

class HourBuildingLegInline(NestedTabularInline):
    model = HourBuildingLeg
    extra = 1
    fk_name = 'hb'

class HourBuildingInLine(NestedTabularInline):
    model = HourBuilding
    extra = 1
    inlines = [HourBuildingLegInline]
    fk_name = 'weekpref'

class TrainingInLIne(NestedTabularInline):
    model = Training
    extra = 0
    fk_name = 'weekpref'

class WeekPreferenceAdmin(NestedModelAdmin):
    inlines = [TrainingInLIne, HourBuildingInLine]

admin.site.register(WeekPreference, WeekPreferenceAdmin)