You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
852 lines
29 KiB
Python
852 lines
29 KiB
Python
6 years ago
|
"""Append module search paths for third-party packages to sys.path.
|
||
|
|
||
|
****************************************************************
|
||
|
* This module is automatically imported during initialization. *
|
||
|
****************************************************************
|
||
|
|
||
|
In earlier versions of Python (up to 1.5a3), scripts or modules that
|
||
|
needed to use site-specific modules would place ``import site''
|
||
|
somewhere near the top of their code. Because of the automatic
|
||
|
import, this is no longer necessary (but code that does it still
|
||
|
works).
|
||
|
|
||
|
This will append site-specific paths to the module search path. On
|
||
|
Unix, it starts with sys.prefix and sys.exec_prefix (if different) and
|
||
|
appends lib/python<version>/site-packages as well as lib/site-python.
|
||
|
It also supports the Debian convention of
|
||
|
lib/python<version>/dist-packages. On other platforms (mainly Mac and
|
||
|
Windows), it uses just sys.prefix (and sys.exec_prefix, if different,
|
||
|
but this is unlikely). The resulting directories, if they exist, are
|
||
|
appended to sys.path, and also inspected for path configuration files.
|
||
|
|
||
|
FOR DEBIAN, this sys.path is augmented with directories in /usr/local.
|
||
|
Local addons go into /usr/local/lib/python<version>/site-packages
|
||
|
(resp. /usr/local/lib/site-python), Debian addons install into
|
||
|
/usr/{lib,share}/python<version>/dist-packages.
|
||
|
|
||
|
A path configuration file is a file whose name has the form
|
||
|
<package>.pth; its contents are additional directories (one per line)
|
||
|
to be added to sys.path. Non-existing directories (or
|
||
|
non-directories) are never added to sys.path; no directory is added to
|
||
|
sys.path more than once. Blank lines and lines beginning with
|
||
|
'#' are skipped. Lines starting with 'import' are executed.
|
||
|
|
||
|
For example, suppose sys.prefix and sys.exec_prefix are set to
|
||
|
/usr/local and there is a directory /usr/local/lib/python2.X/site-packages
|
||
|
with three subdirectories, foo, bar and spam, and two path
|
||
|
configuration files, foo.pth and bar.pth. Assume foo.pth contains the
|
||
|
following:
|
||
|
|
||
|
# foo package configuration
|
||
|
foo
|
||
|
bar
|
||
|
bletch
|
||
|
|
||
|
and bar.pth contains:
|
||
|
|
||
|
# bar package configuration
|
||
|
bar
|
||
|
|
||
|
Then the following directories are added to sys.path, in this order:
|
||
|
|
||
|
/usr/local/lib/python2.X/site-packages/bar
|
||
|
/usr/local/lib/python2.X/site-packages/foo
|
||
|
|
||
|
Note that bletch is omitted because it doesn't exist; bar precedes foo
|
||
|
because bar.pth comes alphabetically before foo.pth; and spam is
|
||
|
omitted because it is not mentioned in either path configuration file.
|
||
|
|
||
|
After these path manipulations, an attempt is made to import a module
|
||
|
named sitecustomize, which can perform arbitrary additional
|
||
|
site-specific customizations. If this import fails with an
|
||
|
ImportError exception, it is silently ignored.
|
||
|
|
||
|
"""
|
||
|
|
||
|
import os
|
||
|
import sys
|
||
|
|
||
|
try:
|
||
|
import __builtin__ as builtins
|
||
|
except ImportError:
|
||
|
import builtins
|
||
|
try:
|
||
|
set
|
||
|
except NameError:
|
||
|
from sets import Set as set
|
||
|
|
||
|
# Prefixes for site-packages; add additional prefixes like /usr/local here
|
||
|
PREFIXES = [sys.prefix, sys.exec_prefix]
|
||
|
# Enable per user site-packages directory
|
||
|
# set it to False to disable the feature or True to force the feature
|
||
|
ENABLE_USER_SITE = None
|
||
|
# for distutils.commands.install
|
||
|
USER_SITE = None
|
||
|
USER_BASE = None
|
||
|
|
||
|
_is_64bit = (getattr(sys, "maxsize", None) or getattr(sys, "maxint")) > 2 ** 32
|
||
|
_is_pypy = hasattr(sys, "pypy_version_info")
|
||
|
_is_jython = sys.platform[:4] == "java"
|
||
|
if _is_jython:
|
||
|
ModuleType = type(os)
|
||
|
|
||
|
|
||
|
def makepath(*paths):
|
||
|
dir = os.path.join(*paths)
|
||
|
if _is_jython and (dir == "__classpath__" or dir.startswith("__pyclasspath__")):
|
||
|
return dir, dir
|
||
|
dir = os.path.abspath(dir)
|
||
|
return dir, os.path.normcase(dir)
|
||
|
|
||
|
|
||
|
def abs__file__():
|
||
|
"""Set all module' __file__ attribute to an absolute path"""
|
||
|
for m in sys.modules.values():
|
||
|
if (_is_jython and not isinstance(m, ModuleType)) or hasattr(m, "__loader__"):
|
||
|
# only modules need the abspath in Jython. and don't mess
|
||
|
# with a PEP 302-supplied __file__
|
||
|
continue
|
||
|
f = getattr(m, "__file__", None)
|
||
|
if f is None:
|
||
|
continue
|
||
|
m.__file__ = os.path.abspath(f)
|
||
|
|
||
|
|
||
|
def removeduppaths():
|
||
|
""" Remove duplicate entries from sys.path along with making them
|
||
|
absolute"""
|
||
|
# This ensures that the initial path provided by the interpreter contains
|
||
|
# only absolute pathnames, even if we're running from the build directory.
|
||
|
L = []
|
||
|
known_paths = set()
|
||
|
for dir in sys.path:
|
||
|
# Filter out duplicate paths (on case-insensitive file systems also
|
||
|
# if they only differ in case); turn relative paths into absolute
|
||
|
# paths.
|
||
|
dir, dircase = makepath(dir)
|
||
|
if not dircase in known_paths:
|
||
|
L.append(dir)
|
||
|
known_paths.add(dircase)
|
||
|
sys.path[:] = L
|
||
|
return known_paths
|
||
|
|
||
|
|
||
|
# XXX This should not be part of site.py, since it is needed even when
|
||
|
# using the -S option for Python. See http://www.python.org/sf/586680
|
||
|
def addbuilddir():
|
||
|
"""Append ./build/lib.<platform> in case we're running in the build dir
|
||
|
(especially for Guido :-)"""
|
||
|
from distutils.util import get_platform
|
||
|
|
||
|
s = "build/lib.{}-{:.3}".format(get_platform(), sys.version)
|
||
|
if hasattr(sys, "gettotalrefcount"):
|
||
|
s += "-pydebug"
|
||
|
s = os.path.join(os.path.dirname(sys.path[-1]), s)
|
||
|
sys.path.append(s)
|
||
|
|
||
|
|
||
|
def _init_pathinfo():
|
||
|
"""Return a set containing all existing directory entries from sys.path"""
|
||
|
d = set()
|
||
|
for dir in sys.path:
|
||
|
try:
|
||
|
if os.path.isdir(dir):
|
||
|
dir, dircase = makepath(dir)
|
||
|
d.add(dircase)
|
||
|
except TypeError:
|
||
|
continue
|
||
|
return d
|
||
|
|
||
|
|
||
|
def addpackage(sitedir, name, known_paths):
|
||
|
"""Add a new path to known_paths by combining sitedir and 'name' or execute
|
||
|
sitedir if it starts with 'import'"""
|
||
|
if known_paths is None:
|
||
|
_init_pathinfo()
|
||
|
reset = 1
|
||
|
else:
|
||
|
reset = 0
|
||
|
fullname = os.path.join(sitedir, name)
|
||
|
try:
|
||
|
f = open(fullname, "r")
|
||
|
except IOError:
|
||
|
return
|
||
|
try:
|
||
|
for line in f:
|
||
|
if line.startswith("#"):
|
||
|
continue
|
||
|
if line.startswith("import"):
|
||
|
exec(line)
|
||
|
continue
|
||
|
line = line.rstrip()
|
||
|
dir, dircase = makepath(sitedir, line)
|
||
|
if not dircase in known_paths and os.path.exists(dir):
|
||
|
sys.path.append(dir)
|
||
|
known_paths.add(dircase)
|
||
|
finally:
|
||
|
f.close()
|
||
|
if reset:
|
||
|
known_paths = None
|
||
|
return known_paths
|
||
|
|
||
|
|
||
|
def addsitedir(sitedir, known_paths=None):
|
||
|
"""Add 'sitedir' argument to sys.path if missing and handle .pth files in
|
||
|
'sitedir'"""
|
||
|
if known_paths is None:
|
||
|
known_paths = _init_pathinfo()
|
||
|
reset = 1
|
||
|
else:
|
||
|
reset = 0
|
||
|
sitedir, sitedircase = makepath(sitedir)
|
||
|
if not sitedircase in known_paths:
|
||
|
sys.path.append(sitedir) # Add path component
|
||
|
try:
|
||
|
names = os.listdir(sitedir)
|
||
|
except os.error:
|
||
|
return
|
||
|
names.sort()
|
||
|
for name in names:
|
||
|
if name.endswith(os.extsep + "pth"):
|
||
|
addpackage(sitedir, name, known_paths)
|
||
|
if reset:
|
||
|
known_paths = None
|
||
|
return known_paths
|
||
|
|
||
|
|
||
|
def addsitepackages(known_paths, sys_prefix=sys.prefix, exec_prefix=sys.exec_prefix):
|
||
|
"""Add site-packages (and possibly site-python) to sys.path"""
|
||
|
prefixes = [os.path.join(sys_prefix, "local"), sys_prefix]
|
||
|
if exec_prefix != sys_prefix:
|
||
|
prefixes.append(os.path.join(exec_prefix, "local"))
|
||
|
|
||
|
for prefix in prefixes:
|
||
|
if prefix:
|
||
|
if sys.platform in ("os2emx", "riscos") or _is_jython:
|
||
|
sitedirs = [os.path.join(prefix, "Lib", "site-packages")]
|
||
|
elif _is_pypy:
|
||
|
sitedirs = [os.path.join(prefix, "site-packages")]
|
||
|
elif sys.platform == "darwin" and prefix == sys_prefix:
|
||
|
|
||
|
if prefix.startswith("/System/Library/Frameworks/"): # Apple's Python
|
||
|
|
||
|
sitedirs = [
|
||
|
os.path.join("/Library/Python", sys.version[:3], "site-packages"),
|
||
|
os.path.join(prefix, "Extras", "lib", "python"),
|
||
|
]
|
||
|
|
||
|
else: # any other Python distros on OSX work this way
|
||
|
sitedirs = [os.path.join(prefix, "lib", "python" + sys.version[:3], "site-packages")]
|
||
|
|
||
|
elif os.sep == "/":
|
||
|
sitedirs = [
|
||
|
os.path.join(prefix, "lib", "python" + sys.version[:3], "site-packages"),
|
||
|
os.path.join(prefix, "lib", "site-python"),
|
||
|
os.path.join(prefix, "python" + sys.version[:3], "lib-dynload"),
|
||
|
]
|
||
|
lib64_dir = os.path.join(prefix, "lib64", "python" + sys.version[:3], "site-packages")
|
||
|
if os.path.exists(lib64_dir) and os.path.realpath(lib64_dir) not in [
|
||
|
os.path.realpath(p) for p in sitedirs
|
||
|
]:
|
||
|
if _is_64bit:
|
||
|
sitedirs.insert(0, lib64_dir)
|
||
|
else:
|
||
|
sitedirs.append(lib64_dir)
|
||
|
try:
|
||
|
# sys.getobjects only available in --with-pydebug build
|
||
|
sys.getobjects
|
||
|
sitedirs.insert(0, os.path.join(sitedirs[0], "debug"))
|
||
|
except AttributeError:
|
||
|
pass
|
||
|
# Debian-specific dist-packages directories:
|
||
|
sitedirs.append(os.path.join(prefix, "local/lib", "python" + sys.version[:3], "dist-packages"))
|
||
|
if sys.version[0] == "2":
|
||
|
sitedirs.append(os.path.join(prefix, "lib", "python" + sys.version[:3], "dist-packages"))
|
||
|
else:
|
||
|
sitedirs.append(os.path.join(prefix, "lib", "python" + sys.version[0], "dist-packages"))
|
||
|
sitedirs.append(os.path.join(prefix, "lib", "dist-python"))
|
||
|
else:
|
||
|
sitedirs = [prefix, os.path.join(prefix, "lib", "site-packages")]
|
||
|
if sys.platform == "darwin":
|
||
|
# for framework builds *only* we add the standard Apple
|
||
|
# locations. Currently only per-user, but /Library and
|
||
|
# /Network/Library could be added too
|
||
|
if "Python.framework" in prefix:
|
||
|
home = os.environ.get("HOME")
|
||
|
if home:
|
||
|
sitedirs.append(os.path.join(home, "Library", "Python", sys.version[:3], "site-packages"))
|
||
|
for sitedir in sitedirs:
|
||
|
if os.path.isdir(sitedir):
|
||
|
addsitedir(sitedir, known_paths)
|
||
|
return None
|
||
|
|
||
|
|
||
|
def check_enableusersite():
|
||
|
"""Check if user site directory is safe for inclusion
|
||
|
|
||
|
The function tests for the command line flag (including environment var),
|
||
|
process uid/gid equal to effective uid/gid.
|
||
|
|
||
|
None: Disabled for security reasons
|
||
|
False: Disabled by user (command line option)
|
||
|
True: Safe and enabled
|
||
|
"""
|
||
|
if hasattr(sys, "flags") and getattr(sys.flags, "no_user_site", False):
|
||
|
return False
|
||
|
|
||
|
if hasattr(os, "getuid") and hasattr(os, "geteuid"):
|
||
|
# check process uid == effective uid
|
||
|
if os.geteuid() != os.getuid():
|
||
|
return None
|
||
|
if hasattr(os, "getgid") and hasattr(os, "getegid"):
|
||
|
# check process gid == effective gid
|
||
|
if os.getegid() != os.getgid():
|
||
|
return None
|
||
|
|
||
|
return True
|
||
|
|
||
|
|
||
|
def addusersitepackages(known_paths):
|
||
|
"""Add a per user site-package to sys.path
|
||
|
|
||
|
Each user has its own python directory with site-packages in the
|
||
|
home directory.
|
||
|
|
||
|
USER_BASE is the root directory for all Python versions
|
||
|
|
||
|
USER_SITE is the user specific site-packages directory
|
||
|
|
||
|
USER_SITE/.. can be used for data.
|
||
|
"""
|
||
|
global USER_BASE, USER_SITE, ENABLE_USER_SITE
|
||
|
env_base = os.environ.get("PYTHONUSERBASE", None)
|
||
|
|
||
|
def joinuser(*args):
|
||
|
return os.path.expanduser(os.path.join(*args))
|
||
|
|
||
|
# if sys.platform in ('os2emx', 'riscos'):
|
||
|
# # Don't know what to put here
|
||
|
# USER_BASE = ''
|
||
|
# USER_SITE = ''
|
||
|
if os.name == "nt":
|
||
|
base = os.environ.get("APPDATA") or "~"
|
||
|
if env_base:
|
||
|
USER_BASE = env_base
|
||
|
else:
|
||
|
USER_BASE = joinuser(base, "Python")
|
||
|
USER_SITE = os.path.join(USER_BASE, "Python" + sys.version[0] + sys.version[2], "site-packages")
|
||
|
else:
|
||
|
if env_base:
|
||
|
USER_BASE = env_base
|
||
|
else:
|
||
|
USER_BASE = joinuser("~", ".local")
|
||
|
USER_SITE = os.path.join(USER_BASE, "lib", "python" + sys.version[:3], "site-packages")
|
||
|
|
||
|
if ENABLE_USER_SITE and os.path.isdir(USER_SITE):
|
||
|
addsitedir(USER_SITE, known_paths)
|
||
|
if ENABLE_USER_SITE:
|
||
|
for dist_libdir in ("lib", "local/lib"):
|
||
|
user_site = os.path.join(USER_BASE, dist_libdir, "python" + sys.version[:3], "dist-packages")
|
||
|
if os.path.isdir(user_site):
|
||
|
addsitedir(user_site, known_paths)
|
||
|
return known_paths
|
||
|
|
||
|
|
||
|
def setBEGINLIBPATH():
|
||
|
"""The OS/2 EMX port has optional extension modules that do double duty
|
||
|
as DLLs (and must use the .DLL file extension) for other extensions.
|
||
|
The library search path needs to be amended so these will be found
|
||
|
during module import. Use BEGINLIBPATH so that these are at the start
|
||
|
of the library search path.
|
||
|
|
||
|
"""
|
||
|
dllpath = os.path.join(sys.prefix, "Lib", "lib-dynload")
|
||
|
libpath = os.environ["BEGINLIBPATH"].split(";")
|
||
|
if libpath[-1]:
|
||
|
libpath.append(dllpath)
|
||
|
else:
|
||
|
libpath[-1] = dllpath
|
||
|
os.environ["BEGINLIBPATH"] = ";".join(libpath)
|
||
|
|
||
|
|
||
|
def setquit():
|
||
|
"""Define new built-ins 'quit' and 'exit'.
|
||
|
These are simply strings that display a hint on how to exit.
|
||
|
|
||
|
"""
|
||
|
if os.sep == ":":
|
||
|
eof = "Cmd-Q"
|
||
|
elif os.sep == "\\":
|
||
|
eof = "Ctrl-Z plus Return"
|
||
|
else:
|
||
|
eof = "Ctrl-D (i.e. EOF)"
|
||
|
|
||
|
class Quitter(object):
|
||
|
def __init__(self, name):
|
||
|
self.name = name
|
||
|
|
||
|
def __repr__(self):
|
||
|
return "Use {}() or {} to exit".format(self.name, eof)
|
||
|
|
||
|
def __call__(self, code=None):
|
||
|
# Shells like IDLE catch the SystemExit, but listen when their
|
||
|
# stdin wrapper is closed.
|
||
|
try:
|
||
|
sys.stdin.close()
|
||
|
except:
|
||
|
pass
|
||
|
raise SystemExit(code)
|
||
|
|
||
|
builtins.quit = Quitter("quit")
|
||
|
builtins.exit = Quitter("exit")
|
||
|
|
||
|
|
||
|
class _Printer(object):
|
||
|
"""interactive prompt objects for printing the license text, a list of
|
||
|
contributors and the copyright notice."""
|
||
|
|
||
|
MAXLINES = 23
|
||
|
|
||
|
def __init__(self, name, data, files=(), dirs=()):
|
||
|
self.__name = name
|
||
|
self.__data = data
|
||
|
self.__files = files
|
||
|
self.__dirs = dirs
|
||
|
self.__lines = None
|
||
|
|
||
|
def __setup(self):
|
||
|
if self.__lines:
|
||
|
return
|
||
|
data = None
|
||
|
for dir in self.__dirs:
|
||
|
for filename in self.__files:
|
||
|
filename = os.path.join(dir, filename)
|
||
|
try:
|
||
|
fp = open(filename, "r")
|
||
|
data = fp.read()
|
||
|
fp.close()
|
||
|
break
|
||
|
except IOError:
|
||
|
pass
|
||
|
if data:
|
||
|
break
|
||
|
if not data:
|
||
|
data = self.__data
|
||
|
self.__lines = data.split("\n")
|
||
|
self.__linecnt = len(self.__lines)
|
||
|
|
||
|
def __repr__(self):
|
||
|
self.__setup()
|
||
|
if len(self.__lines) <= self.MAXLINES:
|
||
|
return "\n".join(self.__lines)
|
||
|
else:
|
||
|
return "Type %s() to see the full %s text" % ((self.__name,) * 2)
|
||
|
|
||
|
def __call__(self):
|
||
|
self.__setup()
|
||
|
prompt = "Hit Return for more, or q (and Return) to quit: "
|
||
|
lineno = 0
|
||
|
while 1:
|
||
|
try:
|
||
|
for i in range(lineno, lineno + self.MAXLINES):
|
||
|
print(self.__lines[i])
|
||
|
except IndexError:
|
||
|
break
|
||
|
else:
|
||
|
lineno += self.MAXLINES
|
||
|
key = None
|
||
|
while key is None:
|
||
|
try:
|
||
|
key = raw_input(prompt)
|
||
|
except NameError:
|
||
|
key = input(prompt)
|
||
|
if key not in ("", "q"):
|
||
|
key = None
|
||
|
if key == "q":
|
||
|
break
|
||
|
|
||
|
|
||
|
def setcopyright():
|
||
|
"""Set 'copyright' and 'credits' in __builtin__"""
|
||
|
builtins.copyright = _Printer("copyright", sys.copyright)
|
||
|
if _is_jython:
|
||
|
builtins.credits = _Printer("credits", "Jython is maintained by the Jython developers (www.jython.org).")
|
||
|
elif _is_pypy:
|
||
|
builtins.credits = _Printer("credits", "PyPy is maintained by the PyPy developers: http://pypy.org/")
|
||
|
else:
|
||
|
builtins.credits = _Printer(
|
||
|
"credits",
|
||
|
"""\
|
||
|
Thanks to CWI, CNRI, BeOpen.com, Zope Corporation and a cast of thousands
|
||
|
for supporting Python development. See www.python.org for more information.""",
|
||
|
)
|
||
|
here = os.path.dirname(os.__file__)
|
||
|
builtins.license = _Printer(
|
||
|
"license",
|
||
|
"See https://www.python.org/psf/license/",
|
||
|
["LICENSE.txt", "LICENSE"],
|
||
|
[os.path.join(here, os.pardir), here, os.curdir],
|
||
|
)
|
||
|
|
||
|
|
||
|
class _Helper(object):
|
||
|
"""Define the built-in 'help'.
|
||
|
This is a wrapper around pydoc.help (with a twist).
|
||
|
|
||
|
"""
|
||
|
|
||
|
def __repr__(self):
|
||
|
return "Type help() for interactive help, " "or help(object) for help about object."
|
||
|
|
||
|
def __call__(self, *args, **kwds):
|
||
|
import pydoc
|
||
|
|
||
|
return pydoc.help(*args, **kwds)
|
||
|
|
||
|
|
||
|
def sethelper():
|
||
|
builtins.help = _Helper()
|
||
|
|
||
|
|
||
|
def aliasmbcs():
|
||
|
"""On Windows, some default encodings are not provided by Python,
|
||
|
while they are always available as "mbcs" in each locale. Make
|
||
|
them usable by aliasing to "mbcs" in such a case."""
|
||
|
if sys.platform == "win32":
|
||
|
import locale, codecs
|
||
|
|
||
|
enc = locale.getdefaultlocale()[1]
|
||
|
if enc.startswith("cp"): # "cp***" ?
|
||
|
try:
|
||
|
codecs.lookup(enc)
|
||
|
except LookupError:
|
||
|
import encodings
|
||
|
|
||
|
encodings._cache[enc] = encodings._unknown
|
||
|
encodings.aliases.aliases[enc] = "mbcs"
|
||
|
|
||
|
|
||
|
def setencoding():
|
||
|
"""Set the string encoding used by the Unicode implementation. The
|
||
|
default is 'ascii', but if you're willing to experiment, you can
|
||
|
change this."""
|
||
|
encoding = "ascii" # Default value set by _PyUnicode_Init()
|
||
|
if 0:
|
||
|
# Enable to support locale aware default string encodings.
|
||
|
import locale
|
||
|
|
||
|
loc = locale.getdefaultlocale()
|
||
|
if loc[1]:
|
||
|
encoding = loc[1]
|
||
|
if 0:
|
||
|
# Enable to switch off string to Unicode coercion and implicit
|
||
|
# Unicode to string conversion.
|
||
|
encoding = "undefined"
|
||
|
if encoding != "ascii":
|
||
|
# On Non-Unicode builds this will raise an AttributeError...
|
||
|
sys.setdefaultencoding(encoding) # Needs Python Unicode build !
|
||
|
|
||
|
|
||
|
def execsitecustomize():
|
||
|
"""Run custom site specific code, if available."""
|
||
|
try:
|
||
|
import sitecustomize
|
||
|
except ImportError:
|
||
|
pass
|
||
|
|
||
|
|
||
|
def virtual_install_main_packages():
|
||
|
f = open(os.path.join(os.path.dirname(__file__), "orig-prefix.txt"))
|
||
|
sys.real_prefix = f.read().strip()
|
||
|
f.close()
|
||
|
pos = 2
|
||
|
hardcoded_relative_dirs = []
|
||
|
if sys.path[0] == "":
|
||
|
pos += 1
|
||
|
if _is_jython:
|
||
|
paths = [os.path.join(sys.real_prefix, "Lib")]
|
||
|
elif _is_pypy:
|
||
|
if sys.version_info > (3, 2):
|
||
|
cpyver = "%d" % sys.version_info[0]
|
||
|
elif sys.pypy_version_info >= (1, 5):
|
||
|
cpyver = "%d.%d" % sys.version_info[:2]
|
||
|
else:
|
||
|
cpyver = "%d.%d.%d" % sys.version_info[:3]
|
||
|
paths = [os.path.join(sys.real_prefix, "lib_pypy"), os.path.join(sys.real_prefix, "lib-python", cpyver)]
|
||
|
if sys.pypy_version_info < (1, 9):
|
||
|
paths.insert(1, os.path.join(sys.real_prefix, "lib-python", "modified-%s" % cpyver))
|
||
|
hardcoded_relative_dirs = paths[:] # for the special 'darwin' case below
|
||
|
#
|
||
|
# This is hardcoded in the Python executable, but relative to sys.prefix:
|
||
|
for path in paths[:]:
|
||
|
plat_path = os.path.join(path, "plat-%s" % sys.platform)
|
||
|
if os.path.exists(plat_path):
|
||
|
paths.append(plat_path)
|
||
|
elif sys.platform == "win32":
|
||
|
paths = [os.path.join(sys.real_prefix, "Lib"), os.path.join(sys.real_prefix, "DLLs")]
|
||
|
else:
|
||
|
paths = [os.path.join(sys.real_prefix, "lib", "python" + sys.version[:3])]
|
||
|
hardcoded_relative_dirs = paths[:] # for the special 'darwin' case below
|
||
|
lib64_path = os.path.join(sys.real_prefix, "lib64", "python" + sys.version[:3])
|
||
|
if os.path.exists(lib64_path):
|
||
|
if _is_64bit:
|
||
|
paths.insert(0, lib64_path)
|
||
|
else:
|
||
|
paths.append(lib64_path)
|
||
|
# This is hardcoded in the Python executable, but relative to
|
||
|
# sys.prefix. Debian change: we need to add the multiarch triplet
|
||
|
# here, which is where the real stuff lives. As per PEP 421, in
|
||
|
# Python 3.3+, this lives in sys.implementation, while in Python 2.7
|
||
|
# it lives in sys.
|
||
|
try:
|
||
|
arch = getattr(sys, "implementation", sys)._multiarch
|
||
|
except AttributeError:
|
||
|
# This is a non-multiarch aware Python. Fallback to the old way.
|
||
|
arch = sys.platform
|
||
|
plat_path = os.path.join(sys.real_prefix, "lib", "python" + sys.version[:3], "plat-%s" % arch)
|
||
|
if os.path.exists(plat_path):
|
||
|
paths.append(plat_path)
|
||
|
# This is hardcoded in the Python executable, but
|
||
|
# relative to sys.prefix, so we have to fix up:
|
||
|
for path in list(paths):
|
||
|
tk_dir = os.path.join(path, "lib-tk")
|
||
|
if os.path.exists(tk_dir):
|
||
|
paths.append(tk_dir)
|
||
|
|
||
|
# These are hardcoded in the Apple's Python executable,
|
||
|
# but relative to sys.prefix, so we have to fix them up:
|
||
|
if sys.platform == "darwin":
|
||
|
hardcoded_paths = [
|
||
|
os.path.join(relative_dir, module)
|
||
|
for relative_dir in hardcoded_relative_dirs
|
||
|
for module in ("plat-darwin", "plat-mac", "plat-mac/lib-scriptpackages")
|
||
|
]
|
||
|
|
||
|
for path in hardcoded_paths:
|
||
|
if os.path.exists(path):
|
||
|
paths.append(path)
|
||
|
|
||
|
sys.path.extend(paths)
|
||
|
|
||
|
|
||
|
def force_global_eggs_after_local_site_packages():
|
||
|
"""
|
||
|
Force easy_installed eggs in the global environment to get placed
|
||
|
in sys.path after all packages inside the virtualenv. This
|
||
|
maintains the "least surprise" result that packages in the
|
||
|
virtualenv always mask global packages, never the other way
|
||
|
around.
|
||
|
|
||
|
"""
|
||
|
egginsert = getattr(sys, "__egginsert", 0)
|
||
|
for i, path in enumerate(sys.path):
|
||
|
if i > egginsert and path.startswith(sys.prefix):
|
||
|
egginsert = i
|
||
|
sys.__egginsert = egginsert + 1
|
||
|
|
||
|
|
||
|
def virtual_addsitepackages(known_paths):
|
||
|
force_global_eggs_after_local_site_packages()
|
||
|
return addsitepackages(known_paths, sys_prefix=sys.real_prefix)
|
||
|
|
||
|
|
||
|
def fixclasspath():
|
||
|
"""Adjust the special classpath sys.path entries for Jython. These
|
||
|
entries should follow the base virtualenv lib directories.
|
||
|
"""
|
||
|
paths = []
|
||
|
classpaths = []
|
||
|
for path in sys.path:
|
||
|
if path == "__classpath__" or path.startswith("__pyclasspath__"):
|
||
|
classpaths.append(path)
|
||
|
else:
|
||
|
paths.append(path)
|
||
|
sys.path = paths
|
||
|
sys.path.extend(classpaths)
|
||
|
|
||
|
|
||
|
def execusercustomize():
|
||
|
"""Run custom user specific code, if available."""
|
||
|
try:
|
||
|
import usercustomize
|
||
|
except ImportError:
|
||
|
pass
|
||
|
|
||
|
|
||
|
def enablerlcompleter():
|
||
|
"""Enable default readline configuration on interactive prompts, by
|
||
|
registering a sys.__interactivehook__.
|
||
|
If the readline module can be imported, the hook will set the Tab key
|
||
|
as completion key and register ~/.python_history as history file.
|
||
|
This can be overridden in the sitecustomize or usercustomize module,
|
||
|
or in a PYTHONSTARTUP file.
|
||
|
"""
|
||
|
|
||
|
def register_readline():
|
||
|
import atexit
|
||
|
|
||
|
try:
|
||
|
import readline
|
||
|
import rlcompleter
|
||
|
except ImportError:
|
||
|
return
|
||
|
|
||
|
# Reading the initialization (config) file may not be enough to set a
|
||
|
# completion key, so we set one first and then read the file.
|
||
|
readline_doc = getattr(readline, "__doc__", "")
|
||
|
if readline_doc is not None and "libedit" in readline_doc:
|
||
|
readline.parse_and_bind("bind ^I rl_complete")
|
||
|
else:
|
||
|
readline.parse_and_bind("tab: complete")
|
||
|
|
||
|
try:
|
||
|
readline.read_init_file()
|
||
|
except OSError:
|
||
|
# An OSError here could have many causes, but the most likely one
|
||
|
# is that there's no .inputrc file (or .editrc file in the case of
|
||
|
# Mac OS X + libedit) in the expected location. In that case, we
|
||
|
# want to ignore the exception.
|
||
|
pass
|
||
|
|
||
|
if readline.get_current_history_length() == 0:
|
||
|
# If no history was loaded, default to .python_history.
|
||
|
# The guard is necessary to avoid doubling history size at
|
||
|
# each interpreter exit when readline was already configured
|
||
|
# through a PYTHONSTARTUP hook, see:
|
||
|
# http://bugs.python.org/issue5845#msg198636
|
||
|
history = os.path.join(os.path.expanduser("~"), ".python_history")
|
||
|
try:
|
||
|
readline.read_history_file(history)
|
||
|
except OSError:
|
||
|
pass
|
||
|
|
||
|
def write_history():
|
||
|
try:
|
||
|
readline.write_history_file(history)
|
||
|
except (FileNotFoundError, PermissionError):
|
||
|
# home directory does not exist or is not writable
|
||
|
# https://bugs.python.org/issue19891
|
||
|
pass
|
||
|
|
||
|
atexit.register(write_history)
|
||
|
|
||
|
sys.__interactivehook__ = register_readline
|
||
|
|
||
|
|
||
|
if _is_pypy:
|
||
|
|
||
|
def import_builtin_stuff():
|
||
|
"""PyPy specific: some built-in modules should be pre-imported because
|
||
|
some programs expect them to be in sys.modules on startup. This is ported
|
||
|
from PyPy's site.py.
|
||
|
"""
|
||
|
import encodings
|
||
|
|
||
|
if "exceptions" in sys.builtin_module_names:
|
||
|
import exceptions
|
||
|
|
||
|
if "zipimport" in sys.builtin_module_names:
|
||
|
import zipimport
|
||
|
|
||
|
|
||
|
def main():
|
||
|
global ENABLE_USER_SITE
|
||
|
virtual_install_main_packages()
|
||
|
if _is_pypy:
|
||
|
import_builtin_stuff()
|
||
|
abs__file__()
|
||
|
paths_in_sys = removeduppaths()
|
||
|
if os.name == "posix" and sys.path and os.path.basename(sys.path[-1]) == "Modules":
|
||
|
addbuilddir()
|
||
|
if _is_jython:
|
||
|
fixclasspath()
|
||
|
GLOBAL_SITE_PACKAGES = not os.path.exists(os.path.join(os.path.dirname(__file__), "no-global-site-packages.txt"))
|
||
|
if not GLOBAL_SITE_PACKAGES:
|
||
|
ENABLE_USER_SITE = False
|
||
|
if ENABLE_USER_SITE is None:
|
||
|
ENABLE_USER_SITE = check_enableusersite()
|
||
|
paths_in_sys = addsitepackages(paths_in_sys)
|
||
|
paths_in_sys = addusersitepackages(paths_in_sys)
|
||
|
if GLOBAL_SITE_PACKAGES:
|
||
|
paths_in_sys = virtual_addsitepackages(paths_in_sys)
|
||
|
if sys.platform == "os2emx":
|
||
|
setBEGINLIBPATH()
|
||
|
setquit()
|
||
|
setcopyright()
|
||
|
sethelper()
|
||
|
if sys.version_info[0] == 3:
|
||
|
enablerlcompleter()
|
||
|
aliasmbcs()
|
||
|
setencoding()
|
||
|
execsitecustomize()
|
||
|
if ENABLE_USER_SITE:
|
||
|
execusercustomize()
|
||
|
# Remove sys.setdefaultencoding() so that users cannot change the
|
||
|
# encoding after initialization. The test for presence is needed when
|
||
|
# this module is run as a script, because this code is executed twice.
|
||
|
if hasattr(sys, "setdefaultencoding"):
|
||
|
del sys.setdefaultencoding
|
||
|
|
||
|
|
||
|
main()
|
||
|
|
||
|
|
||
|
def _script():
|
||
|
help = """\
|
||
|
%s [--user-base] [--user-site]
|
||
|
|
||
|
Without arguments print some useful information
|
||
|
With arguments print the value of USER_BASE and/or USER_SITE separated
|
||
|
by '%s'.
|
||
|
|
||
|
Exit codes with --user-base or --user-site:
|
||
|
0 - user site directory is enabled
|
||
|
1 - user site directory is disabled by user
|
||
|
2 - uses site directory is disabled by super user
|
||
|
or for security reasons
|
||
|
>2 - unknown error
|
||
|
"""
|
||
|
args = sys.argv[1:]
|
||
|
if not args:
|
||
|
print("sys.path = [")
|
||
|
for dir in sys.path:
|
||
|
print(" {!r},".format(dir))
|
||
|
print("]")
|
||
|
|
||
|
def exists(path):
|
||
|
if os.path.isdir(path):
|
||
|
return "exists"
|
||
|
else:
|
||
|
return "doesn't exist"
|
||
|
|
||
|
print("USER_BASE: {!r} ({})".format(USER_BASE, exists(USER_BASE)))
|
||
|
print("USER_SITE: {!r} ({})".format(USER_SITE, exists(USER_SITE)))
|
||
|
print("ENABLE_USER_SITE: %r" % ENABLE_USER_SITE)
|
||
|
sys.exit(0)
|
||
|
|
||
|
buffer = []
|
||
|
if "--user-base" in args:
|
||
|
buffer.append(USER_BASE)
|
||
|
if "--user-site" in args:
|
||
|
buffer.append(USER_SITE)
|
||
|
|
||
|
if buffer:
|
||
|
print(os.pathsep.join(buffer))
|
||
|
if ENABLE_USER_SITE:
|
||
|
sys.exit(0)
|
||
|
elif ENABLE_USER_SITE is False:
|
||
|
sys.exit(1)
|
||
|
elif ENABLE_USER_SITE is None:
|
||
|
sys.exit(2)
|
||
|
else:
|
||
|
sys.exit(3)
|
||
|
else:
|
||
|
import textwrap
|
||
|
|
||
|
print(textwrap.dedent(help % (sys.argv[0], os.pathsep)))
|
||
|
sys.exit(10)
|
||
|
|
||
|
|
||
|
if __name__ == "__main__":
|
||
|
_script()
|