Location: /

History

                  
14.3 KiB
text/x-python
gravatar
self._data_walker -> self.data_walker
 1
 2
 3
 4
 5
 6
 7
 8
 9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
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
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458

                        
#!python

                        
"""Bootstrap distribute installation

                        

                        
If you want to use setuptools in your package's setup.py, just include this

                        
file in the same directory with it, and add this to the top of your setup.py::

                        

                        
from distribute_setup import use_setuptools

                        
use_setuptools()

                        

                        
If you want to require a specific version of setuptools, set a download

                        
mirror, or use an alternate download directory, you can do so by supplying

                        
the appropriate options to ``use_setuptools()``.

                        

                        
This file can also be run as a script to install or upgrade setuptools.

                        
"""

                        
import os

                        
import sys

                        
import time

                        
import fnmatch

                        
import tempfile

                        
import tarfile

                        
from distutils import log

                        

                        
try:

                        
from site import USER_SITE

                        
except ImportError:

                        
USER_SITE = None

                        

                        
try:

                        
import subprocess

                        

                        
def _python_cmd(*args):

                        
args = (sys.executable,) + args

                        
return subprocess.call(args) == 0

                        

                        
except ImportError:

                        
# will be used for python 2.3

                        
def _python_cmd(*args):

                        
args = (sys.executable,) + args

                        
# quoting arguments if windows

                        
if sys.platform == 'win32':

                        
def quote(arg):

                        
if ' ' in arg:

                        
return '"%s"' % arg

                        
return arg

                        
args = [quote(arg) for arg in args]

                        
return os.spawnl(os.P_WAIT, sys.executable, *args) == 0

                        

                        
DEFAULT_VERSION = "0.6.8"

                        
DEFAULT_URL = "http://pypi.python.org/packages/source/d/distribute/"

                        
SETUPTOOLS_PKG_INFO = """\

                        
Metadata-Version: 1.0

                        
Name: setuptools

                        
Version: 0.6c9

                        
Summary: xxxx

                        
Home-page: xxx

                        
Author: xxx

                        
Author-email: xxx

                        
License: xxx

                        
Description: xxx

                        
"""

                        

                        

                        
def _install(tarball):

                        
# extracting the tarball

                        
tmpdir = tempfile.mkdtemp()

                        
log.warn('Extracting in %s', tmpdir)

                        
old_wd = os.getcwd()

                        
try:

                        
os.chdir(tmpdir)

                        
tar = tarfile.open(tarball)

                        
_extractall(tar)

                        
tar.close()

                        

                        
# going in the directory

                        
subdir = os.path.join(tmpdir, os.listdir(tmpdir)[0])

                        
os.chdir(subdir)

                        
log.warn('Now working in %s', subdir)

                        

                        
# installing

                        
log.warn('Installing Distribute')

                        
assert _python_cmd('setup.py', 'install')

                        
finally:

                        
os.chdir(old_wd)

                        

                        

                        
def _build_egg(egg, tarball, to_dir):

                        
# extracting the tarball

                        
tmpdir = tempfile.mkdtemp()

                        
log.warn('Extracting in %s', tmpdir)

                        
old_wd = os.getcwd()

                        
try:

                        
os.chdir(tmpdir)

                        
tar = tarfile.open(tarball)

                        
_extractall(tar)

                        
tar.close()

                        

                        
# going in the directory

                        
subdir = os.path.join(tmpdir, os.listdir(tmpdir)[0])

                        
os.chdir(subdir)

                        
log.warn('Now working in %s', subdir)

                        

                        
# building an egg

                        
log.warn('Building a Distribute egg in %s', to_dir)

                        
_python_cmd('setup.py', '-q', 'bdist_egg', '--dist-dir', to_dir)

                        

                        
finally:

                        
os.chdir(old_wd)

                        
# returning the result

                        
log.warn(egg)

                        
if not os.path.exists(egg):

                        
raise IOError('Could not build the egg.')

                        

                        

                        
def _do_download(version, download_base, to_dir, download_delay):

                        
egg = os.path.join(to_dir, 'distribute-%s-py%d.%d.egg'

                        
% (version, sys.version_info[0], sys.version_info[1]))

                        
if not os.path.exists(egg):

                        
tarball = download_setuptools(version, download_base,

                        
to_dir, download_delay)

                        
_build_egg(egg, tarball, to_dir)

                        
sys.path.insert(0, egg)

                        
import setuptools

                        
setuptools.bootstrap_install_from = egg

                        

                        

                        
def use_setuptools(version=DEFAULT_VERSION, download_base=DEFAULT_URL,

                        
to_dir=os.curdir, download_delay=15, no_fake=True):

                        
# making sure we use the absolute path

                        
to_dir = os.path.abspath(to_dir)

                        
was_imported = 'pkg_resources' in sys.modules or \

                        
'setuptools' in sys.modules

                        
try:

                        
try:

                        
import pkg_resources

                        
if not hasattr(pkg_resources, '_distribute'):

                        
if not no_fake:

                        
_fake_setuptools()

                        
raise ImportError

                        
except ImportError:

                        
return _do_download(version, download_base, to_dir, download_delay)

                        
try:

                        
pkg_resources.require("distribute>="+version)

                        
return

                        
except pkg_resources.VersionConflict:

                        
e = sys.exc_info()[1]

                        
if was_imported:

                        
sys.stderr.write(

                        
"The required version of distribute (>=%s) is not available,\n"

                        
"and can't be installed while this script is running. Please\n"

                        
"install a more recent version first, using\n"

                        
"'easy_install -U distribute'."

                        
"\n\n(Currently using %r)\n" % (version, e.args[0]))

                        
sys.exit(2)

                        
else:

                        
del pkg_resources, sys.modules['pkg_resources'] # reload ok

                        
return _do_download(version, download_base, to_dir,

                        
download_delay)

                        
except pkg_resources.DistributionNotFound:

                        
return _do_download(version, download_base, to_dir,

                        
download_delay)

                        
finally:

                        
if not no_fake:

                        
_create_fake_setuptools_pkg_info(to_dir)

                        

                        
def download_setuptools(version=DEFAULT_VERSION, download_base=DEFAULT_URL,

                        
to_dir=os.curdir, delay=15):

                        
"""Download distribute from a specified location and return its filename

                        

                        
`version` should be a valid distribute version number that is available

                        
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.

                        
`delay` is the number of seconds to pause before an actual download

                        
attempt.

                        
"""

                        
# making sure we use the absolute path

                        
to_dir = os.path.abspath(to_dir)

                        
try:

                        
from urllib.request import urlopen

                        
except ImportError:

                        
from urllib2 import urlopen

                        
tgz_name = "distribute-%s.tar.gz" % version

                        
url = download_base + tgz_name

                        
saveto = os.path.join(to_dir, tgz_name)

                        
src = dst = None

                        
if not os.path.exists(saveto): # Avoid repeated downloads

                        
try:

                        
log.warn("Downloading %s", url)

                        
src = urlopen(url)

                        
# Read/write all in one block, so we don't create a corrupt file

                        
# if the download is interrupted.

                        
data = src.read()

                        
dst = open(saveto, "wb")

                        
dst.write(data)

                        
finally:

                        
if src:

                        
src.close()

                        
if dst:

                        
dst.close()

                        
return os.path.realpath(saveto)

                        

                        

                        
def _patch_file(path, content):

                        
"""Will backup the file then patch it"""

                        
existing_content = open(path).read()

                        
if existing_content == content:

                        
# already patched

                        
log.warn('Already patched.')

                        
return False

                        
log.warn('Patching...')

                        
_rename_path(path)

                        
f = open(path, 'w')

                        
try:

                        
f.write(content)

                        
finally:

                        
f.close()

                        
return True

                        

                        

                        
def _same_content(path, content):

                        
return open(path).read() == content

                        

                        

                        
def _rename_path(path):

                        
new_name = path + '.OLD.%s' % time.time()

                        
log.warn('Renaming %s into %s', path, new_name)

                        
try:

                        
from setuptools.sandbox import DirectorySandbox

                        
def _violation(*args):

                        
pass

                        
DirectorySandbox._violation = _violation

                        
except ImportError:

                        
pass

                        

                        
os.rename(path, new_name)

                        
return new_name

                        

                        

                        
def _remove_flat_installation(placeholder):

                        
if not os.path.isdir(placeholder):

                        
log.warn('Unkown installation at %s', placeholder)

                        
return False

                        
found = False

                        
for file in os.listdir(placeholder):

                        
if fnmatch.fnmatch(file, 'setuptools*.egg-info'):

                        
found = True

                        
break

                        
if not found:

                        
log.warn('Could not locate setuptools*.egg-info')

                        
return

                        

                        
log.warn('Removing elements out of the way...')

                        
pkg_info = os.path.join(placeholder, file)

                        
if os.path.isdir(pkg_info):

                        
patched = _patch_egg_dir(pkg_info)

                        
else:

                        
patched = _patch_file(pkg_info, SETUPTOOLS_PKG_INFO)

                        

                        
if not patched:

                        
log.warn('%s already patched.', pkg_info)

                        
return False

                        
# now let's move the files out of the way

                        
for element in ('setuptools', 'pkg_resources.py', 'site.py'):

                        
element = os.path.join(placeholder, element)

                        
if os.path.exists(element):

                        
_rename_path(element)

                        
else:

                        
log.warn('Could not find the %s element of the '

                        
'Setuptools distribution', element)

                        
return True

                        

                        

                        
def _after_install(dist):

                        
log.warn('After install bootstrap.')

                        
placeholder = dist.get_command_obj('install').install_purelib

                        
_create_fake_setuptools_pkg_info(placeholder)

                        

                        
def _create_fake_setuptools_pkg_info(placeholder):

                        
if not placeholder or not os.path.exists(placeholder):

                        
log.warn('Could not find the install location')

                        
return

                        
pyver = '%s.%s' % (sys.version_info[0], sys.version_info[1])

                        
setuptools_file = 'setuptools-0.6c9-py%s.egg-info' % pyver

                        
pkg_info = os.path.join(placeholder, setuptools_file)

                        
if os.path.exists(pkg_info):

                        
log.warn('%s already exists', pkg_info)

                        
return

                        
log.warn('Creating %s', pkg_info)

                        
f = open(pkg_info, 'w')

                        
try:

                        
f.write(SETUPTOOLS_PKG_INFO)

                        
finally:

                        
f.close()

                        
pth_file = os.path.join(placeholder, 'setuptools.pth')

                        
log.warn('Creating %s', pth_file)

                        
f = open(pth_file, 'w')

                        
try:

                        
f.write(os.path.join(os.curdir, setuptools_file))

                        
finally:

                        
f.close()

                        

                        

                        
def _patch_egg_dir(path):

                        
# let's check if it's already patched

                        
pkg_info = os.path.join(path, 'EGG-INFO', 'PKG-INFO')

                        
if os.path.exists(pkg_info):

                        
if _same_content(pkg_info, SETUPTOOLS_PKG_INFO):

                        
log.warn('%s already patched.', pkg_info)

                        
return False

                        
_rename_path(path)

                        
os.mkdir(path)

                        
os.mkdir(os.path.join(path, 'EGG-INFO'))

                        
pkg_info = os.path.join(path, 'EGG-INFO', 'PKG-INFO')

                        
f = open(pkg_info, 'w')

                        
try:

                        
f.write(SETUPTOOLS_PKG_INFO)

                        
finally:

                        
f.close()

                        
return True

                        

                        

                        
def _before_install():

                        
log.warn('Before install bootstrap.')

                        
_fake_setuptools()

                        

                        

                        
def _under_prefix(location):

                        
if 'install' not in sys.argv:

                        
return True

                        
args = sys.argv[sys.argv.index('install')+1:]

                        
for index, arg in enumerate(args):

                        
for option in ('--root', '--prefix'):

                        
if arg.startswith('%s=' % option):

                        
top_dir = arg.split('root=')[-1]

                        
return location.startswith(top_dir)

                        
elif arg == option:

                        
if len(args) > index:

                        
top_dir = args[index+1]

                        
return location.startswith(top_dir)

                        
elif option == '--user' and USER_SITE is not None:

                        
return location.startswith(USER_SITE)

                        
return True

                        

                        

                        
def _fake_setuptools():

                        
log.warn('Scanning installed packages')

                        
try:

                        
import pkg_resources

                        
except ImportError:

                        
# we're cool

                        
log.warn('Setuptools or Distribute does not seem to be installed.')

                        
return

                        
ws = pkg_resources.working_set

                        
try:

                        
setuptools_dist = ws.find(pkg_resources.Requirement.parse('setuptools',

                        
replacement=False))

                        
except TypeError:

                        
# old distribute API

                        
setuptools_dist = ws.find(pkg_resources.Requirement.parse('setuptools'))

                        

                        
if setuptools_dist is None:

                        
log.warn('No setuptools distribution found')

                        
return

                        
# detecting if it was already faked

                        
setuptools_location = setuptools_dist.location

                        
log.warn('Setuptools installation detected at %s', setuptools_location)

                        

                        
# if --root or --preix was provided, and if

                        
# setuptools is not located in them, we don't patch it

                        
if not _under_prefix(setuptools_location):

                        
log.warn('Not patching, --root or --prefix is installing Distribute'

                        
' in another location')

                        
return

                        

                        
# let's see if its an egg

                        
if not setuptools_location.endswith('.egg'):

                        
log.warn('Non-egg installation')

                        
res = _remove_flat_installation(setuptools_location)

                        
if not res:

                        
return

                        
else:

                        
log.warn('Egg installation')

                        
pkg_info = os.path.join(setuptools_location, 'EGG-INFO', 'PKG-INFO')

                        
if (os.path.exists(pkg_info) and

                        
_same_content(pkg_info, SETUPTOOLS_PKG_INFO)):

                        
log.warn('Already patched.')

                        
return

                        
log.warn('Patching...')

                        
# let's create a fake egg replacing setuptools one

                        
res = _patch_egg_dir(setuptools_location)

                        
if not res:

                        
return

                        
log.warn('Patched done.')

                        
_relaunch()

                        

                        

                        
def _relaunch():

                        
log.warn('Relaunching...')

                        
# we have to relaunch the process

                        
args = [sys.executable] + sys.argv

                        
sys.exit(subprocess.call(args))

                        

                        

                        
def _extractall(self, path=".", members=None):

                        
"""Extract all members from the archive to the current working

                        
directory and set owner, modification time and permissions on

                        
directories afterwards. `path' specifies a different directory

                        
to extract to. `members' is optional and must be a subset of the

                        
list returned by getmembers().

                        
"""

                        
import copy

                        
import operator

                        
from tarfile import ExtractError

                        
directories = []

                        

                        
if members is None:

                        
members = self

                        

                        
for tarinfo in members:

                        
if tarinfo.isdir():

                        
# Extract directories with a safe mode.

                        
directories.append(tarinfo)

                        
tarinfo = copy.copy(tarinfo)

                        
tarinfo.mode = 448 # decimal for oct 0700

                        
self.extract(tarinfo, path)

                        

                        
# Reverse sort directories.

                        
if sys.version_info < (2, 4):

                        
def sorter(dir1, dir2):

                        
return cmp(dir1.name, dir2.name)

                        
directories.sort(sorter)

                        
directories.reverse()

                        
else:

                        
directories.sort(key=operator.attrgetter('name'), reverse=True)

                        

                        
# Set correct owner, mtime and filemode on directories.

                        
for tarinfo in directories:

                        
dirpath = os.path.join(path, tarinfo.name)

                        
try:

                        
self.chown(tarinfo, dirpath)

                        
self.utime(tarinfo, dirpath)

                        
self.chmod(tarinfo, dirpath)

                        
except ExtractError:

                        
e = sys.exc_info()[1]

                        
if self.errorlevel > 1:

                        
raise

                        
else:

                        
self._dbg(1, "tarfile: %s" % e)

                        

                        

                        
def main(argv, version=DEFAULT_VERSION):

                        
"""Install or upgrade setuptools and EasyInstall"""

                        
tarball = download_setuptools()

                        
_install(tarball)

                        

                        

                        
if __name__ == '__main__':

                        
main(sys.argv[1:])