from django.contrib import admin
from . import models


@admin.register(models.RotaSchedule)
class RotaScheduleAdmin(admin.ModelAdmin):
	list_display = ("name", "start_date", "end_date", "created_at")
	search_fields = ("name",)


@admin.register(models.Worker)
class WorkerAdmin(admin.ModelAdmin):
	list_display = ("name", "email", "site", "grade", "fte", "active")
	list_filter = ("site", "active")
	search_fields = ("name", "email")


@admin.register(models.Assignment)
class AssignmentAdmin(admin.ModelAdmin):
	list_display = ("worker", "rota", "role", "created_at")
	list_select_related = ("worker", "rota")


@admin.register(models.Leave)
class LeaveAdmin(admin.ModelAdmin):
	list_display = ("worker", "start_date", "end_date", "reason")
	list_filter = ("start_date", "end_date")
	search_fields = ("worker__name",)


@admin.register(models.RotaRun)
class RotaRunAdmin(admin.ModelAdmin):
	list_display = ("rota", "status", "created_at", "started_at", "finished_at")
	list_filter = ("status", "created_at")
	readonly_fields = ("log", "result")