"""Establishment admin conf."""
from django.contrib import admin
from django.contrib.contenttypes.admin import GenericTabularInline

from establishment import models
from main.models import Award, MetaDataContent


@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


class MetaDataContentInline(GenericTabularInline):
    model = MetaDataContent
    extra = 0


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


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


@admin.register(models.Comment)
class EstablishmentComment(admin.ModelAdmin):
    """Establishment comments."""