git.ozlabs.org Git - patchwork/blob - 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