diff options
| -rw-r--r-- | src/meetingtools/urlmiddleware.py | 126 |
1 files changed, 126 insertions, 0 deletions
diff --git a/src/meetingtools/urlmiddleware.py b/src/meetingtools/urlmiddleware.py new file mode 100644 index 0000000..3d77caa --- /dev/null +++ b/src/meetingtools/urlmiddleware.py @@ -0,0 +1,126 @@ +""" +URL Middleware +Stefano J. Attardi (attardi.org) + +$Id$ +$URL$ + +Cleans up urls by adding/removing trailing slashes, adding/removing +the www. prefix, and allowing the language to be set from the url. + +If APPEND_SLASH is set to False, trailing slashes are removed from the +urls, except for urls which have an explicit trailing slash in +urls.py, in which case a trailing slash is added. + +If REMOVE_WWW is set to True, the www. prefix is removed. + +Finally, ?lang=xx can be appended to any url to override the default +language setting. This override is remembered for the following +requests. For example, /article?lang=it would show the article in +Italian regardless of brower settings or cookies, and any following +request to the site would be shown in Italian by default. + +Changelog + +1.3.2 +Fixed an indentation issue. Added a check for those backends +which set an empty path (e.g. runfcgi). + +1.3.1 +Added support for running in a test +suite (doesn't assume that HTTP_HOST is set) + +1.3 +Only use sessions for the language preference if the session +cookie has already been set (regardless of whether session middleware +is active). Otherwise use the plain django_language cookie. +Only import the FlatPages module if it is active. + +1.2 +Added support for FlatPages. +Switched to Django's resolve function (with workaround for when it +returns None). + +1.1 +Various bugfixes. + +1.0 +First release. +""" +__version__ = "1.3.2" +__license__ = "Python" +__copyright__ = "Copyright (C) 2006-2007, Stefano J. Attardi" +__author__ = "Stefano J. Attardi <http://attardi.org/>" +__contributors__ = ["Antonio Cavedoni <http://cavedoni.com/>"] + +from django.conf import settings +from django.http import HttpResponseRedirect, Http404 +from django.core.urlresolvers import resolve +from django.utils.translation import check_for_language +import os + +class UrlMiddleware: + + def process_request(self, request): + + # Change the language setting for the current page + if "lang" in request.GET and check_for_language(request.GET["lang"]): + if hasattr(request, "session"): + request.session["django_language"] = request.GET["lang"] + else: + request.COOKIES["django_language"] = request.GET["lang"] + + # work-around for runfcgi + if request.path == "": request.path = "/" + request.path = '/'+request.path.lstrip('/') + + # Check for a redirect based on settings.APPEND_SLASH and settings.PREPEND_WWW + old_url = [request.META.get("HTTP_HOST", "localhost"), request.path] + new_url = old_url[:] + + # if REMOVE_WWW is True, remove the www. from the urls if necessary + if hasattr(settings, "REMOVE_WWW") and settings.REMOVE_WWW and old_url[0].startswith("www."): + new_url[0] = old_url[0][4:] + + if hasattr(settings, "APPEND_SLASH") and not settings.APPEND_SLASH: + # if the url is not found, try with(out) the trailing slash + if old_url[1] != "/" and not self._urlExists(old_url[1]): + + if old_url[1][-1] == "/": + other = old_url[1][:-1] + else: + other = old_url[1] + "/" + + if self._urlExists(other): + new_url[1] = other + + if new_url != old_url: + # Redirect + newurl = "%s://%s%s" % (os.environ.get("HTTPS") == "on" and "https" or "http", new_url[0], new_url[1]) + if request.GET: + newurl += "?" + request.GET.urlencode() + + return HttpResponseRedirect(newurl) + + return None + + def process_response(self, request, response): + + # Change the language setting for future pages + if "lang" in request.GET and check_for_language(request.GET["lang"]): + if "sessionid" in request.COOKIES: + request.session["django_language"] = request.GET["lang"] + else: + response.set_cookie("django_language", request.GET["lang"]) + + return response + + def _urlExists(self, path): + try: + if resolve(path) is None: raise Http404 # None?!? You mean 404... + return True + except Http404: + # check for flatpages + if "django.contrib.flatpages.middleware.FlatpageFallbackMiddleware" in settings.MIDDLEWARE_CLASSES: + from django.contrib.flatpages.models import FlatPage + return FlatPage.objects.filter(url=path, sites__id=settings.SITE_ID).count() == 1 |