app/django/contrib/flatpages/middleware.py
author Pawel Solyga <Pawel.Solyga@gmail.com>
Thu, 21 May 2009 13:25:11 +0200
changeset 2330 068540d91bde
parent 323 ff1a9aa48cfd
permissions -rw-r--r--
Now clicking cancel button returns user to previous page. Fixes issue 569. If you don't want that behaviour you can define cancel_redirect and user will be redirected to given url on "Cancel" button click instead of redirecting to previous page. Patch by: Daniel Hans & Pawel Solyga

from django.contrib.flatpages.views import flatpage
from django.http import Http404
from django.conf import settings

class FlatpageFallbackMiddleware(object):
    def process_response(self, request, response):
        if response.status_code != 404:
            return response # No need to check for a flatpage for non-404 responses.
        try:
            return flatpage(request, request.path_info)
        # Return the original response if any errors happened. Because this
        # is a middleware, we can't assume the errors will be caught elsewhere.
        except Http404:
            return response
        except:
            if settings.DEBUG:
                raise
            return response