1 from django import template
2 from django.conf import settings
3 from django.contrib.sites.models import Site
4 from django.core.urlresolvers import reverse, NoReverseMatch
5 from django.template.defaulttags import kwarg_re
6 from philo.exceptions import ViewCanNotProvideSubpath
9 register = template.Library()
12 class NodeURLNode(template.Node):
13 def __init__(self, node, as_var, with_obj=None, view_name=None, args=None, kwargs=None):
15 self.view_name = view_name
17 # Because the following variables have already been compiled as filters if they exist, they don't need to be re-scanned as template variables.
19 self.with_obj = with_obj
23 def render(self, context):
25 node = self.node.resolve(context)
27 node = context['node']
29 if self.with_obj is not None:
31 view_name, args, kwargs = node.view.get_reverse_params(self.with_obj.resolve(context))
32 except ViewCanNotProvideSubpath:
33 return settings.TEMPLATE_STRING_IF_INVALID
34 elif self.view_name is not None:
35 view_name = self.view_name
36 args = [arg.resolve(context) for arg in self.args]
37 kwargs = dict([(smart_str(k, 'ascii'), v.resolve(context)) for k, v in self.kwargs.items()])
39 return node.get_absolute_url()
41 if not node.view.accepts_subpath:
42 return settings.TEMPLATE_STRING_IF_INVALID
46 subpath = reverse(view_name, urlconf=node.view, args=args, kwargs=kwargs)
47 except NoReverseMatch:
48 if self.as_var is None:
54 url = node.get_absolute_url() + subpath
57 context[self.as_var] = url
63 @register.tag(name='node_url')
64 def do_node_url(parser, token):
66 {% node_url [for <node>] [as <var] %}
67 {% node_url with <obj> [for <node>] [as <var>] %}
68 {% node_url <view_name> [<arg1> [<arg2> ...] ] [for <node>] [as <var>] %}
69 {% node_url <view_name> [<key1>=<value1> [<key2>=<value2> ...] ] [for <node>] [as <var>]%}
71 params = token.split_contents()
78 if len(params) >= 2 and params[-2] == 'as':
82 if len(params) >= 2 and params[-2] == 'for':
83 node = parser.compile_filter(params[-1])
86 if len(params) >= 2 and params[-2] == 'with':
87 with_obj = parser.compile_filter(params[-1])
90 if with_obj is not None:
92 raise template.TemplateSyntaxError('`%s` template tag accepts no arguments or keyword arguments if with <obj> is specified.' % tag)
93 return NodeURLNode(with_obj=with_obj, node=node, as_var=as_var)
98 view_name = params.pop(0)
100 match = kwarg_re.match(param)
102 raise TemplateSyntaxError("Malformed arguments to `%s` tag" % tag)
103 name, value = match.groups()
105 kwargs[name] = parser.compile_filter(value)
107 args.append(parser.compile_filter(value))
108 return NodeURLNode(view_name=view_name, args=args, kwargs=kwargs, node=node, as_var=as_var)
110 return NodeURLNode(node=node, as_var=as_var)