-
Notifications
You must be signed in to change notification settings - Fork 0
/
widgets.py
executable file
·37 lines (33 loc) · 1.6 KB
/
widgets.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
from django.contrib.admin.widgets import AdminFileWidget
from django.utils.translation import ugettext as _
from django.utils.safestring import mark_safe
from django.conf import settings
from PIL import Image
import os
try:
from sorl.thumbnail.main import DjangoThumbnail
def thumbnail(image_path):
t = DjangoThumbnail(relative_source=image_path, requested_size=(200,200))
return u'<img src="%s" alt="%s" />' % (t.absolute_url, image_path)
except ImportError:
def thumbnail(image_path):
return u'<img src="%s%s" alt="%s" />' % (settings.MEDIA_URL, image_path, image_path)
class AdminImageWidget(AdminFileWidget):
"""
A FileField Widget that displays an image instead of a file path
if the current file is an image.
"""
def render(self, name, value, attrs=None):
output = []
file_name = str(value)
if file_name:
file_path = '%s%s' % (settings.MEDIA_URL, file_name)
try: # is image
Image.open(os.path.join(settings.MEDIA_ROOT, file_name))
output.append('<a target="_blank" href="%s">%s</a><br />%s <a target="_blank" href="%s">%s</a><br />%s ' % \
(file_path, thumbnail(file_name), _('Currently:'), file_path, file_name, _('Change:')))
except IOError: # not image
output.append('%s <a target="_blank" href="%s">%s</a> <br />%s ' % \
(_('Currently:'), file_path, file_name, _('Change:')))
output.append(super(AdminFileWidget, self).render(name, value, attrs))
return mark_safe(u''.join(output))