52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.apps import apps
|
|
from django.conf import settings
|
|
from django.contrib.redirects.models import Redirect
|
|
from django.contrib.sites.models import get_current_site
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
from django import http
|
|
|
|
|
|
class RedirectFallbackMiddleware(object):
|
|
|
|
# Defined as class-level attributes to be subclassing-friendly.
|
|
response_gone_class = http.HttpResponseGone
|
|
response_redirect_class = http.HttpResponsePermanentRedirect
|
|
|
|
def __init__(self):
|
|
if not apps.is_installed('django.contrib.sites'):
|
|
raise ImproperlyConfigured(
|
|
"You cannot use RedirectFallbackMiddleware when "
|
|
"django.contrib.sites is not installed."
|
|
)
|
|
|
|
def process_response(self, request, response):
|
|
# No need to check for a redirect for non-404 responses.
|
|
if response.status_code != 404:
|
|
return response
|
|
|
|
full_path = request.get_full_path()
|
|
current_site = get_current_site(request)
|
|
|
|
r = None
|
|
try:
|
|
r = Redirect.objects.get(site=current_site, old_path=full_path)
|
|
except Redirect.DoesNotExist:
|
|
pass
|
|
if settings.APPEND_SLASH and not request.path.endswith('/'):
|
|
# Try appending a trailing slash.
|
|
path_len = len(request.path)
|
|
full_path = full_path[:path_len] + '/' + full_path[path_len:]
|
|
try:
|
|
r = Redirect.objects.get(site=current_site, old_path=full_path)
|
|
except Redirect.DoesNotExist:
|
|
pass
|
|
if r is not None:
|
|
if r.new_path == '':
|
|
return self.response_gone_class()
|
|
return self.response_redirect_class(r.new_path)
|
|
|
|
# No redirect was found. Return the response.
|
|
return response
|