21 lines
784 B
Python
21 lines
784 B
Python
|
from django.conf import settings
|
||
|
|
from django.contrib.flatpages.views import flatpage
|
||
|
|
from django.http import Http404
|
||
|
|
from django.utils.deprecation import MiddlewareMixin
|
||
|
|
|
||
|
|
|
||
|
|
class FlatpageFallbackMiddleware(MiddlewareMixin):
|
||
|
|
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 Exception:
|
||
|
|
if settings.DEBUG:
|
||
|
|
raise
|
||
|
|
return response
|