extended photologue photo to add an owner field

This commit is contained in:
aeltheos 2021-10-11 20:40:48 +02:00
parent e91e79ad85
commit 7c0ba18824
4 changed files with 86 additions and 16 deletions

View file

@ -1,8 +1,9 @@
from django.contrib import admin
from photologue.admin import GalleryAdmin as GalleryAdminDefault
from photologue.models import Gallery
from photologue.admin import PhotoAdmin as PhotoAdminDefault
from photologue.models import Gallery, Photo
from .models import GalleryExtended
from .models import GalleryExtended, PhotoExtended
class GalleryExtendedInline(admin.StackedInline):
@ -18,5 +19,20 @@ class GalleryAdmin(GalleryAdminDefault):
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.register(Gallery, GalleryAdmin)
admin.site.register(Photo, PhotoAdmin)