muistox/gallery/admin.py
2025-01-18 09:40:47 +02:00

75 lines
2.4 KiB
Python

from django.contrib import admin
from django.utils.html import format_html
from imagekit import ImageSpec
from imagekit.admin import AdminThumbnail
from imagekit.cachefiles import ImageCacheFile
from imagekit.processors import ResizeToFill
from gallery.models import Album, City, Location, Photo
class AdminThumbnailSpec(ImageSpec):
processors = [ResizeToFill(50, 50)]
format = 'JPEG'
options = {'quality': 60}
def cached_admin_thumb(instance):
cached = ImageCacheFile(AdminThumbnailSpec(instance.photo))
cached.generate()
return cached
class CityAdmin(admin.ModelAdmin):
list_display = ('name',)
search_fields = ('name',)
ordering = ('name',)
list_per_page = 30
class LocationAdmin(admin.ModelAdmin):
list_display = ('city', 'place')
list_filter = ('city',)
ordering = ('city__name',)
search_fields = ('place', 'city__name',)
list_per_page = 30
class AlbumAdmin(admin.ModelAdmin):
prepopulated_fields = {'slug': ('name',)}
list_display = ('__str__', 'location', 'album_date', 'is_public', 'thumbnail', )
search_fields = ('name',)
ordering = ('-album_date',)
list_per_page = 30
list_editable = ('is_public', 'location')
def formfield_for_foreignkey(self, db_field, request, **kwargs):
if db_field.name == "cover":
if hasattr(request, 'resolver_match') and request.resolver_match.kwargs.get('object_id'):
album_id = request.resolver_match.kwargs['object_id']
kwargs["queryset"] = Photo.objects.filter(album_id=album_id)
return super().formfield_for_foreignkey(db_field, request, **kwargs)
def thumbnail(self, obj):
if obj.cover and obj.cover.photo:
return format_html(
'<img src="{}" style="width: 50px; height: 50px; object-fit: cover;"/>',
obj.cover.photo.url,
)
return "-"
thumbnail.short_description = "Thumbnail"
class PhotoAdmin(admin.ModelAdmin):
list_display = ('slug', 'album', 'admin_thumbnail',)
list_display_links = ('slug',)
search_fields = ('slug',)
readonly_fields = ['slug', 'taken_at', 'height', 'width', 'exif', ]
admin_thumbnail = AdminThumbnail(image_field=cached_admin_thumb)
admin.site.register(City, CityAdmin)
admin.site.register(Location, LocationAdmin)
admin.site.register(Album, AlbumAdmin)
admin.site.register(Photo, PhotoAdmin)