194 lines
6.8 KiB
Python
194 lines
6.8 KiB
Python
|
from distutils.util import convert_path
|
||
|
|
from distutils import log
|
||
|
|
from distutils.errors import DistutilsError, DistutilsOptionError
|
||
|
|
import os
|
||
|
|
import glob
|
||
|
|
import io
|
||
|
|
|
||
|
|
import pkg_resources
|
||
|
|
from setuptools.command.easy_install import easy_install
|
||
|
|
from setuptools import namespaces
|
||
|
|
import setuptools
|
||
|
|
|
||
|
|
|
||
|
|
class develop(namespaces.DevelopInstaller, easy_install):
|
||
|
|
"""Set up package for development"""
|
||
|
|
|
||
|
|
description = "install package in 'development mode'"
|
||
|
|
|
||
|
|
user_options = easy_install.user_options + [
|
||
|
|
("uninstall", "u", "Uninstall this source package"),
|
||
|
|
("egg-path=", None, "Set the path to be used in the .egg-link file"),
|
||
|
|
]
|
||
|
|
|
||
|
|
boolean_options = easy_install.boolean_options + ['uninstall']
|
||
|
|
|
||
|
|
command_consumes_arguments = False # override base
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
if self.uninstall:
|
||
|
|
self.multi_version = True
|
||
|
|
self.uninstall_link()
|
||
|
|
self.uninstall_namespaces()
|
||
|
|
else:
|
||
|
|
self.install_for_development()
|
||
|
|
self.warn_deprecated_options()
|
||
|
|
|
||
|
|
def initialize_options(self):
|
||
|
|
self.uninstall = None
|
||
|
|
self.egg_path = None
|
||
|
|
easy_install.initialize_options(self)
|
||
|
|
self.setup_path = None
|
||
|
|
self.always_copy_from = '.' # always copy eggs installed in curdir
|
||
|
|
|
||
|
|
def finalize_options(self):
|
||
|
|
ei = self.get_finalized_command("egg_info")
|
||
|
|
if ei.broken_egg_info:
|
||
|
|
template = "Please rename %r to %r before using 'develop'"
|
||
|
|
args = ei.egg_info, ei.broken_egg_info
|
||
|
|
raise DistutilsError(template % args)
|
||
|
|
self.args = [ei.egg_name]
|
||
|
|
|
||
|
|
easy_install.finalize_options(self)
|
||
|
|
self.expand_basedirs()
|
||
|
|
self.expand_dirs()
|
||
|
|
# pick up setup-dir .egg files only: no .egg-info
|
||
|
|
self.package_index.scan(glob.glob('*.egg'))
|
||
|
|
|
||
|
|
egg_link_fn = ei.egg_name + '.egg-link'
|
||
|
|
self.egg_link = os.path.join(self.install_dir, egg_link_fn)
|
||
|
|
self.egg_base = ei.egg_base
|
||
|
|
if self.egg_path is None:
|
||
|
|
self.egg_path = os.path.abspath(ei.egg_base)
|
||
|
|
|
||
|
|
target = pkg_resources.normalize_path(self.egg_base)
|
||
|
|
egg_path = pkg_resources.normalize_path(
|
||
|
|
os.path.join(self.install_dir, self.egg_path)
|
||
|
|
)
|
||
|
|
if egg_path != target:
|
||
|
|
raise DistutilsOptionError(
|
||
|
|
"--egg-path must be a relative path from the install"
|
||
|
|
" directory to " + target
|
||
|
|
)
|
||
|
|
|
||
|
|
# Make a distribution for the package's source
|
||
|
|
self.dist = pkg_resources.Distribution(
|
||
|
|
target,
|
||
|
|
pkg_resources.PathMetadata(target, os.path.abspath(ei.egg_info)),
|
||
|
|
project_name=ei.egg_name,
|
||
|
|
)
|
||
|
|
|
||
|
|
self.setup_path = self._resolve_setup_path(
|
||
|
|
self.egg_base,
|
||
|
|
self.install_dir,
|
||
|
|
self.egg_path,
|
||
|
|
)
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _resolve_setup_path(egg_base, install_dir, egg_path):
|
||
|
|
"""
|
||
|
|
Generate a path from egg_base back to '.' where the
|
||
|
|
setup script resides and ensure that path points to the
|
||
|
|
setup path from $install_dir/$egg_path.
|
||
|
|
"""
|
||
|
|
path_to_setup = egg_base.replace(os.sep, '/').rstrip('/')
|
||
|
|
if path_to_setup != os.curdir:
|
||
|
|
path_to_setup = '../' * (path_to_setup.count('/') + 1)
|
||
|
|
resolved = pkg_resources.normalize_path(
|
||
|
|
os.path.join(install_dir, egg_path, path_to_setup)
|
||
|
|
)
|
||
|
|
if resolved != pkg_resources.normalize_path(os.curdir):
|
||
|
|
raise DistutilsOptionError(
|
||
|
|
"Can't get a consistent path to setup script from"
|
||
|
|
" installation directory",
|
||
|
|
resolved,
|
||
|
|
pkg_resources.normalize_path(os.curdir),
|
||
|
|
)
|
||
|
|
return path_to_setup
|
||
|
|
|
||
|
|
def install_for_development(self):
|
||
|
|
self.run_command('egg_info')
|
||
|
|
|
||
|
|
# Build extensions in-place
|
||
|
|
self.reinitialize_command('build_ext', inplace=1)
|
||
|
|
self.run_command('build_ext')
|
||
|
|
|
||
|
|
if setuptools.bootstrap_install_from:
|
||
|
|
self.easy_install(setuptools.bootstrap_install_from)
|
||
|
|
setuptools.bootstrap_install_from = None
|
||
|
|
|
||
|
|
self.install_namespaces()
|
||
|
|
|
||
|
|
# create an .egg-link in the installation dir, pointing to our egg
|
||
|
|
log.info("Creating %s (link to %s)", self.egg_link, self.egg_base)
|
||
|
|
if not self.dry_run:
|
||
|
|
with open(self.egg_link, "w") as f:
|
||
|
|
f.write(self.egg_path + "\n" + self.setup_path)
|
||
|
|
# postprocess the installed distro, fixing up .pth, installing scripts,
|
||
|
|
# and handling requirements
|
||
|
|
self.process_distribution(None, self.dist, not self.no_deps)
|
||
|
|
|
||
|
|
def uninstall_link(self):
|
||
|
|
if os.path.exists(self.egg_link):
|
||
|
|
log.info("Removing %s (link to %s)", self.egg_link, self.egg_base)
|
||
|
|
egg_link_file = open(self.egg_link)
|
||
|
|
contents = [line.rstrip() for line in egg_link_file]
|
||
|
|
egg_link_file.close()
|
||
|
|
if contents not in ([self.egg_path], [self.egg_path, self.setup_path]):
|
||
|
|
log.warn("Link points to %s: uninstall aborted", contents)
|
||
|
|
return
|
||
|
|
if not self.dry_run:
|
||
|
|
os.unlink(self.egg_link)
|
||
|
|
if not self.dry_run:
|
||
|
|
self.update_pth(self.dist) # remove any .pth link to us
|
||
|
|
if self.distribution.scripts:
|
||
|
|
# XXX should also check for entry point scripts!
|
||
|
|
log.warn("Note: you must uninstall or replace scripts manually!")
|
||
|
|
|
||
|
|
def install_egg_scripts(self, dist):
|
||
|
|
if dist is not self.dist:
|
||
|
|
# Installing a dependency, so fall back to normal behavior
|
||
|
|
return easy_install.install_egg_scripts(self, dist)
|
||
|
|
|
||
|
|
# create wrapper scripts in the script dir, pointing to dist.scripts
|
||
|
|
|
||
|
|
# new-style...
|
||
|
|
self.install_wrapper_scripts(dist)
|
||
|
|
|
||
|
|
# ...and old-style
|
||
|
|
for script_name in self.distribution.scripts or []:
|
||
|
|
script_path = os.path.abspath(convert_path(script_name))
|
||
|
|
script_name = os.path.basename(script_path)
|
||
|
|
with io.open(script_path) as strm:
|
||
|
|
script_text = strm.read()
|
||
|
|
self.install_script(dist, script_name, script_text, script_path)
|
||
|
|
|
||
|
|
def install_wrapper_scripts(self, dist):
|
||
|
|
dist = VersionlessRequirement(dist)
|
||
|
|
return easy_install.install_wrapper_scripts(self, dist)
|
||
|
|
|
||
|
|
|
||
|
|
class VersionlessRequirement:
|
||
|
|
"""
|
||
|
|
Adapt a pkg_resources.Distribution to simply return the project
|
||
|
|
name as the 'requirement' so that scripts will work across
|
||
|
|
multiple versions.
|
||
|
|
|
||
|
|
>>> from pkg_resources import Distribution
|
||
|
|
>>> dist = Distribution(project_name='foo', version='1.0')
|
||
|
|
>>> str(dist.as_requirement())
|
||
|
|
'foo==1.0'
|
||
|
|
>>> adapted_dist = VersionlessRequirement(dist)
|
||
|
|
>>> str(adapted_dist.as_requirement())
|
||
|
|
'foo'
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(self, dist):
|
||
|
|
self.__dist = dist
|
||
|
|
|
||
|
|
def __getattr__(self, name):
|
||
|
|
return getattr(self.__dist, name)
|
||
|
|
|
||
|
|
def as_requirement(self):
|
||
|
|
return self.project_name
|