"""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
from review import models as review_models


@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


class ContactPhoneInline(admin.TabularInline):
    """Contact phone inline admin."""
    model = models.ContactPhone
    extra = 0


class ContactEmailInline(admin.TabularInline):
    """Contact email inline admin."""
    model = models.ContactEmail
    extra = 0


@admin.register(models.Contact)
class ContactAdmin(admin.ModelAdmin):
    """Contact admin."""
    inlines = [ContactPhoneInline, ContactEmailInline, ]


class ContactsInline(admin.TabularInline):
    model = models.Contact
    extra = 0


class ReviewInline(GenericTabularInline):
    model = review_models.Review
    extra = 0


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


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