mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2024-12-29 01:08:54 +00:00
124 lines
3.6 KiB
Python
124 lines
3.6 KiB
Python
from setuptools import Command
|
|
from setuptools.archive_util import unpack_archive
|
|
from distutils import log, dir_util
|
|
import os, shutil, pkg_resources
|
|
|
|
class install_egg_info(Command):
|
|
"""Install an .egg-info directory for the package"""
|
|
|
|
description = "Install an .egg-info directory for the package"
|
|
|
|
user_options = [
|
|
('install-dir=', 'd', "directory to install to"),
|
|
]
|
|
|
|
def initialize_options(self):
|
|
self.install_dir = None
|
|
|
|
def finalize_options(self):
|
|
self.set_undefined_options('install_lib',('install_dir','install_dir'))
|
|
ei_cmd = self.get_finalized_command("egg_info")
|
|
basename = pkg_resources.Distribution(
|
|
None, None, ei_cmd.egg_name, ei_cmd.egg_version
|
|
).egg_name()+'.egg-info'
|
|
self.source = ei_cmd.egg_info
|
|
self.target = os.path.join(self.install_dir, basename)
|
|
self.outputs = [self.target]
|
|
|
|
def run(self):
|
|
self.run_command('egg_info')
|
|
target = self.target
|
|
if os.path.isdir(self.target) and not os.path.islink(self.target):
|
|
dir_util.remove_tree(self.target, dry_run=self.dry_run)
|
|
elif os.path.exists(self.target):
|
|
self.execute(os.unlink,(self.target,),"Removing "+self.target)
|
|
if not self.dry_run:
|
|
pkg_resources.ensure_directory(self.target)
|
|
self.execute(self.copytree, (),
|
|
"Copying %s to %s" % (self.source, self.target)
|
|
)
|
|
self.install_namespaces()
|
|
|
|
def get_outputs(self):
|
|
return self.outputs
|
|
|
|
def copytree(self):
|
|
# Copy the .egg-info tree to site-packages
|
|
def skimmer(src,dst):
|
|
# filter out source-control directories; note that 'src' is always
|
|
# a '/'-separated path, regardless of platform. 'dst' is a
|
|
# platform-specific path.
|
|
for skip in '.svn/','CVS/':
|
|
if src.startswith(skip) or '/'+skip in src:
|
|
return None
|
|
self.outputs.append(dst)
|
|
log.debug("Copying %s to %s", src, dst)
|
|
return dst
|
|
unpack_archive(self.source, self.target, skimmer)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def install_namespaces(self):
|
|
nsp = self._get_all_ns_packages()
|
|
if not nsp: return
|
|
filename,ext = os.path.splitext(self.target)
|
|
filename += '-nspkg.pth'; self.outputs.append(filename)
|
|
log.info("Installing %s",filename)
|
|
if not self.dry_run:
|
|
f = open(filename,'wb')
|
|
for pkg in nsp:
|
|
pth = tuple(pkg.split('.'))
|
|
trailer = '\n'
|
|
if '.' in pkg:
|
|
trailer = (
|
|
"; m and setattr(sys.modules[%r], %r, m)\n"
|
|
% ('.'.join(pth[:-1]), pth[-1])
|
|
)
|
|
f.write(
|
|
"import sys,new,os; "
|
|
"p = os.path.join(sys._getframe(1).f_locals['sitedir'], "
|
|
"*%(pth)r); "
|
|
"ie = os.path.exists(os.path.join(p,'__init__.py')); "
|
|
"m = not ie and "
|
|
"sys.modules.setdefault(%(pkg)r,new.module(%(pkg)r)); "
|
|
"mp = (m or []) and m.__dict__.setdefault('__path__',[]); "
|
|
"(p not in mp) and mp.append(p)%(trailer)s"
|
|
% locals()
|
|
)
|
|
f.close()
|
|
|
|
def _get_all_ns_packages(self):
|
|
nsp = {}
|
|
for pkg in self.distribution.namespace_packages or []:
|
|
pkg = pkg.split('.')
|
|
while pkg:
|
|
nsp['.'.join(pkg)] = 1
|
|
pkg.pop()
|
|
nsp=list(nsp)
|
|
nsp.sort() # set up shorter names first
|
|
return nsp
|
|
|
|
|