| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193 | from distutils.util import convert_pathfrom distutils import logfrom distutils.errors import DistutilsError, DistutilsOptionErrorimport osimport globimport ioimport pkg_resourcesfrom setuptools.command.easy_install import easy_installfrom setuptools import namespacesimport setuptoolsclass 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
 |