setup.py
441 lines
| 15.0 KiB
| text/x-python
|
PythonLexer
|
r575 | # | ||
# This is the mercurial setup script. | ||||
|
r0 | # | ||
|
r4816 | # 'python setup.py install', or | ||
# 'python setup.py --help' for more options | ||||
|
r0 | |||
|
r14295 | import sys, platform | ||
|
r8286 | if not hasattr(sys, 'version_info') or sys.version_info < (2, 4, 0, 'final'): | ||
raise SystemExit("Mercurial requires Python 2.4 or later.") | ||||
|
r1873 | |||
|
r11532 | if sys.version_info[0] >= 3: | ||
def b(s): | ||||
'''A helper function to emulate 2.6+ bytes literals using string | ||||
literals.''' | ||||
return s.encode('latin1') | ||||
else: | ||||
def b(s): | ||||
'''A helper function to emulate 2.6+ bytes literals using string | ||||
literals.''' | ||||
return s | ||||
|
r7558 | # Solaris Python packaging brain damage | ||
try: | ||||
import hashlib | ||||
sha = hashlib.sha1() | ||||
except: | ||||
try: | ||||
import sha | ||||
except: | ||||
raise SystemExit( | ||||
"Couldn't import standard hashlib (incomplete Python install).") | ||||
try: | ||||
import zlib | ||||
except: | ||||
raise SystemExit( | ||||
"Couldn't import standard zlib (incomplete Python install).") | ||||
|
r14295 | # The base IronPython distribution (as of 2.7.1) doesn't support bz2 | ||
isironpython = False | ||||
|
r10761 | try: | ||
|
r14295 | isironpython = platform.python_implementation().lower().find("ironpython") != -1 | ||
|
r10761 | except: | ||
|
r14295 | pass | ||
if isironpython: | ||||
print "warning: IronPython detected (no bz2 support)" | ||||
else: | ||||
try: | ||||
import bz2 | ||||
except: | ||||
raise SystemExit( | ||||
"Couldn't import standard bz2 (incomplete Python install).") | ||||
|
r10761 | |||
|
r8547 | import os, subprocess, time | ||
|
r6251 | import shutil | ||
import tempfile | ||||
|
r11468 | from distutils import log | ||
|
r14538 | from distutils.core import setup, Command, Extension | ||
|
r7722 | from distutils.dist import Distribution | ||
|
r7649 | from distutils.command.build import build | ||
|
r11468 | from distutils.command.build_ext import build_ext | ||
|
r7722 | from distutils.command.build_py import build_py | ||
|
r12661 | from distutils.command.install_scripts import install_scripts | ||
|
r7649 | from distutils.spawn import spawn, find_executable | ||
|
r6245 | from distutils.ccompiler import new_compiler | ||
|
r14538 | from distutils.errors import CCompilerError, DistutilsExecError | ||
|
r12649 | from distutils.sysconfig import get_python_inc | ||
|
r13594 | from distutils.version import StrictVersion | ||
|
r157 | |||
|
r6513 | scripts = ['hg'] | ||
if os.name == 'nt': | ||||
scripts.append('contrib/win32/hg.bat') | ||||
|
r3893 | |||
|
r6251 | # simplified version of distutils.ccompiler.CCompiler.has_function | ||
# that actually removes its temporary files. | ||||
|
r10000 | def hasfunction(cc, funcname): | ||
|
r6251 | tmpdir = tempfile.mkdtemp(prefix='hg-install-') | ||
|
r6373 | devnull = oldstderr = None | ||
|
r6251 | try: | ||
try: | ||||
|
r6373 | fname = os.path.join(tmpdir, 'funcname.c') | ||
f = open(fname, 'w') | ||||
f.write('int main(void) {\n') | ||||
f.write(' %s();\n' % funcname) | ||||
f.write('}\n') | ||||
f.close() | ||||
# Redirect stderr to /dev/null to hide any error messages | ||||
# from the compiler. | ||||
# This will have to be changed if we ever have to check | ||||
# for a function on Windows. | ||||
devnull = open('/dev/null', 'w') | ||||
oldstderr = os.dup(sys.stderr.fileno()) | ||||
os.dup2(devnull.fileno(), sys.stderr.fileno()) | ||||
|
r9124 | objects = cc.compile([fname], output_dir=tmpdir) | ||
|
r6251 | cc.link_executable(objects, os.path.join(tmpdir, "a.out")) | ||
except: | ||||
return False | ||||
return True | ||||
finally: | ||||
|
r6373 | if oldstderr is not None: | ||
os.dup2(oldstderr, sys.stderr.fileno()) | ||||
if devnull is not None: | ||||
devnull.close() | ||||
|
r6251 | shutil.rmtree(tmpdir) | ||
|
r1283 | # py2exe needs to be installed to work | ||
try: | ||||
|
r1294 | import py2exe | ||
|
r10400 | py2exeloaded = True | ||
|
r1284 | except ImportError: | ||
|
r10400 | py2exeloaded = False | ||
|
r1283 | |||
|
r9807 | def runcmd(cmd, env): | ||
|
r9615 | p = subprocess.Popen(cmd, stdout=subprocess.PIPE, | ||
stderr=subprocess.PIPE, env=env) | ||||
out, err = p.communicate() | ||||
|
r13636 | return out, err | ||
def runhg(cmd, env): | ||||
out, err = runcmd(cmd, env) | ||||
|
r9615 | # If root is executing setup.py, but the repository is owned by | ||
# another user (as in "sudo python setup.py install") we will get | ||||
# trust warnings since the .hg/hgrc file is untrusted. That is | ||||
|
r10120 | # fine, we don't want to load it anyway. Python may warn about | ||
# a missing __init__.py in mercurial/locale, we also ignore that. | ||||
|
r9615 | err = [e for e in err.splitlines() | ||
|
r11532 | if not e.startswith(b('Not trusting file')) \ | ||
and not e.startswith(b('warning: Not importing'))] | ||||
|
r9615 | if err: | ||
return '' | ||||
return out | ||||
version = '' | ||||
|
r8548 | |||
|
r8547 | if os.path.isdir('.hg'): | ||
|
r8629 | # Execute hg out of this directory with a custom environment which | ||
# includes the pure Python modules in mercurial/pure. We also take | ||||
# care to not use any hgrc files and do no localization. | ||||
pypath = ['mercurial', os.path.join('mercurial', 'pure')] | ||||
env = {'PYTHONPATH': os.pathsep.join(pypath), | ||||
'HGRCPATH': '', | ||||
'LANGUAGE': 'C'} | ||||
|
r9856 | if 'LD_LIBRARY_PATH' in os.environ: | ||
env['LD_LIBRARY_PATH'] = os.environ['LD_LIBRARY_PATH'] | ||||
|
r8648 | if 'SystemRoot' in os.environ: | ||
# Copy SystemRoot into the custom environment for Python 2.6 | ||||
# under Windows. Otherwise, the subprocess will fail with | ||||
# error 0xc0150004. See: http://bugs.python.org/issue3440 | ||||
env['SystemRoot'] = os.environ['SystemRoot'] | ||||
|
r8547 | cmd = [sys.executable, 'hg', 'id', '-i', '-t'] | ||
|
r13636 | l = runhg(cmd, env).split() | ||
|
r9615 | while len(l) > 1 and l[-1][0].isalpha(): # remove non-numbered tags | ||
l.pop() | ||||
if len(l) > 1: # tag found | ||||
version = l[-1] | ||||
if l[0].endswith('+'): # propagate the dirty status to the tag | ||||
version += '+' | ||||
elif len(l) == 1: # no tag found | ||||
cmd = [sys.executable, 'hg', 'parents', '--template', | ||||
'{latesttag}+{latesttagdistance}-'] | ||||
|
r13636 | version = runhg(cmd, env) + l[0] | ||
|
r9615 | if version.endswith('+'): | ||
version += time.strftime('%Y%m%d') | ||||
elif os.path.exists('.hg_archival.txt'): | ||||
|
r10124 | kw = dict([[t.strip() for t in l.split(':', 1)] | ||
for l in open('.hg_archival.txt')]) | ||||
|
r9615 | if 'tag' in kw: | ||
version = kw['tag'] | ||||
elif 'latesttag' in kw: | ||||
version = '%(latesttag)s+%(latesttagdistance)s-%(node).12s' % kw | ||||
|
r8547 | else: | ||
|
r9615 | version = kw.get('node', '')[:12] | ||
|
r7632 | |||
|
r8548 | if version: | ||
|
r9031 | f = open("mercurial/__version__.py", "w") | ||
|
r8548 | f.write('# this file is autogenerated by setup.py\n') | ||
f.write('version = "%s"\n' % version) | ||||
f.close() | ||||
|
r8493 | |||
try: | ||||
from mercurial import __version__ | ||||
version = __version__.version | ||||
except ImportError: | ||||
version = 'unknown' | ||||
|
r7632 | |||
|
r10000 | class hgbuildmo(build): | ||
|
r7649 | |||
description = "build translations (.mo files)" | ||||
def run(self): | ||||
if not find_executable('msgfmt'): | ||||
self.warn("could not find msgfmt executable, no translations " | ||||
"will be built") | ||||
return | ||||
podir = 'i18n' | ||||
if not os.path.isdir(podir): | ||||
self.warn("could not find %s/ directory" % podir) | ||||
return | ||||
join = os.path.join | ||||
for po in os.listdir(podir): | ||||
if not po.endswith('.po'): | ||||
continue | ||||
pofile = join(podir, po) | ||||
modir = join('locale', po[:-3], 'LC_MESSAGES') | ||||
mofile = join(modir, 'hg.mo') | ||||
|
r9999 | mobuildfile = join('mercurial', mofile) | ||
cmd = ['msgfmt', '-v', '-o', mobuildfile, pofile] | ||||
|
r7720 | if sys.platform != 'sunos5': | ||
# msgfmt on Solaris does not know about -c | ||||
cmd.append('-c') | ||||
|
r9999 | self.mkpath(join('mercurial', modir)) | ||
self.make_file([pofile], mobuildfile, spawn, (cmd,)) | ||||
|
r7649 | |||
|
r12661 | |||
|
r10000 | # Insert hgbuildmo first so that files in mercurial/locale/ are found | ||
|
r9999 | # when build_py is run next. | ||
build.sub_commands.insert(0, ('build_mo', None)) | ||||
|
r7649 | |||
|
r7722 | Distribution.pure = 0 | ||
Distribution.global_options.append(('pure', None, "use pure (slow) Python " | ||||
"code instead of C extensions")) | ||||
|
r11468 | class hgbuildext(build_ext): | ||
def build_extension(self, ext): | ||||
try: | ||||
build_ext.build_extension(self, ext) | ||||
except CCompilerError: | ||||
|
r12501 | if not getattr(ext, 'optional', False): | ||
|
r11468 | raise | ||
log.warn("Failed to build optional extension '%s' (skipping)", | ||||
ext.name) | ||||
|
r10000 | class hgbuildpy(build_py): | ||
|
r7722 | |||
def finalize_options(self): | ||||
build_py.finalize_options(self) | ||||
if self.distribution.pure: | ||||
if self.py_modules is None: | ||||
self.py_modules = [] | ||||
for ext in self.distribution.ext_modules: | ||||
if ext.name.startswith("mercurial."): | ||||
self.py_modules.append("mercurial.pure.%s" % ext.name[10:]) | ||||
self.distribution.ext_modules = [] | ||||
|
r12649 | else: | ||
if not os.path.exists(os.path.join(get_python_inc(), 'Python.h')): | ||||
raise SystemExit("Python headers are required to build Mercurial") | ||||
|
r7722 | |||
def find_modules(self): | ||||
modules = build_py.find_modules(self) | ||||
for module in modules: | ||||
if module[0] == "mercurial.pure": | ||||
if module[1] != "__init__": | ||||
yield ("mercurial", module[1], module[2]) | ||||
else: | ||||
yield module | ||||
|
r14538 | class buildhgextindex(Command): | ||
description = 'generate prebuilt index of hgext (for frozen package)' | ||||
user_options = [] | ||||
_indexfilename = 'hgext/__index__.py' | ||||
def initialize_options(self): | ||||
pass | ||||
def finalize_options(self): | ||||
pass | ||||
def run(self): | ||||
if os.path.exists(self._indexfilename): | ||||
os.unlink(self._indexfilename) | ||||
# here no extension enabled, disabled() lists up everything | ||||
code = ('import pprint; from mercurial import extensions; ' | ||||
'pprint.pprint(extensions.disabled())') | ||||
out, err = runcmd([sys.executable, '-c', code], env) | ||||
if err: | ||||
raise DistutilsExecError(err) | ||||
f = open(self._indexfilename, 'w') | ||||
f.write('# this file is autogenerated by setup.py\n') | ||||
f.write('docs = ') | ||||
f.write(out) | ||||
f.close() | ||||
|
r12661 | class hginstallscripts(install_scripts): | ||
''' | ||||
This is a specialization of install_scripts that replaces the @LIBDIR@ with | ||||
the configured directory for modules. If possible, the path is made relative | ||||
to the directory for scripts. | ||||
''' | ||||
def initialize_options(self): | ||||
install_scripts.initialize_options(self) | ||||
self.install_lib = None | ||||
def finalize_options(self): | ||||
install_scripts.finalize_options(self) | ||||
self.set_undefined_options('install', | ||||
('install_lib', 'install_lib')) | ||||
def run(self): | ||||
install_scripts.run(self) | ||||
if (os.path.splitdrive(self.install_dir)[0] != | ||||
os.path.splitdrive(self.install_lib)[0]): | ||||
# can't make relative paths from one drive to another, so use an | ||||
# absolute path instead | ||||
libdir = self.install_lib | ||||
else: | ||||
common = os.path.commonprefix((self.install_dir, self.install_lib)) | ||||
rest = self.install_dir[len(common):] | ||||
uplevel = len([n for n in os.path.split(rest) if n]) | ||||
libdir = uplevel * ('..' + os.sep) + self.install_lib[len(common):] | ||||
for outfile in self.outfiles: | ||||
|
r13400 | fp = open(outfile, 'rb') | ||
data = fp.read() | ||||
fp.close() | ||||
|
r12661 | |||
# skip binary files | ||||
if '\0' in data: | ||||
continue | ||||
|
r12676 | data = data.replace('@LIBDIR@', libdir.encode('string_escape')) | ||
|
r13400 | fp = open(outfile, 'wb') | ||
fp.write(data) | ||||
fp.close() | ||||
|
r12661 | |||
|
r10000 | cmdclass = {'build_mo': hgbuildmo, | ||
|
r11468 | 'build_ext': hgbuildext, | ||
|
r12661 | 'build_py': hgbuildpy, | ||
|
r14538 | 'build_hgextindex': buildhgextindex, | ||
|
r12661 | 'install_scripts': hginstallscripts} | ||
|
r3238 | |||
|
r14243 | packages = ['mercurial', 'mercurial.hgweb', | ||
'mercurial.httpclient', 'mercurial.httpclient.tests', | ||||
'hgext', 'hgext.convert', 'hgext.highlight', 'hgext.zeroconf'] | ||||
|
r10521 | |||
pymodules = [] | ||||
|
r10000 | extmodules = [ | ||
|
r6389 | Extension('mercurial.base85', ['mercurial/base85.c']), | ||
|
r5396 | Extension('mercurial.bdiff', ['mercurial/bdiff.c']), | ||
|
r6389 | Extension('mercurial.diffhelpers', ['mercurial/diffhelpers.c']), | ||
Extension('mercurial.mpatch', ['mercurial/mpatch.c']), | ||||
Extension('mercurial.parsers', ['mercurial/parsers.c']), | ||||
|
r5396 | ] | ||
|
r13734 | osutil_ldflags = [] | ||
if sys.platform == 'darwin': | ||||
osutil_ldflags += ['-framework', 'ApplicationServices'] | ||||
|
r10521 | # disable osutil.c under windows + python 2.4 (issue1364) | ||
if sys.platform == 'win32' and sys.version_info < (2, 5, 0, 'final'): | ||||
pymodules.append('mercurial.pure.osutil') | ||||
else: | ||||
|
r13734 | extmodules.append(Extension('mercurial.osutil', ['mercurial/osutil.c'], | ||
extra_link_args=osutil_ldflags)) | ||||
|
r6239 | |||
|
r15163 | if sys.platform.startswith('linux') and os.uname()[2] > '2.6': | ||
|
r8283 | # The inotify extension is only usable with Linux 2.6 kernels. | ||
# You also need a reasonably recent C library. | ||||
|
r11468 | # In any case, if it fails to build the error will be skipped ('optional'). | ||
|
r8283 | cc = new_compiler() | ||
|
r10000 | if hasfunction(cc, 'inotify_add_watch'): | ||
|
r11468 | inotify = Extension('hgext.inotify.linux._inotify', | ||
|
r11533 | ['hgext/inotify/linux/_inotify.c'], | ||
['mercurial']) | ||||
|
r11468 | inotify.optional = True | ||
extmodules.append(inotify) | ||||
|
r8283 | packages.extend(['hgext.inotify', 'hgext.inotify.linux']) | ||
|
r5396 | |||
|
r9999 | packagedata = {'mercurial': ['locale/*/LC_MESSAGES/hg.mo', | ||
'help/*.txt']} | ||||
def ordinarypath(p): | ||||
return p and p[0] != '.' and p[-1] != '~' | ||||
|
r10282 | for root in ('templates',): | ||
|
r9999 | for curdir, dirs, files in os.walk(os.path.join('mercurial', root)): | ||
curdir = curdir.split(os.sep, 1)[1] | ||||
dirs[:] = filter(ordinarypath, dirs) | ||||
for f in filter(ordinarypath, files): | ||||
f = os.path.join(curdir, f) | ||||
packagedata['mercurial'].append(f) | ||||
|
r7648 | datafiles = [] | ||
|
r10400 | setupversion = version | ||
extra = {} | ||||
if py2exeloaded: | ||||
extra['console'] = [ | ||||
{'script':'hg', | ||||
'copyright':'Copyright (C) 2005-2010 Matt Mackall and others', | ||||
'product_version':version}] | ||||
|
r14538 | # sub command of 'build' because 'py2exe' does not handle sub_commands | ||
build.sub_commands.insert(0, ('build_hgextindex', None)) | ||||
|
r10400 | |||
if os.name == 'nt': | ||||
# Windows binary file versions for exe/dll files must have the | ||||
# form W.X.Y.Z, where W,X,Y,Z are numbers in the range 0..65535 | ||||
setupversion = version.split('+', 1)[0] | ||||
|
r7648 | |||
|
r13583 | if sys.platform == 'darwin' and os.path.exists('/usr/bin/xcodebuild'): | ||
# XCode 4.0 dropped support for ppc architecture, which is hardcoded in | ||||
# distutils.sysconfig | ||||
|
r13636 | version = runcmd(['/usr/bin/xcodebuild', '-version'], {})[0].splitlines()[0] | ||
|
r13583 | # Also parse only first digit, because 3.2.1 can't be parsed nicely | ||
if (version.startswith('Xcode') and | ||||
|
r13594 | StrictVersion(version.split()[1]) >= StrictVersion('4.0')): | ||
|
r14324 | os.environ['ARCHFLAGS'] = '' | ||
|
r13583 | |||
|
r1977 | setup(name='mercurial', | ||
|
r10400 | version=setupversion, | ||
|
r3238 | author='Matt Mackall', | ||
author_email='mpm@selenic.com', | ||||
|
r8936 | url='http://mercurial.selenic.com/', | ||
|
r3238 | description='Scalable distributed SCM', | ||
|
r10263 | license='GNU GPLv2+', | ||
|
r6513 | scripts=scripts, | ||
|
r6239 | packages=packages, | ||
|
r10521 | py_modules=pymodules, | ||
|
r10000 | ext_modules=extmodules, | ||
|
r7648 | data_files=datafiles, | ||
|
r9999 | package_data=packagedata, | ||
|
r3238 | cmdclass=cmdclass, | ||
|
r6789 | options=dict(py2exe=dict(packages=['hgext', 'email']), | ||
|
r4628 | bdist_mpkg=dict(zipdist=True, | ||
|
r3238 | license='COPYING', | ||
readme='contrib/macosx/Readme.html', | ||||
welcome='contrib/macosx/Welcome.html')), | ||||
|
r3893 | **extra) | ||