|
|
This file can also be run as a script to install or upgrade setuptools. |
This file can also be run as a script to install or upgrade setuptools. |
""" |
""" |
|
import sys |
DEFAULT_VERSION = "0.5a13" |
DEFAULT_VERSION = "0.6a7" |
DEFAULT_URL = "http://www.python.org/packages/source/s/setuptools/" |
DEFAULT_URL = "http://cheeseshop.python.org/packages/%s/s/setuptools/" % sys.version[:3] |
|
|
|
md5_data = { |
|
'setuptools-0.5a13-py2.3.egg': '85edcf0ef39bab66e130d3f38f578c86', |
|
'setuptools-0.5a13-py2.4.egg': 'ede4be600e3890e06d4ee5e0148e092a', |
|
'setuptools-0.6a1-py2.3.egg': 'ee819a13b924d9696b0d6ca6d1c5833d', |
|
'setuptools-0.6a1-py2.4.egg': '8256b5f1cd9e348ea6877b5ddd56257d', |
|
'setuptools-0.6a2-py2.3.egg': 'b98da449da411267c37a738f0ab625ba', |
|
'setuptools-0.6a2-py2.4.egg': 'be5b88bc30aed63fdefd2683be135c3b', |
|
'setuptools-0.6a3-py2.3.egg': 'ee0e325de78f23aab79d33106dc2a8c8', |
|
'setuptools-0.6a3-py2.4.egg': 'd95453d525a456d6c23e7a5eea89a063', |
|
'setuptools-0.6a4-py2.3.egg': 'e958cbed4623bbf47dd1f268b99d7784', |
|
'setuptools-0.6a4-py2.4.egg': '7f33c3ac2ef1296f0ab4fac1de4767d8', |
|
'setuptools-0.6a5-py2.3.egg': '748408389c49bcd2d84f6ae0b01695b1', |
|
'setuptools-0.6a5-py2.4.egg': '999bacde623f4284bfb3ea77941d2627', |
|
'setuptools-0.6a6-py2.3.egg': '7858139f06ed0600b0d9383f36aca24c', |
|
'setuptools-0.6a6-py2.4.egg': 'c10d20d29acebce0dc76219dc578d058', |
|
'setuptools-0.6a7-py2.3.egg': 'cfc4125ddb95c07f9500adc5d6abef6f', |
|
'setuptools-0.6a7-py2.4.egg': 'c6d62dab4461f71aed943caea89e6f20', |
|
} |
|
|
import sys, os |
import sys, os |
|
|
|
def _validate_md5(egg_name, data): |
|
if egg_name in md5_data: |
|
from md5 import md5 |
|
digest = md5(data).hexdigest() |
|
if digest != md5_data[egg_name]: |
|
print >>sys.stderr, ( |
|
"md5 validation of %s failed! (Possible download problem?)" |
|
% egg_name |
|
) |
|
sys.exit(2) |
|
return data |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def use_setuptools( |
def use_setuptools( |
version=DEFAULT_VERSION, download_base=DEFAULT_URL, to_dir=os.curdir |
version=DEFAULT_VERSION, download_base=DEFAULT_URL, to_dir=os.curdir, |
|
download_delay=15 |
): |
): |
"""Automatically find/download setuptools and make it available on sys.path |
"""Automatically find/download setuptools and make it available on sys.path |
|
|
`version` should be a valid setuptools version number that is available |
`version` should be a valid setuptools version number that is available |
as an egg for download under the `download_base` URL (which should end with |
as an egg for download under the `download_base` URL (which should end with |
a '/'). `to_dir` is the directory where setuptools will be downloaded, if |
a '/'). `to_dir` is the directory where setuptools will be downloaded, if |
it is not already available. |
it is not already available. If `download_delay` is specified, it should |
|
be the number of seconds that will be paused before initiating a download, |
If an older version of setuptools is installed, this will print a message |
should one be required. If an older version of setuptools is installed, |
to ``sys.stderr`` and raise SystemExit in an attempt to abort the calling |
this routine will print a message to ``sys.stderr`` and raise SystemExit in |
script. |
an attempt to abort the calling script. |
""" |
""" |
try: |
try: |
import setuptools |
import setuptools |
"remove it from your system entirely before rerunning this script." |
"remove it from your system entirely before rerunning this script." |
) |
) |
sys.exit(2) |
sys.exit(2) |
|
|
except ImportError: |
except ImportError: |
egg = download_setuptools(version, download_base, to_dir) |
egg = download_setuptools(version, download_base, to_dir, download_delay) |
sys.path.insert(0, egg) |
sys.path.insert(0, egg) |
import setuptools; setuptools.bootstrap_install_from = egg |
import setuptools; setuptools.bootstrap_install_from = egg |
|
|
sys.exit(2) |
sys.exit(2) |
|
|
def download_setuptools( |
def download_setuptools( |
version=DEFAULT_VERSION, download_base=DEFAULT_URL, to_dir=os.curdir |
version=DEFAULT_VERSION, download_base=DEFAULT_URL, to_dir=os.curdir, |
|
delay = 15 |
): |
): |
"""Download setuptools from a specified location and return its filename |
"""Download setuptools from a specified location and return its filename |
|
|
`version` should be a valid setuptools version number that is available |
`version` should be a valid setuptools version number that is available |
as an egg for download under the `download_base` URL (which should end |
as an egg for download under the `download_base` URL (which should end |
with a '/'). `to_dir` is the directory where the egg will be downloaded. |
with a '/'). `to_dir` is the directory where the egg will be downloaded. |
|
`delay` is the number of seconds to pause before an actual download attempt. |
""" |
""" |
import urllib2, shutil |
import urllib2, shutil |
egg_name = "setuptools-%s-py%s.egg" % (version,sys.version[:3]) |
egg_name = "setuptools-%s-py%s.egg" % (version,sys.version[:3]) |
url = download_base + egg_name + '.zip' # XXX |
url = download_base + egg_name |
saveto = os.path.join(to_dir, egg_name) |
saveto = os.path.join(to_dir, egg_name) |
src = dst = None |
src = dst = None |
|
|
if not os.path.exists(saveto): # Avoid repeated downloads |
if not os.path.exists(saveto): # Avoid repeated downloads |
try: |
try: |
from distutils import log |
from distutils import log |
|
if delay: |
|
log.warn(""" |
|
--------------------------------------------------------------------------- |
|
This script requires setuptools version %s to run (even to display |
|
help). I will attempt to download it for you (from |
|
%s), but |
|
you may need to enable firewall access for this script first. |
|
I will start the download in %d seconds. |
|
---------------------------------------------------------------------------""", |
|
version, download_base, delay |
|
); from time import sleep; sleep(delay) |
log.warn("Downloading %s", url) |
log.warn("Downloading %s", url) |
src = urllib2.urlopen(url) |
src = urllib2.urlopen(url) |
# Read/write all in one block, so we don't create a corrupt file |
# Read/write all in one block, so we don't create a corrupt file |
# if the download is interrupted. |
# if the download is interrupted. |
data = src.read() |
data = _validate_md5(egg_name, src.read()) |
dst = open(saveto,"wb") |
dst = open(saveto,"wb"); dst.write(data) |
dst.write(data) |
|
finally: |
finally: |
if src: src.close() |
if src: src.close() |
if dst: dst.close() |
if dst: dst.close() |
|
|
return os.path.realpath(saveto) |
return os.path.realpath(saveto) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def main(argv, version=DEFAULT_VERSION): |
def main(argv, version=DEFAULT_VERSION): |
"""Install or upgrade setuptools and EasyInstall""" |
"""Install or upgrade setuptools and EasyInstall""" |
|
|
import tempfile, shutil |
import tempfile, shutil |
tmpdir = tempfile.mkdtemp(prefix="easy_install-") |
tmpdir = tempfile.mkdtemp(prefix="easy_install-") |
try: |
try: |
egg = download_setuptools(version, to_dir=tmpdir) |
egg = download_setuptools(version, to_dir=tmpdir, delay=0) |
sys.path.insert(0,egg) |
sys.path.insert(0,egg) |
from setuptools.command.easy_install import main |
from setuptools.command.easy_install import main |
main(list(argv)+[egg]) |
main(list(argv)+[egg]) |
from setuptools.command.easy_install import main |
from setuptools.command.easy_install import main |
except ImportError: |
except ImportError: |
from easy_install import main |
from easy_install import main |
main(list(argv)+[download_setuptools()]) |
main(list(argv)+[download_setuptools(delay=0)]) |
sys.exit(0) # try to force an exit |
sys.exit(0) # try to force an exit |
else: |
else: |
if argv: |
if argv: |
else: |
else: |
print "Setuptools version",version,"or greater has been installed." |
print "Setuptools version",version,"or greater has been installed." |
print '(Run "ez_setup.py -U setuptools" to reinstall or upgrade.)' |
print '(Run "ez_setup.py -U setuptools" to reinstall or upgrade.)' |
|
|
|
|
|
|
|
def update_md5(filenames): |
|
"""Update our built-in md5 registry""" |
|
|
|
import re |
|
from md5 import md5 |
|
|
|
for name in filenames: |
|
base = os.path.basename(name) |
|
f = open(name,'rb') |
|
md5_data[base] = md5(f.read()).hexdigest() |
|
f.close() |
|
|
|
data = [" %r: %r,\n" % it for it in md5_data.items()] |
|
data.sort() |
|
repl = "".join(data) |
|
|
|
import inspect |
|
srcfile = inspect.getsourcefile(sys.modules[__name__]) |
|
f = open(srcfile, 'rb'); src = f.read(); f.close() |
|
|
|
match = re.search("\nmd5_data = {\n([^}]+)}", src) |
|
if not match: |
|
print >>sys.stderr, "Internal error!" |
|
sys.exit(2) |
|
|
|
src = src[:match.start(1)] + repl + src[match.end(1):] |
|
f = open(srcfile,'w') |
|
f.write(src) |
|
f.close() |
|
|
|
|
if __name__=='__main__': |
if __name__=='__main__': |
|
if len(sys.argv)>2 and sys.argv[1]=='--md5update': |
|
update_md5(sys.argv[2:]) |
|
else: |
main(sys.argv[1:]) |
main(sys.argv[1:]) |
|
|
|
|
|
|
|
|
|
|