cpython/Lib/distutils/command/install_py.py

81 lines
2.9 KiB
Python

# created 1999/03/13, Greg Ward
__rcsid__ = "$Id$"
import sys, string
from distutils.core import Command
from distutils.util import copy_tree
class InstallPy (Command):
options = [('dir=', 'd', "directory to install to"),
('build-dir=','b', "build directory (where to install from)"),
('compile', 'c', "compile .py to .pyc"),
('optimize', 'o', "compile .py to .pyo (optimized)"),
]
def set_default_options (self):
# let the 'install' command dictate our installation directory
self.dir = None
self.build_dir = None
self.compile = 1
self.optimize = 1
def set_final_options (self):
# Find out from the 'build_ext' command if we were asked to build
# any extensions. If so, that means even pure-Python modules in
# this distribution have to be installed to the "platlib"
# directory.
extensions = self.get_peer_option ('build_ext', 'extensions')
if extensions:
dir_option = 'install_site_platlib'
else:
dir_option = 'install_site_lib'
# Get all the information we need to install pure Python modules
# from the umbrella 'install' command -- build (source) directory,
# install (target) directory, and whether to compile .py files.
self.set_undefined_options ('install',
('build_lib', 'build_dir'),
(dir_option, 'dir'),
('compile_py', 'compile'),
('optimize_py', 'optimize'))
def run (self):
self.set_final_options ()
# Dump entire contents of the build directory to the installation
# directory (that's the beauty of having a build directory!)
outfiles = self.copy_tree (self.build_dir, self.dir)
# (Optionally) compile .py to .pyc
# XXX hey! we can't control whether we optimize or not; that's up
# to the invocation of the current Python interpreter (at least
# according to the py_compile docs). That sucks.
if self.compile:
from py_compile import compile
for f in outfiles:
# XXX can't assume this filename mapping!
# only compile the file if it is actually a .py file
if f[-3:] == '.py':
out_fn = string.replace (f, '.py', '.pyc')
self.make_file (f, out_fn, compile, (f,),
"compiling %s -> %s" % (f, out_fn),
"compilation of %s skipped" % f)
# XXX ignore self.optimize for now, since we don't really know if
# we're compiling optimally or not, and couldn't pick what to do
# even if we did know. ;-(
# run ()
# class InstallPy