From 2c5978d3bb4f91f7f837e974c4abdc8248ed8d3f Mon Sep 17 00:00:00 2001
From: PJ Eby <distutils-sig@python.org>
Date: Mon, 25 Sep 2006 21:31:59 +0000
Subject: Fixed SF downloads aborting when a SF mirror returns an HTML page
 when it should've returned a 404.  Fall back to
 ``sf-mirrors.telecommunity.com`` round-robin address for SF mirrors if
 ``dl.sourceforge.net`` doesn't work. (backport from trunk)

--HG--
branch : setuptools-0.6
extra : convert_revision : svn%3A6015fed2-1504-0410-9fe1-9d1591cc4771/sandbox/branches/setuptools-0.6%4052004
---
 setuptools/package_index.py | 44 ++++++++++++++++++++++----------------------
 1 file changed, 22 insertions(+), 22 deletions(-)

(limited to 'setuptools/package_index.py')

diff --git a/setuptools/package_index.py b/setuptools/package_index.py
index 837eff19..78e89daf 100755
--- a/setuptools/package_index.py
+++ b/setuptools/package_index.py
@@ -533,7 +533,6 @@ class PackageIndex(Environment):
 
     dl_blocksize = 8192
     def _download_to(self, url, filename):
-        self.url_ok(url,True)   # raises error if not allowed
         self.info("Downloading %s", url)
         # Download the file
         fp, tfp, info = None, None, None
@@ -572,9 +571,18 @@ class PackageIndex(Environment):
     def reporthook(self, url, filename, blocknum, blksize, size):
         pass    # no-op
 
-    def retry_sf_download(self, url, filename):
+
+    def _attempt_download(self, url, filename):
+        headers = self._download_to(url, filename)
+        if 'html' in headers['content-type'].lower():
+            return self._download_html(url, headers, filename)
+        else:
+            return filename
+
+    def _retry_sf_download(self, url, filename):
+        self.url_ok(url, True)   # raises error if not allowed
         try:
-            return self._download_to(url, filename)
+            return self._attempt_download(url, filename)
         except (KeyboardInterrupt,SystemExit):
             raise
         except:
@@ -588,7 +596,9 @@ class PackageIndex(Environment):
             self.warn("Download failed: %s", sys.exc_info()[1])
             url = urlparse.urlunparse((scheme, mirror, path, param, '', frag))
             try:
-                return self._download_to(url, filename)
+                return self._attempt_download(url, filename)
+            except (KeyboardInterrupt,SystemExit):
+                raise
             except:
                 _sf_mirrors.remove(mirror)  # don't retry the same mirror
                 mirror = get_sf_ip()
@@ -603,16 +613,6 @@ class PackageIndex(Environment):
 
 
 
-
-
-
-
-
-
-
-
-
-
     def open_url(self, url):
         if url.startswith('file:'):
             return local_open(url)
@@ -648,16 +648,16 @@ class PackageIndex(Environment):
         elif scheme=='file':
             return urllib2.url2pathname(urlparse.urlparse(url)[2])
         else:
-            headers = self.retry_sf_download(url, filename)
-            if 'html' in headers['content-type'].lower():
-                return self._download_html(url, headers, filename, tmpdir)
-            else:
-                return filename
+            return self._retry_sf_download(url, filename)
+
+
+
+
 
     def scan_url(self, url):
         self.process_url(url, True)
 
-    def _download_html(self, url, headers, filename, tmpdir):
+    def _download_html(self, url, headers, filename):
         file = open(filename)
         for line in file:
             if line.strip():
@@ -700,7 +700,8 @@ _sf_mirrors = []
 def get_sf_ip():
     if not _sf_mirrors:
         try:
-            _sf_mirrors[:] = socket.gethostbyname_ex('dl.sourceforge.net')[-1]
+            _sf_mirrors[:] = socket.gethostbyname_ex(
+                'sf-mirrors.telecommunity.com')[-1]
         except socket.error:
             # DNS-bl0ck1n9 f1r3w4llz sUx0rs!
             _sf_mirrors[:] = ['dl.sourceforge.net']
@@ -734,5 +735,4 @@ def local_open(url):
 
 
 
-
 # this line is a kludge to keep the trailing blank lines for pje's editor
-- 
cgit v1.2.3