sommitrealweird/generic/templatetags/thumbnail.py

   1 from django import template
   2 from django.conf import settings
   3 from generic import helpers
   4 import sys
   5
   6 register = template.Library()
   7
   8 def thumbnail(image_url, args=''):
   9     options = {}
  10
  11     if ',' not in args:
  12         args = "%s," %(args,)
  13
  14     for arg in args.split(','):
  15         arg = arg.strip()
  16         try:
  17             (kw,val) = arg.split('=', 1)
  18             try:
  19                 options[kw] = int(val)
  20             except:
  21                 pass
  22         except:
  23             pass
  24
  25     if "height" in options or "width" in options:
  26         if "width" in options and "height" in options:
  27             return helpers.make_thumbnail(image_url[len(settings.MEDIA_URL):], width=options["width"], height=options["height"])
  28         elif "width" in options:
  29             return helpers.make_thumbnail(image_url[len(settings.MEDIA_URL):], width=options["width"])
  30         else:
  31             return helpers.make_thumbnail(image_url[len(settings.MEDIA_URL):], height=options["height"])
  32     else:
  33         pass
  34
  35     return "%s" %(image_url)
  36
  37 register.filter(thumbnail)