diff options
| -rw-r--r-- | paste/packaging/__init__.py | 1 | ||||
| -rw-r--r-- | paste/packaging/install.py | 63 | ||||
| -rw-r--r-- | setup.py | 1 |
3 files changed, 65 insertions, 0 deletions
diff --git a/paste/packaging/__init__.py b/paste/packaging/__init__.py new file mode 100644 index 0000000..792d600 --- /dev/null +++ b/paste/packaging/__init__.py @@ -0,0 +1 @@ +# diff --git a/paste/packaging/install.py b/paste/packaging/install.py new file mode 100644 index 0000000..bff8d1b --- /dev/null +++ b/paste/packaging/install.py @@ -0,0 +1,63 @@ +import zipfile +import os +import urllib2 +import shutil + +def install_egg_file(filename, dest_dir, fileobj=None): + """ + Install the egg file at `filename` into `dest_dir`. + + Can also be installed from a `fileobj`; if given then `filename` + is just for error reporting. + """ + # Since the filename may be a URL, we can't trust os.path: + canonical_filename = filename.replace(os.sep, '/') + # This is the convention for egg file naming: + base_name = canonical_filename.split('/')[-1] + package_name = base_name.split('-')[0] + dest_dir = os.path.join(dest_dir, base_name) + zf = zipfile.ZipFile(fileobj or filename) + badfile = zf.testzip() + if badfile: + raise ValueError( + "Zip file %s is corrupted; first bad file: %r" + % (filename, badfile)) + for file_info in zf.infolist(): + fn = file_info.filename.lstrip('/') # make sure it's relative + if fn.startswith('EGG-INFO/'): + fn = package_name + '.egg-info' + fn[len('EGG-INFO'):] + dest_fn = os.path.join(dest_dir, fn) + if not os.path.exists(os.path.dirname(dest_fn)): + os.makedirs(os.path.dirname(dest_fn)) + if fn.endswith('/') and not os.path.exists(dest_fn): + os.makedir(dest_fn) + continue + f = open(dest_fn, 'wb') + f.write(zf.read(file_info.filename)) + f.close() + +def install_egg_from_url(url, dest_dir, package_name=None, + unpack=True): + """ + Install the egg file located at `url` into `dest_dir`. If + `package_name` is not given, then the package name is determined + from the url. + """ + f = urllib2.urlopen(url) + try: + if unpack: + install_egg_file(url, dest_dir, package_name=package_name, + fileobj=f) + else: + fn_out = os.path.join(dest_dir, url.split('/')[-1]) + if not os.path.exists(os.path.dirname(fn_out)): + os.makedirs(os.path.dirname(fn_out)) + fout = open(fn_out, 'wb') + shutil.copyfileobj(f, fout) + fout.close() + finally: + f.close() + +if __name__ == '__main__': + import sys + install_egg_file(*sys.argv[1:]) @@ -90,6 +90,7 @@ functionality. "paste.servers.scgi_server", "paste.wareweb", "paste.docsupport", + "paste.packaging", ], scripts=['scripts/paste-server', 'scripts/paste-setup'], download_url="", |