from django.contrib import admin
from polymorphic.admin import (
    PolymorphicParentModelAdmin,
    PolymorphicChildModelAdmin,
    PolymorphicChildModelFilter,
)

from .models import Content, Image, ImageText, Text, Video


class ContentChildAdmin(PolymorphicChildModelAdmin):
    base_model = Content


@admin.register(Image)
class ImageAdmin(ContentChildAdmin):
    base_model = Image


@admin.register(Text)
class TextAdmin(ContentChildAdmin):
    base_model = Text


@admin.register(ImageText)
class ImageTextAdmin(ContentChildAdmin):
    base_model = ImageText


@admin.register(Video)
class VideoAdmin(ContentChildAdmin):
    base_model = Video


@admin.register(Content)
class ContentAdmin(PolymorphicParentModelAdmin):
    base_model = Content
    child_models = (
        Image,
        Text,
        ImageText,
        Video
    )