Author: mtredinnick
Date: 2008-10-07 03:26:31 -0500 (Tue, 07 Oct 2008)
New Revision: 9185

Modified:
   django/branches/releases/1.0.X/django/middleware/common.py
   django/branches/releases/1.0.X/tests/regressiontests/middleware/tests.py
Log:
[1.0.X] Fixed #9199 -- We were erroneously only prepending "www" to the domain
if we also needed to append a slash (when PREPEND_WWW=True).

Based on a patch and tests from gonz. Thanks.

Backport of r9184 from trunk.


Modified: django/branches/releases/1.0.X/django/middleware/common.py
===================================================================
--- django/branches/releases/1.0.X/django/middleware/common.py  2008-10-07 
08:22:50 UTC (rev 9184)
+++ django/branches/releases/1.0.X/django/middleware/common.py  2008-10-07 
08:26:31 UTC (rev 9185)
@@ -53,9 +53,8 @@
         # Append a slash if APPEND_SLASH is set and the URL doesn't have a
         # trailing slash and there is no pattern for the current path
         if settings.APPEND_SLASH and (not old_url[1].endswith('/')):
-            try:
-                urlresolvers.resolve(request.path_info)
-            except urlresolvers.Resolver404:
+            if (not _is_valid_path(request.path_info) and
+                    _is_valid_path("%s/" % request.path_info)):
                 new_url[1] = new_url[1] + '/'
                 if settings.DEBUG and request.method == 'POST':
                     raise RuntimeError, (""
@@ -66,25 +65,19 @@
                     "slash), or set APPEND_SLASH=False in your Django "
                     "settings.") % (new_url[0], new_url[1])
 
-        if new_url != old_url:
-            # Redirect if the target url exists
-            try:
-                urlresolvers.resolve("%s/" % request.path_info)
-            except urlresolvers.Resolver404:
-                pass
-            else:
-                if new_url[0]:
-                    newurl = "%s://%s%s" % (
-                        request.is_secure() and 'https' or 'http',
-                        new_url[0], urlquote(new_url[1]))
-                else:
-                    newurl = urlquote(new_url[1])
-                if request.GET:
-                    newurl += '?' + request.META['QUERY_STRING']
-                return http.HttpResponsePermanentRedirect(newurl)
+        if new_url == old_url:
+            # No redirects required.
+            return
+        if new_url[0]:
+            newurl = "%s://%s%s" % (
+                request.is_secure() and 'https' or 'http',
+                new_url[0], urlquote(new_url[1]))
+        else:
+            newurl = urlquote(new_url[1])
+        if request.GET:
+            newurl += '?' + request.META['QUERY_STRING']
+        return http.HttpResponsePermanentRedirect(newurl)
 
-        return None
-
     def process_response(self, request, response):
         "Check for a flat page (for 404s) and calculate the Etag, if needed."
         if response.status_code == 404:
@@ -119,7 +112,9 @@
         return response
 
 def _is_ignorable_404(uri):
-    "Returns True if a 404 at the given URL *shouldn't* notify the site 
managers"
+    """
+    Returns True if a 404 at the given URL *shouldn't* notify the site 
managers.
+    """
     for start in settings.IGNORABLE_404_STARTS:
         if uri.startswith(start):
             return True
@@ -129,6 +124,23 @@
     return False
 
 def _is_internal_request(domain, referer):
-    "Return true if the referring URL is the same domain as the current 
request"
+    """
+    Returns true if the referring URL is the same domain as the current 
request.
+    """
     # Different subdomains are treated as different domains.
     return referer is not None and re.match("^https?://%s/" % 
re.escape(domain), referer)
+
+def _is_valid_path(path):
+    """
+    Returns True if the given path resolves against the default URL resolver,
+    False otherwise.
+
+    This is a convenience method to make working with "is this a match?" cases
+    easier, avoiding unnecessarily indented try...except blocks.
+    """
+    try:
+        urlresolvers.resolve(path)
+        return True
+    except urlresolvers.Resolver404:
+        return False
+

Modified: 
django/branches/releases/1.0.X/tests/regressiontests/middleware/tests.py
===================================================================
--- django/branches/releases/1.0.X/tests/regressiontests/middleware/tests.py    
2008-10-07 08:22:50 UTC (rev 9184)
+++ django/branches/releases/1.0.X/tests/regressiontests/middleware/tests.py    
2008-10-07 08:26:31 UTC (rev 9185)
@@ -89,3 +89,31 @@
         self.assertEquals(
             r['Location'],
             'http://testserver/middleware/needsquoting%23/')
+
+    def test_prepend_www(self):
+        settings.PREPEND_WWW = True
+        settings.APPEND_SLASH = False
+        request = self._get_request('path/')
+        r = CommonMiddleware().process_request(request)
+        self.assertEquals(r.status_code, 301)
+        self.assertEquals(
+            r['Location'],
+            'http://www.testserver/middleware/path/')
+
+    def test_prepend_www_append_slash_have_slash(self):
+        settings.PREPEND_WWW = True
+        settings.APPEND_SLASH = True
+        request = self._get_request('slash/')
+        r = CommonMiddleware().process_request(request)
+        self.assertEquals(r.status_code, 301)
+        self.assertEquals(r['Location'],
+                          'http://www.testserver/middleware/slash/')
+
+    def test_prepend_www_append_slash_slashless(self):
+        settings.PREPEND_WWW = True
+        settings.APPEND_SLASH = True
+        request = self._get_request('slash')
+        r = CommonMiddleware().process_request(request)
+        self.assertEquals(r.status_code, 301)
+        self.assertEquals(r['Location'],
+                          'http://www.testserver/middleware/slash/')


--~--~---------~--~----~------------~-------~--~----~
You received this message because you are subscribed to the Google Groups 
"Django updates" group.
To post to this group, send email to [email protected]
To unsubscribe from this group, send email to [EMAIL PROTECTED]
For more options, visit this group at 
http://groups.google.com/group/django-updates?hl=en
-~----------~----~----~----~------~----~------~--~---

Reply via email to