from django.contrib import admin
import nested_admin
from .models import *

# Register your models here.
class HourBuildingLegInline(nested_admin.NestedTabularInline):
    model = HourBuildingLeg
    extra = 1
    fk_name = 'hb'

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

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

class WeekPreferenceAdmin(nested_admin.NestedModelAdmin):
    list_display = ('week', 'student')
    list_filter = ['week', 'student']
    inlines = [TrainingInLIne, HourBuildingInLine]

class StudentAdmin(admin.ModelAdmin):
    list_display = ("name", "surname", "course", "active")
    list_filter = ["course", "active"]

class CourseAdmin(admin.ModelAdmin):
    list_filter = ["ctype"]

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