470 lines
16 KiB
Python
470 lines
16 KiB
Python
"""Provide access to Python's configuration information. The specific
|
|
configuration variables available depend heavily on the platform and
|
|
configuration. The values may be retrieved using
|
|
get_config_var(name), and the list of variables is available via
|
|
get_config_vars().keys(). Additional convenience functions are also
|
|
available.
|
|
|
|
Written by: Fred L. Drake, Jr.
|
|
Email: <fdrake@acm.org>
|
|
Initial date: 17-Dec-1998
|
|
"""
|
|
|
|
__revision__ = "$Id$"
|
|
|
|
import os
|
|
import re
|
|
import string
|
|
import sys
|
|
|
|
from errors import DistutilsPlatformError
|
|
|
|
# These are needed in a couple of spots, so just compute them once.
|
|
PREFIX = os.path.normpath(sys.prefix)
|
|
EXEC_PREFIX = os.path.normpath(sys.exec_prefix)
|
|
|
|
# python_build: (Boolean) if true, we're either building Python or
|
|
# building an extension with an un-installed Python, so we use
|
|
# different (hard-wired) directories.
|
|
|
|
argv0_path = os.path.dirname(os.path.abspath(sys.executable))
|
|
landmark = os.path.join(argv0_path, "Modules", "Setup")
|
|
if not os.path.isfile(landmark):
|
|
python_build = 0
|
|
elif os.path.isfile(os.path.join(argv0_path, "Lib", "os.py")):
|
|
python_build = 1
|
|
else:
|
|
python_build = os.path.isfile(os.path.join(os.path.dirname(argv0_path),
|
|
"Lib", "os.py"))
|
|
del argv0_path, landmark
|
|
|
|
|
|
def get_python_inc(plat_specific=0, prefix=None):
|
|
"""Return the directory containing installed Python header files.
|
|
|
|
If 'plat_specific' is false (the default), this is the path to the
|
|
non-platform-specific header files, i.e. Python.h and so on;
|
|
otherwise, this is the path to platform-specific header files
|
|
(namely pyconfig.h).
|
|
|
|
If 'prefix' is supplied, use it instead of sys.prefix or
|
|
sys.exec_prefix -- i.e., ignore 'plat_specific'.
|
|
"""
|
|
if prefix is None:
|
|
prefix = plat_specific and EXEC_PREFIX or PREFIX
|
|
if os.name == "posix":
|
|
if python_build:
|
|
base = os.path.dirname(os.path.abspath(sys.executable))
|
|
if plat_specific:
|
|
inc_dir = base
|
|
else:
|
|
inc_dir = os.path.join(base, "Include")
|
|
if not os.path.exists(inc_dir):
|
|
inc_dir = os.path.join(os.path.dirname(base), "Include")
|
|
return inc_dir
|
|
return os.path.join(prefix, "include", "python" + sys.version[:3])
|
|
elif os.name == "nt":
|
|
return os.path.join(prefix, "include")
|
|
elif os.name == "mac":
|
|
if plat_specific:
|
|
return os.path.join(prefix, "Mac", "Include")
|
|
else:
|
|
return os.path.join(prefix, "Include")
|
|
elif os.name == "os2":
|
|
return os.path.join(prefix, "Include")
|
|
else:
|
|
raise DistutilsPlatformError(
|
|
"I don't know where Python installs its C header files "
|
|
"on platform '%s'" % os.name)
|
|
|
|
|
|
def get_python_lib(plat_specific=0, standard_lib=0, prefix=None):
|
|
"""Return the directory containing the Python library (standard or
|
|
site additions).
|
|
|
|
If 'plat_specific' is true, return the directory containing
|
|
platform-specific modules, i.e. any module from a non-pure-Python
|
|
module distribution; otherwise, return the platform-shared library
|
|
directory. If 'standard_lib' is true, return the directory
|
|
containing standard Python library modules; otherwise, return the
|
|
directory for site-specific modules.
|
|
|
|
If 'prefix' is supplied, use it instead of sys.prefix or
|
|
sys.exec_prefix -- i.e., ignore 'plat_specific'.
|
|
"""
|
|
if prefix is None:
|
|
prefix = plat_specific and EXEC_PREFIX or PREFIX
|
|
|
|
if os.name == "posix":
|
|
libpython = os.path.join(prefix,
|
|
"lib", "python" + sys.version[:3])
|
|
if standard_lib:
|
|
return libpython
|
|
else:
|
|
return os.path.join(libpython, "site-packages")
|
|
|
|
elif os.name == "nt":
|
|
if standard_lib:
|
|
return os.path.join(prefix, "Lib")
|
|
else:
|
|
if sys.version < "2.2":
|
|
return prefix
|
|
else:
|
|
return os.path.join(PREFIX, "Lib", "site-packages")
|
|
|
|
elif os.name == "mac":
|
|
if plat_specific:
|
|
if standard_lib:
|
|
return os.path.join(prefix, "Lib", "lib-dynload")
|
|
else:
|
|
return os.path.join(prefix, "Lib", "site-packages")
|
|
else:
|
|
if standard_lib:
|
|
return os.path.join(prefix, "Lib")
|
|
else:
|
|
return os.path.join(prefix, "Lib", "site-packages")
|
|
|
|
elif os.name == "os2":
|
|
if standard_lib:
|
|
return os.path.join(PREFIX, "Lib")
|
|
else:
|
|
return os.path.join(PREFIX, "Lib", "site-packages")
|
|
|
|
else:
|
|
raise DistutilsPlatformError(
|
|
"I don't know where Python installs its library "
|
|
"on platform '%s'" % os.name)
|
|
|
|
|
|
def customize_compiler(compiler):
|
|
"""Do any platform-specific customization of a CCompiler instance.
|
|
|
|
Mainly needed on Unix, so we can plug in the information that
|
|
varies across Unices and is stored in Python's Makefile.
|
|
"""
|
|
if compiler.compiler_type == "unix":
|
|
(cc, opt, ccshared, ldshared, so_ext) = \
|
|
get_config_vars('CC', 'OPT', 'CCSHARED', 'LDSHARED', 'SO')
|
|
|
|
cc_cmd = cc + ' ' + opt
|
|
compiler.set_executables(
|
|
preprocessor=cc + " -E", # not always!
|
|
compiler=cc_cmd,
|
|
compiler_so=cc_cmd + ' ' + ccshared,
|
|
linker_so=ldshared,
|
|
linker_exe=cc)
|
|
|
|
compiler.shared_lib_extension = so_ext
|
|
|
|
|
|
def get_config_h_filename():
|
|
"""Return full pathname of installed pyconfig.h file."""
|
|
if python_build:
|
|
inc_dir = os.curdir
|
|
else:
|
|
inc_dir = get_python_inc(plat_specific=1)
|
|
if sys.version < '2.2':
|
|
config_h = 'config.h'
|
|
else:
|
|
# The name of the config.h file changed in 2.2
|
|
config_h = 'pyconfig.h'
|
|
return os.path.join(inc_dir, config_h)
|
|
|
|
|
|
def get_makefile_filename():
|
|
"""Return full pathname of installed Makefile from the Python build."""
|
|
if python_build:
|
|
return os.path.join(os.path.dirname(sys.executable), "Makefile")
|
|
lib_dir = get_python_lib(plat_specific=1, standard_lib=1)
|
|
return os.path.join(lib_dir, "config", "Makefile")
|
|
|
|
|
|
def parse_config_h(fp, g=None):
|
|
"""Parse a config.h-style file.
|
|
|
|
A dictionary containing name/value pairs is returned. If an
|
|
optional dictionary is passed in as the second argument, it is
|
|
used instead of a new dictionary.
|
|
"""
|
|
if g is None:
|
|
g = {}
|
|
define_rx = re.compile("#define ([A-Z][A-Z0-9_]+) (.*)\n")
|
|
undef_rx = re.compile("/[*] #undef ([A-Z][A-Z0-9_]+) [*]/\n")
|
|
#
|
|
while 1:
|
|
line = fp.readline()
|
|
if not line:
|
|
break
|
|
m = define_rx.match(line)
|
|
if m:
|
|
n, v = m.group(1, 2)
|
|
try: v = string.atoi(v)
|
|
except ValueError: pass
|
|
g[n] = v
|
|
else:
|
|
m = undef_rx.match(line)
|
|
if m:
|
|
g[m.group(1)] = 0
|
|
return g
|
|
|
|
|
|
# Regexes needed for parsing Makefile (and similar syntaxes,
|
|
# like old-style Setup files).
|
|
_variable_rx = re.compile("([a-zA-Z][a-zA-Z0-9_]+)\s*=\s*(.*)")
|
|
_findvar1_rx = re.compile(r"\$\(([A-Za-z][A-Za-z0-9_]*)\)")
|
|
_findvar2_rx = re.compile(r"\${([A-Za-z][A-Za-z0-9_]*)}")
|
|
|
|
def parse_makefile(fn, g=None):
|
|
"""Parse a Makefile-style file.
|
|
|
|
A dictionary containing name/value pairs is returned. If an
|
|
optional dictionary is passed in as the second argument, it is
|
|
used instead of a new dictionary.
|
|
"""
|
|
from distutils.text_file import TextFile
|
|
fp = TextFile(fn, strip_comments=1, skip_blanks=1, join_lines=1)
|
|
|
|
if g is None:
|
|
g = {}
|
|
done = {}
|
|
notdone = {}
|
|
|
|
while 1:
|
|
line = fp.readline()
|
|
if line is None: # eof
|
|
break
|
|
m = _variable_rx.match(line)
|
|
if m:
|
|
n, v = m.group(1, 2)
|
|
v = string.strip(v)
|
|
if "$" in v:
|
|
notdone[n] = v
|
|
else:
|
|
try: v = string.atoi(v)
|
|
except ValueError: pass
|
|
done[n] = v
|
|
|
|
# do variable interpolation here
|
|
while notdone:
|
|
for name in notdone.keys():
|
|
value = notdone[name]
|
|
m = _findvar1_rx.search(value) or _findvar2_rx.search(value)
|
|
if m:
|
|
n = m.group(1)
|
|
if done.has_key(n):
|
|
after = value[m.end():]
|
|
value = value[:m.start()] + str(done[n]) + after
|
|
if "$" in after:
|
|
notdone[name] = value
|
|
else:
|
|
try: value = string.atoi(value)
|
|
except ValueError:
|
|
done[name] = string.strip(value)
|
|
else:
|
|
done[name] = value
|
|
del notdone[name]
|
|
elif notdone.has_key(n):
|
|
# get it on a subsequent round
|
|
pass
|
|
else:
|
|
done[n] = ""
|
|
after = value[m.end():]
|
|
value = value[:m.start()] + after
|
|
if "$" in after:
|
|
notdone[name] = value
|
|
else:
|
|
try: value = string.atoi(value)
|
|
except ValueError:
|
|
done[name] = string.strip(value)
|
|
else:
|
|
done[name] = value
|
|
del notdone[name]
|
|
else:
|
|
# bogus variable reference; just drop it since we can't deal
|
|
del notdone[name]
|
|
|
|
fp.close()
|
|
|
|
# save the results in the global dictionary
|
|
g.update(done)
|
|
return g
|
|
|
|
|
|
def expand_makefile_vars(s, vars):
|
|
"""Expand Makefile-style variables -- "${foo}" or "$(foo)" -- in
|
|
'string' according to 'vars' (a dictionary mapping variable names to
|
|
values). Variables not present in 'vars' are silently expanded to the
|
|
empty string. The variable values in 'vars' should not contain further
|
|
variable expansions; if 'vars' is the output of 'parse_makefile()',
|
|
you're fine. Returns a variable-expanded version of 's'.
|
|
"""
|
|
|
|
# This algorithm does multiple expansion, so if vars['foo'] contains
|
|
# "${bar}", it will expand ${foo} to ${bar}, and then expand
|
|
# ${bar}... and so forth. This is fine as long as 'vars' comes from
|
|
# 'parse_makefile()', which takes care of such expansions eagerly,
|
|
# according to make's variable expansion semantics.
|
|
|
|
while 1:
|
|
m = _findvar1_rx.search(s) or _findvar2_rx.search(s)
|
|
if m:
|
|
(beg, end) = m.span()
|
|
s = s[0:beg] + vars.get(m.group(1)) + s[end:]
|
|
else:
|
|
break
|
|
return s
|
|
|
|
|
|
_config_vars = None
|
|
|
|
def _init_posix():
|
|
"""Initialize the module as appropriate for POSIX systems."""
|
|
g = {}
|
|
# load the installed Makefile:
|
|
try:
|
|
filename = get_makefile_filename()
|
|
parse_makefile(filename, g)
|
|
except IOError, msg:
|
|
my_msg = "invalid Python installation: unable to open %s" % filename
|
|
if hasattr(msg, "strerror"):
|
|
my_msg = my_msg + " (%s)" % msg.strerror
|
|
|
|
raise DistutilsPlatformError(my_msg)
|
|
|
|
|
|
# On AIX, there are wrong paths to the linker scripts in the Makefile
|
|
# -- these paths are relative to the Python source, but when installed
|
|
# the scripts are in another directory.
|
|
if python_build:
|
|
g['LDSHARED'] = g['BLDSHARED']
|
|
|
|
elif sys.version < '2.1':
|
|
# The following two branches are for 1.5.2 compatibility.
|
|
if sys.platform == 'aix4': # what about AIX 3.x ?
|
|
# Linker script is in the config directory, not in Modules as the
|
|
# Makefile says.
|
|
python_lib = get_python_lib(standard_lib=1)
|
|
ld_so_aix = os.path.join(python_lib, 'config', 'ld_so_aix')
|
|
python_exp = os.path.join(python_lib, 'config', 'python.exp')
|
|
|
|
g['LDSHARED'] = "%s %s -bI:%s" % (ld_so_aix, g['CC'], python_exp)
|
|
|
|
elif sys.platform == 'beos':
|
|
# Linker script is in the config directory. In the Makefile it is
|
|
# relative to the srcdir, which after installation no longer makes
|
|
# sense.
|
|
python_lib = get_python_lib(standard_lib=1)
|
|
linkerscript_name = os.path.basename(string.split(g['LDSHARED'])[0])
|
|
linkerscript = os.path.join(python_lib, 'config', linkerscript_name)
|
|
|
|
# XXX this isn't the right place to do this: adding the Python
|
|
# library to the link, if needed, should be in the "build_ext"
|
|
# command. (It's also needed for non-MS compilers on Windows, and
|
|
# it's taken care of for them by the 'build_ext.get_libraries()'
|
|
# method.)
|
|
g['LDSHARED'] = ("%s -L%s/lib -lpython%s" %
|
|
(linkerscript, PREFIX, sys.version[0:3]))
|
|
|
|
global _config_vars
|
|
_config_vars = g
|
|
|
|
|
|
def _init_nt():
|
|
"""Initialize the module as appropriate for NT"""
|
|
g = {}
|
|
# set basic install directories
|
|
g['LIBDEST'] = get_python_lib(plat_specific=0, standard_lib=1)
|
|
g['BINLIBDEST'] = get_python_lib(plat_specific=1, standard_lib=1)
|
|
|
|
# XXX hmmm.. a normal install puts include files here
|
|
g['INCLUDEPY'] = get_python_inc(plat_specific=0)
|
|
|
|
g['SO'] = '.pyd'
|
|
g['EXE'] = ".exe"
|
|
|
|
global _config_vars
|
|
_config_vars = g
|
|
|
|
|
|
def _init_mac():
|
|
"""Initialize the module as appropriate for Macintosh systems"""
|
|
g = {}
|
|
# set basic install directories
|
|
g['LIBDEST'] = get_python_lib(plat_specific=0, standard_lib=1)
|
|
g['BINLIBDEST'] = get_python_lib(plat_specific=1, standard_lib=1)
|
|
|
|
# XXX hmmm.. a normal install puts include files here
|
|
g['INCLUDEPY'] = get_python_inc(plat_specific=0)
|
|
|
|
import MacOS
|
|
if not hasattr(MacOS, 'runtimemodel'):
|
|
g['SO'] = '.ppc.slb'
|
|
else:
|
|
g['SO'] = '.%s.slb' % MacOS.runtimemodel
|
|
|
|
# XXX are these used anywhere?
|
|
g['install_lib'] = os.path.join(EXEC_PREFIX, "Lib")
|
|
g['install_platlib'] = os.path.join(EXEC_PREFIX, "Mac", "Lib")
|
|
|
|
# These are used by the extension module build
|
|
g['srcdir'] = ':'
|
|
global _config_vars
|
|
_config_vars = g
|
|
|
|
|
|
def _init_os2():
|
|
"""Initialize the module as appropriate for OS/2"""
|
|
g = {}
|
|
# set basic install directories
|
|
g['LIBDEST'] = get_python_lib(plat_specific=0, standard_lib=1)
|
|
g['BINLIBDEST'] = get_python_lib(plat_specific=1, standard_lib=1)
|
|
|
|
# XXX hmmm.. a normal install puts include files here
|
|
g['INCLUDEPY'] = get_python_inc(plat_specific=0)
|
|
|
|
g['SO'] = '.pyd'
|
|
g['EXE'] = ".exe"
|
|
|
|
global _config_vars
|
|
_config_vars = g
|
|
|
|
|
|
def get_config_vars(*args):
|
|
"""With no arguments, return a dictionary of all configuration
|
|
variables relevant for the current platform. Generally this includes
|
|
everything needed to build extensions and install both pure modules and
|
|
extensions. On Unix, this means every variable defined in Python's
|
|
installed Makefile; on Windows and Mac OS it's a much smaller set.
|
|
|
|
With arguments, return a list of values that result from looking up
|
|
each argument in the configuration variable dictionary.
|
|
"""
|
|
global _config_vars
|
|
if _config_vars is None:
|
|
func = globals().get("_init_" + os.name)
|
|
if func:
|
|
func()
|
|
else:
|
|
_config_vars = {}
|
|
|
|
# Normalized versions of prefix and exec_prefix are handy to have;
|
|
# in fact, these are the standard versions used most places in the
|
|
# Distutils.
|
|
_config_vars['prefix'] = PREFIX
|
|
_config_vars['exec_prefix'] = EXEC_PREFIX
|
|
|
|
if args:
|
|
vals = []
|
|
for name in args:
|
|
vals.append(_config_vars.get(name))
|
|
return vals
|
|
else:
|
|
return _config_vars
|
|
|
|
def get_config_var(name):
|
|
"""Return the value of a single variable using the dictionary
|
|
returned by 'get_config_vars()'. Equivalent to
|
|
get_config_vars().get(name)
|
|
"""
|
|
return get_config_vars().get(name)
|