1 #!python 2 3 # This file was obtained from: 4 # http://peak.telecommunity.com/dist/ez_setup.py 5 # on 2011/1/21. 6 7 """Bootstrap setuptools installation 8 9 If you want to use setuptools in your package's setup.py, just include this 10 file in the same directory with it, and add this to the top of your setup.py:: 11 12 from ez_setup import use_setuptools 13 use_setuptools() 14 15 If you want to require a specific version of setuptools, set a download 16 mirror, or use an alternate download directory, you can do so by supplying 17 the appropriate options to ``use_setuptools()``. 18 19 This file can also be run as a script to install or upgrade setuptools. 20 """ 21 import sys 22 DEFAULT_VERSION = "0.6c11" 23 DEFAULT_URL = "http://pypi.python.org/packages/%s/s/setuptools/" % sys.version[:3] 24 25 md5_data = { 26 'setuptools-0.6b1-py2.3.egg': '8822caf901250d848b996b7f25c6e6ca', 27 'setuptools-0.6b1-py2.4.egg': 'b79a8a403e4502fbb85ee3f1941735cb', 28 'setuptools-0.6b2-py2.3.egg': '5657759d8a6d8fc44070a9d07272d99b', 29 'setuptools-0.6b2-py2.4.egg': '4996a8d169d2be661fa32a6e52e4f82a', 30 'setuptools-0.6b3-py2.3.egg': 'bb31c0fc7399a63579975cad9f5a0618', 31 'setuptools-0.6b3-py2.4.egg': '38a8c6b3d6ecd22247f179f7da669fac', 32 'setuptools-0.6b4-py2.3.egg': '62045a24ed4e1ebc77fe039aa4e6f7e5', 33 'setuptools-0.6b4-py2.4.egg': '4cb2a185d228dacffb2d17f103b3b1c4', 34 'setuptools-0.6c1-py2.3.egg': 'b3f2b5539d65cb7f74ad79127f1a908c', 35 'setuptools-0.6c1-py2.4.egg': 'b45adeda0667d2d2ffe14009364f2a4b', 36 'setuptools-0.6c10-py2.3.egg': 'ce1e2ab5d3a0256456d9fc13800a7090', 37 'setuptools-0.6c10-py2.4.egg': '57d6d9d6e9b80772c59a53a8433a5dd4', 38 'setuptools-0.6c10-py2.5.egg': 'de46ac8b1c97c895572e5e8596aeb8c7', 39 'setuptools-0.6c10-py2.6.egg': '58ea40aef06da02ce641495523a0b7f5', 40 'setuptools-0.6c11-py2.3.egg': '2baeac6e13d414a9d28e7ba5b5a596de', 41 'setuptools-0.6c11-py2.4.egg': 'bd639f9b0eac4c42497034dec2ec0c2b', 42 'setuptools-0.6c11-py2.5.egg': '64c94f3bf7a72a13ec83e0b24f2749b2', 43 'setuptools-0.6c11-py2.6.egg': 'bfa92100bd772d5a213eedd356d64086', 44 'setuptools-0.6c2-py2.3.egg': 'f0064bf6aa2b7d0f3ba0b43f20817c27', 45 'setuptools-0.6c2-py2.4.egg': '616192eec35f47e8ea16cd6a122b7277', 46 'setuptools-0.6c3-py2.3.egg': 'f181fa125dfe85a259c9cd6f1d7b78fa', 47 'setuptools-0.6c3-py2.4.egg': 'e0ed74682c998bfb73bf803a50e7b71e', 48 'setuptools-0.6c3-py2.5.egg': 'abef16fdd61955514841c7c6bd98965e', 49 'setuptools-0.6c4-py2.3.egg': 'b0b9131acab32022bfac7f44c5d7971f', 50 'setuptools-0.6c4-py2.4.egg': '2a1f9656d4fbf3c97bf946c0a124e6e2', 51 'setuptools-0.6c4-py2.5.egg': '8f5a052e32cdb9c72bcf4b5526f28afc', 52 'setuptools-0.6c5-py2.3.egg': 'ee9fd80965da04f2f3e6b3576e9d8167', 53 'setuptools-0.6c5-py2.4.egg': 'afe2adf1c01701ee841761f5bcd8aa64', 54 'setuptools-0.6c5-py2.5.egg': 'a8d3f61494ccaa8714dfed37bccd3d5d', 55 'setuptools-0.6c6-py2.3.egg': '35686b78116a668847237b69d549ec20', 56 'setuptools-0.6c6-py2.4.egg': '3c56af57be3225019260a644430065ab', 57 'setuptools-0.6c6-py2.5.egg': 'b2f8a7520709a5b34f80946de5f02f53', 58 'setuptools-0.6c7-py2.3.egg': '209fdf9adc3a615e5115b725658e13e2', 59 'setuptools-0.6c7-py2.4.egg': '5a8f954807d46a0fb67cf1f26c55a82e', 60 'setuptools-0.6c7-py2.5.egg': '45d2ad28f9750e7434111fde831e8372', 61 'setuptools-0.6c8-py2.3.egg': '50759d29b349db8cfd807ba8303f1902', 62 'setuptools-0.6c8-py2.4.egg': 'cba38d74f7d483c06e9daa6070cce6de', 63 'setuptools-0.6c8-py2.5.egg': '1721747ee329dc150590a58b3e1ac95b', 64 'setuptools-0.6c9-py2.3.egg': 'a83c4020414807b496e4cfbe08507c03', 65 'setuptools-0.6c9-py2.4.egg': '260a2be2e5388d66bdaee06abec6342a', 66 'setuptools-0.6c9-py2.5.egg': 'fe67c3e5a17b12c0e7c541b7ea43a8e6', 67 'setuptools-0.6c9-py2.6.egg': 'ca37b1ff16fa2ede6e19383e7b59245a', 68 } 69 70 import sys, os 71 try: from hashlib import md5 72 except ImportError: from md5 import md5 73 74 def _validate_md5(egg_name, data): 75 if egg_name in md5_data: 76 digest = md5(data).hexdigest() 77 if digest != md5_data[egg_name]: 78 print >>sys.stderr, ( 79 "md5 validation of %s failed! (Possible download problem?)" 80 % egg_name 81 ) 82 sys.exit(2) 83 return data 84 85 def use_setuptools( 86 version=DEFAULT_VERSION, download_base=DEFAULT_URL, to_dir=os.curdir, 87 download_delay=15 88 ): 89 """Automatically find/download setuptools and make it available on sys.path 90 91 `version` should be a valid setuptools version number that is available 92 as an egg for download under the `download_base` URL (which should end with 93 a '/'). `to_dir` is the directory where setuptools will be downloaded, if 94 it is not already available. If `download_delay` is specified, it should 95 be the number of seconds that will be paused before initiating a download, 96 should one be required. If an older version of setuptools is installed, 97 this routine will print a message to ``sys.stderr`` and raise SystemExit in 98 an attempt to abort the calling script. 99 """ 100 was_imported = 'pkg_resources' in sys.modules or 'setuptools' in sys.modules 101 def do_download(): 102 egg = download_setuptools(version, download_base, to_dir, download_delay) 103 sys.path.insert(0, egg) 104 import setuptools; setuptools.bootstrap_install_from = egg 105 try: 106 import pkg_resources 107 except ImportError: 108 return do_download() 109 try: 110 pkg_resources.require("setuptools>="+version); return 111 except pkg_resources.VersionConflict, e: 112 if was_imported: 113 print >>sys.stderr, ( 114 "The required version of setuptools (>=%s) is not available, and\n" 115 "can't be installed while this script is running. Please install\n" 116 " a more recent version first, using 'easy_install -U setuptools'." 117 "\n\n(Currently using %r)" 118 ) % (version, e.args[0]) 119 sys.exit(2) 120 except pkg_resources.DistributionNotFound: 121 pass 122 123 del pkg_resources, sys.modules['pkg_resources'] # reload ok 124 return do_download() 125 126 def download_setuptools( 127 version=DEFAULT_VERSION, download_base=DEFAULT_URL, to_dir=os.curdir, 128 delay = 15 129 ): 130 """Download setuptools from a specified location and return its filename 131 132 `version` should be a valid setuptools version number that is available 133 as an egg for download under the `download_base` URL (which should end 134 with a '/'). `to_dir` is the directory where the egg will be downloaded. 135 `delay` is the number of seconds to pause before an actual download attempt. 136 """ 137 import urllib2, shutil 138 egg_name = "setuptools-%s-py%s.egg" % (version,sys.version[:3]) 139 url = download_base + egg_name 140 saveto = os.path.join(to_dir, egg_name) 141 src = dst = None 142 if not os.path.exists(saveto): # Avoid repeated downloads 143 try: 144 from distutils import log 145 if delay: 146 log.warn(""" 147 --------------------------------------------------------------------------- 148 This script requires setuptools version %s to run (even to display 149 help). I will attempt to download it for you (from 150 %s), but 151 you may need to enable firewall access for this script first. 152 I will start the download in %d seconds. 153 154 (Note: if this machine does not have network access, please obtain the file 155 156 %s 157 158 and place it in this directory before rerunning this script.) 159 ---------------------------------------------------------------------------""", 160 version, download_base, delay, url 161 ); from time import sleep; sleep(delay) 162 log.warn("Downloading %s", url) 163 src = urllib2.urlopen(url) 164 # Read/write all in one block, so we don't create a corrupt file 165 # if the download is interrupted. 166 data = _validate_md5(egg_name, src.read()) 167 dst = open(saveto,"wb"); dst.write(data) 168 finally: 169 if src: src.close() 170 if dst: dst.close() 171 return os.path.realpath(saveto) 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 def main(argv, version=DEFAULT_VERSION): 209 """Install or upgrade setuptools and EasyInstall""" 210 try: 211 import setuptools 212 except ImportError: 213 egg = None 214 try: 215 egg = download_setuptools(version, delay=0) 216 sys.path.insert(0,egg) 217 from setuptools.command.easy_install import main 218 return main(list(argv)+[egg]) # we're done here 219 finally: 220 if egg and os.path.exists(egg): 221 os.unlink(egg) 222 else: 223 if setuptools.__version__ == '0.0.1': 224 print >>sys.stderr, ( 225 "You have an obsolete version of setuptools installed. Please\n" 226 "remove it from your system entirely before rerunning this script." 227 ) 228 sys.exit(2) 229 230 req = "setuptools>="+version 231 import pkg_resources 232 try: 233 pkg_resources.require(req) 234 except pkg_resources.VersionConflict: 235 try: 236 from setuptools.command.easy_install import main 237 except ImportError: 238 from easy_install import main 239 main(list(argv)+[download_setuptools(delay=0)]) 240 sys.exit(0) # try to force an exit 241 else: 242 if argv: 243 from setuptools.command.easy_install import main 244 main(argv) 245 else: 246 print "Setuptools version",version,"or greater has been installed." 247 print '(Run "ez_setup.py -U setuptools" to reinstall or upgrade.)' 248 249 def update_md5(filenames): 250 """Update our built-in md5 registry""" 251 252 import re 253 254 for name in filenames: 255 base = os.path.basename(name) 256 f = open(name,'rb') 257 md5_data[base] = md5(f.read()).hexdigest() 258 f.close() 259 260 data = [" %r: %r,\n" % it for it in md5_data.items()] 261 data.sort() 262 repl = "".join(data) 263 264 import inspect 265 srcfile = inspect.getsourcefile(sys.modules[__name__]) 266 f = open(srcfile, 'rb'); src = f.read(); f.close() 267 268 match = re.search("\nmd5_data = {\n([^}]+)}", src) 269 if not match: 270 print >>sys.stderr, "Internal error!" 271 sys.exit(2) 272 273 src = src[:match.start(1)] + repl + src[match.end(1):] 274 f = open(srcfile,'w') 275 f.write(src) 276 f.close() 277 278 279 if __name__=='__main__': 280 if len(sys.argv)>2 and sys.argv[1]=='--md5update': 281 update_md5(sys.argv[2:]) 282 else: 283 main(sys.argv[1:]) 284