from django.contrib import admin
from photologue.admin import GalleryAdmin as GalleryAdminDefault
from photologue.admin import PhotoAdmin as PhotoAdminDefault
from photologue.models import Gallery, Photo, PhotoEffect, PhotoSize, Watermark

from .models import GalleryExtended, PhotoExtended


class GalleryExtendedInline(admin.StackedInline):
    model = GalleryExtended
    can_delete = False


class GalleryAdmin(GalleryAdminDefault):
    """
    Define our new one-to-one model as an inline of Photologue's Gallery
    model.
    """
    inlines = [GalleryExtendedInline, ]


class PhotoExtendedInline(admin.StackedInline):
    model = PhotoExtended
    can_delete = True


class PhotoAdmin(PhotoAdminDefault):
    """
    Define our new one-to-one model as an inline of Photologue's Photo
    model.
    """
    inlines = [PhotoExtendedInline, ]


admin.site.unregister(Gallery)
admin.site.unregister(Photo)
admin.site.unregister(PhotoEffect)
admin.site.unregister(PhotoSize)
admin.site.unregister(Watermark)
admin.site.register(Gallery, GalleryAdmin)
admin.site.register(Photo, PhotoAdmin)