view potd/admin.py @ 973:6f55c086db1e

Guess file extension based on content-type. When downloading a file, and no path is supplied to store it, guess the file extension using mimetypes and the content-type header. Also supply a unit test for the HotLinkImageForm.
author Brian Neal <bgneal@gmail.com>
date Thu, 01 Oct 2015 19:44:45 -0500
parents ee87ea74d46b
children
line wrap: on
line source
"""
This file contains the admin definitions for the POTD application.
"""
from django.contrib import admin
from django.conf import settings

from potd.models import Photo
from potd.models import Current
from potd.models import Sequence


IMG_TAG = '<img src="%s" alt="thumbnail" />'


class PhotoAdmin(admin.ModelAdmin):
    fields = ['photo', 'caption', 'description', 'user', 'potd_count']
    raw_id_fields = ['user']
    list_display = ['__unicode__', 'thumbnail']
    actions = ['regen_thumbnail']
    search_fields = ['caption', 'description']

    class Media:
        js = settings.GPP_THIRD_PARTY_JS['tiny_mce']

    def thumbnail(self, obj):
        return IMG_TAG % obj.thumb.url
    thumbnail.allow_tags = True

    def regen_thumbnail(self, request, qs):
        """
        Regenerates the thumbnail for the selected photos.
        """
        for photo in qs:
            photo.generate_thumb()
            photo.save()

    regen_thumbnail.short_description = "Regenerate thumbs for selected photos"


class CurrentAdmin(admin.ModelAdmin):
    list_display = ('__unicode__', 'thumbnail')
    raw_id_fields = ('potd', )

    def thumbnail(self, obj):
        return IMG_TAG % obj.potd.thumb.url
    thumbnail.allow_tags = True


admin.site.register(Photo, PhotoAdmin)
admin.site.register(Current, CurrentAdmin)
admin.site.register(Sequence)