| from distutils import log
|
| import distutils.command.install_scripts as orig
|
| from distutils.errors import DistutilsModuleError
|
| import os
|
| import sys
|
|
|
| from pkg_resources import Distribution, PathMetadata
|
| from .._path import ensure_directory
|
|
|
|
|
| class install_scripts(orig.install_scripts):
|
| """Do normal script install, plus any egg_info wrapper scripts"""
|
|
|
| def initialize_options(self):
|
| orig.install_scripts.initialize_options(self)
|
| self.no_ep = False
|
|
|
| def run(self):
|
| import setuptools.command.easy_install as ei
|
|
|
| self.run_command("egg_info")
|
| if self.distribution.scripts:
|
| orig.install_scripts.run(self)
|
| else:
|
| self.outfiles = []
|
| if self.no_ep:
|
|
|
| return
|
|
|
| ei_cmd = self.get_finalized_command("egg_info")
|
| dist = Distribution(
|
| ei_cmd.egg_base, PathMetadata(ei_cmd.egg_base, ei_cmd.egg_info),
|
| ei_cmd.egg_name, ei_cmd.egg_version,
|
| )
|
| bs_cmd = self.get_finalized_command('build_scripts')
|
| exec_param = getattr(bs_cmd, 'executable', None)
|
| try:
|
| bw_cmd = self.get_finalized_command("bdist_wininst")
|
| is_wininst = getattr(bw_cmd, '_is_running', False)
|
| except (ImportError, DistutilsModuleError):
|
| is_wininst = False
|
| writer = ei.ScriptWriter
|
| if is_wininst:
|
| exec_param = "python.exe"
|
| writer = ei.WindowsScriptWriter
|
| if exec_param == sys.executable:
|
|
|
|
|
| exec_param = [exec_param]
|
|
|
| writer = writer.best()
|
| cmd = writer.command_spec_class.best().from_param(exec_param)
|
| for args in writer.get_args(dist, cmd.as_header()):
|
| self.write_script(*args)
|
|
|
| def write_script(self, script_name, contents, mode="t", *ignored):
|
| """Write an executable file to the scripts directory"""
|
| from setuptools.command.easy_install import chmod, current_umask
|
|
|
| log.info("Installing %s script to %s", script_name, self.install_dir)
|
| target = os.path.join(self.install_dir, script_name)
|
| self.outfiles.append(target)
|
|
|
| mask = current_umask()
|
| if not self.dry_run:
|
| ensure_directory(target)
|
| f = open(target, "w" + mode)
|
| f.write(contents)
|
| f.close()
|
| chmod(target, 0o777 - mask)
|
|
|