amk's fix attached to

Michael W. Hudson · Michael W. Hudson · commit bd3e771a979d · 2002-03-18T13:06:00.000Z
[ 516299 ] urlparse can get fragments wrong
diff --git a/Lib/test/output/test_urlparse b/Lib/test/output/test_urlparse
@@ -1,4 +1,9 @@
 test_urlparse
+http://www.python.org = ('http', 'www.python.org', '', '', '', '')
+http://www.python.org#abc = ('http', 'www.python.org', '', '', '', 'abc')
+http://www.python.org/#abc = ('http', 'www.python.org', '/', '', '', 'abc')
+http://a/b/c/d;p?q#f = ('http', 'a', '/b/c/d', 'p', 'q', 'f')
+
 urlparse.urljoin() tests
 
 g:h           = 'g:h'
diff --git a/Lib/test/test_urlparse.py b/Lib/test/test_urlparse.py
@@ -4,6 +4,24 @@
 
 RFC1808_BASE = "http://a/b/c/d;p?q#f"
 
+for url, expected in [('http://www.python.org',
+                       ('http', 'www.python.org', '', '', '', '')),
+                      ('http://www.python.org#abc',
+                       ('http', 'www.python.org', '', '', '', 'abc')),
+                      ('http://www.python.org/#abc',
+                       ('http', 'www.python.org', '/', '', '', 'abc')),
+                      (RFC1808_BASE,
+                       ('http', 'a', '/b/c/d', 'p', 'q', 'f')),
+                      ]:
+    result = urlparse.urlparse(url)
+    print "%-13s = %r" % (url, result)
+    if result != expected:
+        errors += 1
+        print "urlparse(%r)" % url
+        print ("expected %r,\n"
+               "     got %r") % (expected, result)
+print
+
 def checkJoin(relurl, expected):
     global errors
     result = urlparse.urljoin(RFC1808_BASE, relurl)
diff --git a/Lib/urlparse.py b/Lib/urlparse.py
@@ -87,7 +87,9 @@ def urlsplit(url, scheme='', allow_fragments=1):
             if url[:2] == '//':
                 i = url.find('/', 2)
                 if i < 0:
-                    i = len(url)
+                    i = url.find('#')
+                    if i < 0:
+                        i = len(url)
                 netloc = url[2:i]
                 url = url[i:]
             if allow_fragments and '#' in url:
