X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/bd4e85c2780dc58b34a16b4f2137723e68ba5a07..8fff8672235eafa1cf55739d460119b3be5faf5f:/middleware.py?ds=sidebyside diff --git a/middleware.py b/middleware.py index ad660ec..5bcd569 100644 --- a/middleware.py +++ b/middleware.py @@ -15,12 +15,27 @@ class LazyNode(object): except Site.DoesNotExist: current_site = None + path = request._cached_node_path + trailing_slash = False + if path[-1] == '/': + trailing_slash = True + try: - node, subpath = Node.objects.get_with_path(request._cached_node_path, root=getattr(current_site, 'root_node', None), absolute_result=False) + node, subpath = Node.objects.get_with_path(path, root=getattr(current_site, 'root_node', None), absolute_result=False) except Node.DoesNotExist: node = None + else: + if subpath and not node.handles_subpath(subpath): + node = None if node: + if subpath is None: + subpath = "" + subpath = "/" + subpath + + if trailing_slash and subpath[-1] != "/": + subpath += "/" + node.subpath = subpath request._found_node = node @@ -34,7 +49,10 @@ class RequestNodeMiddleware(object): request.__class__.node = LazyNode() def process_view(self, request, view_func, view_args, view_kwargs): - request._cached_node_path = view_kwargs.get('path', '/') + try: + request._cached_node_path = view_kwargs['path'] + except KeyError: + pass def process_exception(self, request, exception): if settings.DEBUG or not hasattr(request, 'node') or not request.node: