summaryrefslogtreecommitdiffstats
path: root/packages/templatetags/package_extras.py
blob: 8b55221ba501ad1cd4767b81844721a12a742d86 (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
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:
			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 />')