42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
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, ]
|
|
autocomplete_fields = ['photos', ]
|
|
|
|
|
|
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)
|