PageRenderTime 49ms CodeModel.GetById 22ms RepoModel.GetById 0ms app.codeStats 0ms

/ez_setup.py

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