Inital commit
[patchwork] / apps / patchwork / templatetags / pwurl.py
1 # Patchwork - automated patch tracking system
2 # Copyright (C) 2008 Jeremy Kerr <jk@ozlabs.org>
3 #
4 # This file is part of the Patchwork package.
5 #
6 # Patchwork is free software; you can redistribute it and/or modify
7 # it under the terms of the GNU General Public License as published by
8 # the Free Software Foundation; either version 2 of the License, or
9 # (at your option) any later version.
10 #
11 # Patchwork is distributed in the hope that it will be useful,
12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 # GNU General Public License for more details.
15 #
16 # You should have received a copy of the GNU General Public License
17 # along with Patchwork; if not, write to the Free Software
18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19
20 from django import template
21 from django.utils.html import escape
22 from django.utils.safestring import mark_safe
23 from patchwork.filters import filterclasses
24 import re
25
26 register = template.Library()
27
28 # params to preserve across views
29 list_params = [ c.param for c in filterclasses ] + ['order', 'page']
30
31 class ListURLNode(template.defaulttags.URLNode):
32     def __init__(self, *args, **kwargs):
33         super(ListURLNode, self).__init__(*args, **kwargs)
34         self.params = {}
35         for (k, v) in kwargs:
36             if k in list_params:
37                 self.params[k] = v
38
39     def render(self, context):
40         self.view_name = template.Variable('list_view.view')
41         str = super(ListURLNode, self).render(context)
42         if str == '':
43             return str
44         params = []
45         try:
46             qs_var = template.Variable('list_view.params')
47             params = dict(qs_var.resolve(context))
48         except Exception:
49             pass
50
51         params.update(self.params)
52
53         if not params:
54             return str
55
56         return str + '?' + '&'.join(['%s=%s' % (k, escape(v)) \
57                         for (k, v) in params.iteritems()])
58
59 @register.tag
60 def listurl(parser, token):
61     bits = token.contents.split(' ', 1)
62     if len(bits) < 1:
63         raise TemplateSyntaxError("'%s' takes at least one argument"
64                                   " (path to a view)" % bits[0])
65     args = ['']
66     kwargs = {}
67     if len(bits) > 1:
68         for arg in bits[2].split(','):
69             if '=' in arg:
70                 k, v = arg.split('=', 1)
71                 k = k.strip()
72                 kwargs[k] = parser.compile_filter(v)
73             else:
74                 args.append(parser.compile_filter(arg))
75     return PatchworkURLNode(bits[1], args, kwargs)
76