middleware.py

   1 # vim:ts=2:sw=2:et
   2 #
   3 # Django Default 403.html handler
   4 # http://wtanaka.com/django/django403
   5 #
   6 # Copyright (C) 2009 Wesley Tanaka <http://wtanaka.com/>
   7 #
   8 # This program is free software: you can redistribute it and/or modify
   9 # it under the terms of the GNU General Public License as published by
  10 # the Free Software Foundation, either version 3 of the License, or
  11 # (at your option) any later version.
  12 #
  13 # This program is distributed in the hope that it will be useful,
  14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 # GNU General Public License for more details.
  17 #
  18 # You should have received a copy of the GNU General Public License
  19 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20
  21 import django.http
  22 import django.template
  23
  24 class Django403Middleware(object):
  25   """Replaces vanilla django.http.HttpResponseForbidden() responses
  26   with a rendering of 403.html
  27   """
  28   def process_response(self, request, response):
  29     # If the response object is a vanilla 403 constructed with
  30     # django.http.HttpResponseForbidden() then call our custom 403 view
  31     # function
  32     if isinstance(response, django.http.HttpResponseForbidden) and \
  33         set(dir(response)) == set(dir(django.http.HttpResponseForbidden())):
  34       import views
  35       try:
  36         return views.access_denied(request)
  37       except django.template.TemplateDoesNotExist, e:
  38         return views.fallback_403(request)
  39
  40     return response