blob: ff66a9f23afeeb04b4eaa709e30b27a5dceed64b (
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
|
import cgi, urllib
from django import template
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])
|
