view user_photos/admin.py @ 887:9a15f7c27526

Actually save model object upon change. This commit was tested on the comments model. Additional logging added. Added check for Markdown image references. Added TODOs after observing behavior on comments.
author Brian Neal <bgneal@gmail.com>
date Tue, 03 Feb 2015 21:09:44 -0600
parents e0e79451339d
children
line wrap: on
line source
"""Admin definitions for the user_photos application."""
from django.contrib import admin

from user_photos.models import Photo
from user_photos.s3 import delete_photos

IMG_TAG = """<a href="{url}"><img src="{thumb_url}" alt="thumbnail" /></a>"""

class PhotoAdmin(admin.ModelAdmin):
    date_hierarchy = 'upload_date'
    ordering = ['-upload_date']
    raw_id_fields = ['user']
    search_fields = ['user__username', 'user__email']
    list_display = ['__unicode__', 'thumbnail']
    actions = ['custom_delete']

    def thumbnail(self, obj):
        return IMG_TAG.format(url=obj.url, thumb_url=obj.thumb_url)
    thumbnail.allow_tags = True

    def custom_delete(self, request, qs):
        """Custom delete in order to remove images from the S3 bucket in
        addition to removing from the database.

        """
        delete_photos(qs)
        count = len(qs)
        qs.delete()

        if count == 1:
            msg_bit = "1 photo was"
        else:
            msg_bit = "{} photos were".format(count)

        self.message_user(request, "{} successfully deleted.".format(msg_bit))
    custom_delete.short_description = "Delete selected photos from DB & S3"

    def get_actions(self, request):
        """Remove the default delete selected action because we have installed
        our own.

        """
        actions = super(PhotoAdmin, self).get_actions(request)
        del actions['delete_selected']
        return actions


admin.site.register(Photo, PhotoAdmin)