mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 13:41:24 +00:00 
			
		
		
		
	 a19cdad6dc
			
		
	
	
		a19cdad6dc
		
	
	
	
	
		
			
			install.c: support for a 'pre-install-script', run before anything has been installed. Provides a 'message_box' module function for use by either the pre-install or post-install scripts. bdist_wininst.py: support for pre-install script. Typo (build->built), fixes so that --target-version can still work, even when the distribution has extension modules - in this case, we insist on --skip-build, as we still can't actually build other versions.
		
			
				
	
	
		
			268 lines
		
	
	
	
		
			10 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			268 lines
		
	
	
	
		
			10 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| """distutils.command.bdist_wininst
 | |
| 
 | |
| Implements the Distutils 'bdist_wininst' command: create a windows installer
 | |
| exe-program."""
 | |
| 
 | |
| # This module should be kept compatible with Python 1.5.2.
 | |
| 
 | |
| __revision__ = "$Id$"
 | |
| 
 | |
| import sys, os, string
 | |
| from distutils.core import Command
 | |
| from distutils.util import get_platform
 | |
| from distutils.dir_util import create_tree, remove_tree
 | |
| from distutils.errors import *
 | |
| from distutils.sysconfig import get_python_version
 | |
| from distutils import log
 | |
| 
 | |
| class bdist_wininst (Command):
 | |
| 
 | |
|     description = "create an executable installer for MS Windows"
 | |
| 
 | |
|     user_options = [('bdist-dir=', None,
 | |
|                      "temporary directory for creating the distribution"),
 | |
|                     ('keep-temp', 'k',
 | |
|                      "keep the pseudo-installation tree around after " +
 | |
|                      "creating the distribution archive"),
 | |
|                     ('target-version=', 'v',
 | |
|                      "require a specific python version" +
 | |
|                      " on the target system"),
 | |
|                     ('no-target-compile', 'c',
 | |
|                      "do not compile .py to .pyc on the target system"),
 | |
|                     ('no-target-optimize', 'o',
 | |
|                      "do not compile .py to .pyo (optimized)"
 | |
|                      "on the target system"),
 | |
|                     ('dist-dir=', 'd',
 | |
|                      "directory to put final built distributions in"),
 | |
|                     ('bitmap=', 'b',
 | |
|                      "bitmap to use for the installer instead of python-powered logo"),
 | |
|                     ('title=', 't',
 | |
|                      "title to display on the installer background instead of default"),
 | |
|                     ('skip-build', None,
 | |
|                      "skip rebuilding everything (for testing/debugging)"),
 | |
|                     ('install-script=', None,
 | |
|                      "basename of installation script to be run after"
 | |
|                      "installation or before deinstallation"),
 | |
|                     ('pre-install-script=', None,
 | |
|                      "Fully qualified filename of a script to be run before "
 | |
|                      "any files are installed.  This script need not be in the "
 | |
|                      "distribution"),
 | |
|                    ]
 | |
| 
 | |
|     boolean_options = ['keep-temp', 'no-target-compile', 'no-target-optimize',
 | |
|                        'skip-build']
 | |
| 
 | |
|     def initialize_options (self):
 | |
|         self.bdist_dir = None
 | |
|         self.keep_temp = 0
 | |
|         self.no_target_compile = 0
 | |
|         self.no_target_optimize = 0
 | |
|         self.target_version = None
 | |
|         self.dist_dir = None
 | |
|         self.bitmap = None
 | |
|         self.title = None
 | |
|         self.skip_build = 0
 | |
|         self.install_script = None
 | |
|         self.pre_install_script = None
 | |
| 
 | |
|     # initialize_options()
 | |
| 
 | |
| 
 | |
|     def finalize_options (self):
 | |
|         if self.bdist_dir is None:
 | |
|             bdist_base = self.get_finalized_command('bdist').bdist_base
 | |
|             self.bdist_dir = os.path.join(bdist_base, 'wininst')
 | |
|         if not self.target_version:
 | |
|             self.target_version = ""
 | |
|         if not self.skip_build and self.distribution.has_ext_modules():
 | |
|             short_version = get_python_version()
 | |
|             if self.target_version and self.target_version != short_version:
 | |
|                 raise DistutilsOptionError, \
 | |
|                       "target version can only be %s, or the '--skip_build'" \
 | |
|                       " option must be specified" % (short_version,)
 | |
|             self.target_version = short_version
 | |
| 
 | |
|         self.set_undefined_options('bdist', ('dist_dir', 'dist_dir'))
 | |
| 
 | |
|         if self.install_script:
 | |
|             for script in self.distribution.scripts:
 | |
|                 if self.install_script == os.path.basename(script):
 | |
|                     break
 | |
|             else:
 | |
|                 raise DistutilsOptionError, \
 | |
|                       "install_script '%s' not found in scripts" % \
 | |
|                       self.install_script
 | |
|     # finalize_options()
 | |
| 
 | |
| 
 | |
|     def run (self):
 | |
|         if (sys.platform != "win32" and
 | |
|             (self.distribution.has_ext_modules() or
 | |
|              self.distribution.has_c_libraries())):
 | |
|             raise DistutilsPlatformError \
 | |
|                   ("distribution contains extensions and/or C libraries; "
 | |
|                    "must be compiled on a Windows 32 platform")
 | |
| 
 | |
|         if not self.skip_build:
 | |
|             self.run_command('build')
 | |
| 
 | |
|         install = self.reinitialize_command('install', reinit_subcommands=1)
 | |
|         install.root = self.bdist_dir
 | |
|         install.skip_build = self.skip_build
 | |
|         install.warn_dir = 0
 | |
| 
 | |
|         install_lib = self.reinitialize_command('install_lib')
 | |
|         # we do not want to include pyc or pyo files
 | |
|         install_lib.compile = 0
 | |
|         install_lib.optimize = 0
 | |
|         
 | |
|         # If we are building an installer for a Python version other
 | |
|         # than the one we are currently running, then we need to ensure
 | |
|         # our build_lib reflects the other Python version rather than ours.
 | |
|         # Note that for target_version!=sys.version, we must have skipped the
 | |
|         # build step, so there is no issue with enforcing the build of this
 | |
|         # version.
 | |
|         target_version = self.target_version
 | |
|         if not target_version:
 | |
|             assert self.skip_build, "Should have already checked this"
 | |
|             target_version = sys.version[0:3]
 | |
|         plat_specifier = ".%s-%s" % (get_platform(), target_version)
 | |
|         build = self.get_finalized_command('build')
 | |
|         build.build_lib = os.path.join(build.build_base,
 | |
|                                        'lib' + plat_specifier)
 | |
| 
 | |
|         # Use a custom scheme for the zip-file, because we have to decide
 | |
|         # at installation time which scheme to use.
 | |
|         for key in ('purelib', 'platlib', 'headers', 'scripts', 'data'):
 | |
|             value = string.upper(key)
 | |
|             if key == 'headers':
 | |
|                 value = value + '/Include/$dist_name'
 | |
|             setattr(install,
 | |
|                     'install_' + key,
 | |
|                     value)
 | |
| 
 | |
|         log.info("installing to %s", self.bdist_dir)
 | |
|         install.ensure_finalized()
 | |
| 
 | |
|         # avoid warning of 'install_lib' about installing
 | |
|         # into a directory not in sys.path
 | |
|         sys.path.insert(0, os.path.join(self.bdist_dir, 'PURELIB'))
 | |
| 
 | |
|         install.run()
 | |
| 
 | |
|         del sys.path[0]
 | |
| 
 | |
|         # And make an archive relative to the root of the
 | |
|         # pseudo-installation tree.
 | |
|         from tempfile import mktemp
 | |
|         archive_basename = mktemp()
 | |
|         fullname = self.distribution.get_fullname()
 | |
|         arcname = self.make_archive(archive_basename, "zip",
 | |
|                                     root_dir=self.bdist_dir)
 | |
|         # create an exe containing the zip-file
 | |
|         self.create_exe(arcname, fullname, self.bitmap)
 | |
|         # remove the zip-file again
 | |
|         log.debug("removing temporary file '%s'", arcname)
 | |
|         os.remove(arcname)
 | |
| 
 | |
|         if not self.keep_temp:
 | |
|             remove_tree(self.bdist_dir, dry_run=self.dry_run)
 | |
| 
 | |
|     # run()
 | |
| 
 | |
|     def get_inidata (self):
 | |
|         # Return data describing the installation.
 | |
| 
 | |
|         lines = []
 | |
|         metadata = self.distribution.metadata
 | |
| 
 | |
|         # Write the [metadata] section.  Values are written with
 | |
|         # repr()[1:-1], so they do not contain unprintable characters, and
 | |
|         # are not surrounded by quote chars.
 | |
|         lines.append("[metadata]")
 | |
| 
 | |
|         # 'info' will be displayed in the installer's dialog box,
 | |
|         # describing the items to be installed.
 | |
|         info = (metadata.long_description or '') + '\n'
 | |
| 
 | |
|         for name in ["author", "author_email", "description", "maintainer",
 | |
|                      "maintainer_email", "name", "url", "version"]:
 | |
|             data = getattr(metadata, name, "")
 | |
|             if data:
 | |
|                 info = info + ("\n    %s: %s" % \
 | |
|                                (string.capitalize(name), data))
 | |
|                 lines.append("%s=%s" % (name, repr(data)[1:-1]))
 | |
| 
 | |
|         # The [setup] section contains entries controlling
 | |
|         # the installer runtime.
 | |
|         lines.append("\n[Setup]")
 | |
|         if self.install_script:
 | |
|             lines.append("install_script=%s" % self.install_script)
 | |
|         lines.append("info=%s" % repr(info)[1:-1])
 | |
|         lines.append("target_compile=%d" % (not self.no_target_compile))
 | |
|         lines.append("target_optimize=%d" % (not self.no_target_optimize))
 | |
|         if self.target_version:
 | |
|             lines.append("target_version=%s" % self.target_version)
 | |
| 
 | |
|         title = self.title or self.distribution.get_fullname()
 | |
|         lines.append("title=%s" % repr(title)[1:-1])
 | |
|         import time
 | |
|         import distutils
 | |
|         build_info = "Built %s with distutils-%s" % \
 | |
|                      (time.ctime(time.time()), distutils.__version__)
 | |
|         lines.append("build_info=%s" % build_info)
 | |
|         return string.join(lines, "\n")
 | |
| 
 | |
|     # get_inidata()
 | |
| 
 | |
|     def create_exe (self, arcname, fullname, bitmap=None):
 | |
|         import struct
 | |
| 
 | |
|         self.mkpath(self.dist_dir)
 | |
| 
 | |
|         cfgdata = self.get_inidata()
 | |
| 
 | |
|         if self.target_version:
 | |
|             # if we create an installer for a specific python version,
 | |
|             # it's better to include this in the name
 | |
|             installer_name = os.path.join(self.dist_dir,
 | |
|                                           "%s.win32-py%s.exe" %
 | |
|                                            (fullname, self.target_version))
 | |
|         else:
 | |
|             installer_name = os.path.join(self.dist_dir,
 | |
|                                           "%s.win32.exe" % fullname)
 | |
|         self.announce("creating %s" % installer_name)
 | |
| 
 | |
|         if bitmap:
 | |
|             bitmapdata = open(bitmap, "rb").read()
 | |
|             bitmaplen = len(bitmapdata)
 | |
|         else:
 | |
|             bitmaplen = 0
 | |
| 
 | |
|         file = open(installer_name, "wb")
 | |
|         file.write(self.get_exe_bytes())
 | |
|         if bitmap:
 | |
|             file.write(bitmapdata)
 | |
| 
 | |
|         # Append the pre-install script
 | |
|         cfgdata = cfgdata + "\0"
 | |
|         if self.pre_install_script:
 | |
|             script_data = open(self.pre_install_script, "r").read()
 | |
|             cfgdata = cfgdata + script_data + "\n\0"
 | |
|         file.write(cfgdata)
 | |
|         header = struct.pack("<iii",
 | |
|                              0x1234567A,       # tag
 | |
|                              len(cfgdata),     # length
 | |
|                              bitmaplen,        # number of bytes in bitmap
 | |
|                              )
 | |
|         file.write(header)
 | |
|         file.write(open(arcname, "rb").read())
 | |
| 
 | |
|     # create_exe()
 | |
| 
 | |
|     def get_exe_bytes (self):
 | |
|         # wininst.exe is in the same directory as this file
 | |
|         directory = os.path.dirname(__file__)
 | |
|         filename = os.path.join(directory, "wininst.exe")
 | |
|         return open(filename, "rb").read()
 | |
| # class bdist_wininst
 |