90 lines
3.1 KiB
Python
90 lines
3.1 KiB
Python
import distutils.command.install_lib as orig
|
|
import os
|
|
import sys
|
|
|
|
|
|
class install_lib(orig.install_lib):
|
|
"""Don't add compiled flags to filenames of non-Python files"""
|
|
|
|
def initialize_options(self):
|
|
orig.install_lib.initialize_options(self)
|
|
self.multiarch = None
|
|
self.install_layout = None
|
|
|
|
def finalize_options(self):
|
|
orig.install_lib.finalize_options(self)
|
|
self.set_undefined_options('install',('install_layout','install_layout'))
|
|
if self.install_layout == 'deb' and sys.version_info[:2] >= (3, 3):
|
|
import sysconfig
|
|
self.multiarch = sysconfig.get_config_var('MULTIARCH')
|
|
|
|
def run(self):
|
|
self.build()
|
|
outfiles = self.install()
|
|
if outfiles is not None:
|
|
# always compile, in case we have any extension stubs to deal with
|
|
self.byte_compile(outfiles)
|
|
|
|
def get_exclusions(self):
|
|
exclude = {}
|
|
nsp = self.distribution.namespace_packages
|
|
svem = (nsp and self.get_finalized_command('install')
|
|
.single_version_externally_managed)
|
|
if svem:
|
|
for pkg in nsp:
|
|
parts = pkg.split('.')
|
|
while parts:
|
|
pkgdir = os.path.join(self.install_dir, *parts)
|
|
for f in '__init__.py', '__init__.pyc', '__init__.pyo':
|
|
exclude[os.path.join(pkgdir, f)] = 1
|
|
parts.pop()
|
|
return exclude
|
|
|
|
def copy_tree(
|
|
self, infile, outfile,
|
|
preserve_mode=1, preserve_times=1, preserve_symlinks=0, level=1
|
|
):
|
|
assert preserve_mode and preserve_times and not preserve_symlinks
|
|
exclude = self.get_exclusions()
|
|
|
|
if not exclude:
|
|
import distutils.dir_util
|
|
distutils.dir_util._multiarch = self.multiarch
|
|
return orig.install_lib.copy_tree(self, infile, outfile)
|
|
|
|
# Exclude namespace package __init__.py* files from the output
|
|
|
|
from setuptools.archive_util import unpack_directory
|
|
from distutils import log
|
|
|
|
outfiles = []
|
|
|
|
if self.multiarch:
|
|
import sysconfig
|
|
ext_suffix = sysconfig.get_config_var ('EXT_SUFFIX')
|
|
new_suffix = "%s-%s%s" % (ext_suffix[:-3], self.multiarch, ext_suffix[-3:])
|
|
|
|
def pf(src, dst):
|
|
if dst in exclude:
|
|
log.warn("Skipping installation of %s (namespace package)",
|
|
dst)
|
|
return False
|
|
|
|
if self.multiarch and dst.endswith(ext_suffix) and not dst.endswith(new_suffix):
|
|
dst = dst.replace(ext_suffix, new_suffix)
|
|
log.info("renaming extension to %s", os.path.basename(dst))
|
|
|
|
log.info("copying %s -> %s", src, os.path.dirname(dst))
|
|
outfiles.append(dst)
|
|
return dst
|
|
|
|
unpack_directory(infile, outfile, pf)
|
|
return outfiles
|
|
|
|
def get_outputs(self):
|
|
outputs = orig.install_lib.get_outputs(self)
|
|
exclude = self.get_exclusions()
|
|
if exclude:
|
|
return [f for f in outputs if f not in exclude]
|
|
return outputs
|