summaryrefslogtreecommitdiff
path: root/packages/templatetags/package_extras.py
blob: 5a92565ca053db1f9f15789f1eb1aa2b249a006f (plain)
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
37
38
39
40
41
42
43
from django import template

register = template.Library()

class BuildQueryStringNode(template.Node):
    def __init__(self, sortfield):
        self.sortfield = sortfield
    def render(self, context):
        qs = context['querystring'].copy()
        if qs.has_key('sort') and qs['sort'] == self.sortfield:
            if self.sortfield.startswith('-'):
                qs['sort'] = self.sortfield[1:]
            else:
                qs['sort'] = '-' + self.sortfield
        else:
            qs['sort'] = self.sortfield
        return '?' + qs.urlencode()

@register.tag(name='buildsortqs')
def do_buildsortqs(parser, token):
    try:
        tagname, sortfield = token.split_contents()
    except ValueError:
        raise template.TemplateSyntaxError, "%r tag requires a single argument" % tagname
    if not (sortfield[0] == sortfield[-1] and sortfield[0] in ('"', "'")):
        raise template.TemplateSyntaxError, "%r tag's argument should be in quotes" % tagname
    return BuildQueryStringNode(sortfield[1:-1])

@register.filter(name='space2br')
def space2br(value):
    return value.replace(' ', '<br />')

@register.inclusion_tag('errors.html')
def print_errors(errors):
    errs = []
    for e,msg in errors.iteritems():
        errmsg = str(msg[0])
        # hack -- I'm a python idiot
        errs.append( (e, errmsg[2:-2]) )
    return {'errors': errs}

# vim: set ts=4 sw=4 et: