summaryrefslogtreecommitdiffstats
path: root/packages/templatetags/package_extras.py
blob: 3cd2b91b8f9b3d4e28a8af0f4311041a1326c819 (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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import cgi, urllib
from django import template
from django.utils.html import escape

register = template.Library()

class BuildQueryStringNode(template.Node):
    def __init__(self, sortfield):
        self.sortfield = sortfield

    def render(self, context):
        qs = dict(cgi.parse_qsl(context['current_query'][1:]))
        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 '?' + urllib.urlencode(qs)

@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.tag
def userpkgs(parser, token):
    try:
        tagname, user = token.split_contents()
    except ValueError:
        raise template.TemplateSyntaxError(
                "%r tag requires a single argument" % tagname)
    return UserPkgsNode(user)

class UserPkgsNode(template.Node):
    def __init__(self, user):
        self.user = template.Variable(user)

    def render(self, context):
        try:
            real_user = self.user.resolve(context)
			# TODO don't hardcode
            title = escape('View packages maintained by ' + real_user.get_full_name())
            return '<a href="/packages/search/?maintainer=%s" title="%s">%s</a>' % (
					real_user.username,
                    title,
                    real_user.get_full_name(),
                    )
        except template.VariableDoesNotExist:
            return ''
        pass