packages/templatetags/package_extras.py

   1 from django import template
   2
   3 register = template.Library()
   4
   5 class BuildQueryStringNode(template.Node):
   6         def __init__(self, sortfield):
   7                 self.sortfield = sortfield
   8         def render(self, context):
   9                 qs = context['querystring'].copy()
  10                 if qs.has_key('sort') and qs['sort'] == self.sortfield:
  11                         qs['sort'] = '-' + self.sortfield
  12                 else:
  13                         qs['sort'] = self.sortfield
  14                 return '?' + qs.urlencode()
  15
  16 @register.tag(name='buildsortqs')
  17 def do_buildsortqs(parser, token):
  18         try:
  19                 tagname, sortfield = token.split_contents()
  20         except ValueError:
  21                 raise template.TemplateSyntaxError, "%r tag requires a single argument" % tagname
  22         if not (sortfield[0] == sortfield[-1] and sortfield[0] in ('"', "'")):
  23                 raise template.TemplateSyntaxError, "%r tag's argument should be in quotes" % tagname
  24         return BuildQueryStringNode(sortfield[1:-1])
  25
  26 @register.filter(name='space2br')
  27 def space2br(value):
  28         return value.replace(' ', '<br />')