middleware.py

   1 from django.conf import settings
   2 from django.contrib.sites.models import Site
   3 from django.http import Http404
   4 from philo.models import Node, View
   5
   6
   7 class LazyNode(object):
   8         def __get__(self, request, obj_type=None):
   9                 if not hasattr(request, '_cached_node_path'):
  10                         return None
  11
  12                 if not hasattr(request, '_found_node'):
  13                         try:
  14                                 current_site = Site.objects.get_current()
  15                         except Site.DoesNotExist:
  16                                 current_site = None
  17
  18                         try:
  19                                 node, subpath = Node.objects.get_with_path(request._cached_node_path, root=getattr(current_site, 'root_node', None), absolute_result=False)
  20                         except Node.DoesNotExist:
  21                                 node = None
  22
  23                         if subpath is None:
  24                                 subpath = ""
  25                         subpath = "/" + subpath
  26
  27                         if node:
  28                                 node.subpath = subpath
  29
  30                         request._found_node = node
  31
  32                 return request._found_node
  33
  34
  35 class RequestNodeMiddleware(object):
  36         """Middleware to process the request's path and attach the closest ancestor node."""
  37         def process_request(self, request):
  38                 request.__class__.node = LazyNode()
  39
  40         def process_view(self, request, view_func, view_args, view_kwargs):
  41                 request._cached_node_path = view_kwargs.get('path', '/')
  42
  43         def process_exception(self, request, exception):
  44                 if settings.DEBUG or not hasattr(request, 'node') or not request.node:
  45                         return
  46
  47                 if isinstance(exception, Http404):
  48                         error_view = request.node.attributes.get('Http404', None)
  49                 else:
  50                         error_view = request.node.attributes.get('Http500', None)
  51
  52                 if error_view is None or not isinstance(error_view, View):
  53                         # Should this be duck-typing? Perhaps even no testing?
  54                         return
  55
  56                 extra_context = {'exception': exception}
  57                 return error_view.render_to_response(request, extra_context)