"""Establishment admin conf."""
from django.contrib import admin
from establishment import models
from django.contrib.contenttypes.admin import GenericTabularInline
from main.models import Award


@admin.register(models.EstablishmentType)
class EstablishmentTypeAdmin(admin.ModelAdmin):
    """EstablishmentType admin."""


@admin.register(models.EstablishmentSubType)
class EstablishmentSubTypeAdmin(admin.ModelAdmin):
    """EstablishmentSubType admin."""


class AwardInline(GenericTabularInline):
    model = Award
    extra = 0


@admin.register(models.Establishment)
class EstablishmentAdmin(admin.ModelAdmin):
    """Establishment admin."""
    inlines = [AwardInline, ]


@admin.register(models.EstablishmentSchedule)
class EstablishmentSchedule(admin.ModelAdmin):
    """Establishment schedule"""