mirror of
https://github.com/python/cpython.git
synced 2024-11-24 02:15:30 +08:00
38cb9f1f17
__file__ attributes of already-imported modules to be absolute. This helps
robustify the interpreter against os.chdir() calls from the application.
Only remove setdefaultencoding() from sys if it exists; if this module is
run as a script (since there is a _test() function that gets run), it broke
because the script attempts to remove it again after the import of site
has already done so. This allows the module to be run as a script again.
makepath(): New function, standardizes all pathname normalization in one
place.
240 lines
7.1 KiB
Python
240 lines
7.1 KiB
Python
"""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 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.
|
|
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.
|
|
|
|
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
|
|
\code{#} are skipped.
|
|
|
|
For example, suppose sys.prefix and sys.exec_prefix are set to
|
|
/usr/local and there is a directory /usr/local/lib/python1.5/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/python1.5/site-packages/bar
|
|
/usr/local/lib/python1.5/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 sys, os
|
|
|
|
def makepath(*paths):
|
|
dir = os.path.join(*paths)
|
|
return os.path.normcase(os.path.abspath(dir))
|
|
|
|
L = sys.modules.values()
|
|
for m in L:
|
|
if hasattr(m, "__file__"):
|
|
m.__file__ = makepath(m.__file__)
|
|
del m, L
|
|
|
|
# This ensures that the initial path provided by the interpreter contains
|
|
# only absolute pathnames, even if we're running from the build directory.
|
|
L = []
|
|
for dir in sys.path:
|
|
dir = makepath(dir)
|
|
if dir not in L:
|
|
L.append(dir)
|
|
sys.path[:] = L
|
|
del dir, L
|
|
|
|
def addsitedir(sitedir):
|
|
sitedir = makepath(sitedir)
|
|
if sitedir not in sys.path:
|
|
sys.path.append(sitedir) # Add path component
|
|
try:
|
|
names = os.listdir(sitedir)
|
|
except os.error:
|
|
return
|
|
names = map(os.path.normcase, names)
|
|
names.sort()
|
|
for name in names:
|
|
if name[-4:] == ".pth":
|
|
addpackage(sitedir, name)
|
|
|
|
def addpackage(sitedir, name):
|
|
fullname = os.path.join(sitedir, name)
|
|
try:
|
|
f = open(fullname)
|
|
except IOError:
|
|
return
|
|
while 1:
|
|
dir = f.readline()
|
|
if not dir:
|
|
break
|
|
if dir[0] == '#':
|
|
continue
|
|
if dir[-1] == '\n':
|
|
dir = dir[:-1]
|
|
dir = makepath(sitedir, dir)
|
|
if dir not in sys.path and os.path.exists(dir):
|
|
sys.path.append(dir)
|
|
|
|
prefixes = [sys.prefix]
|
|
if sys.exec_prefix != sys.prefix:
|
|
prefixes.append(sys.exec_prefix)
|
|
for prefix in prefixes:
|
|
if prefix:
|
|
if os.sep == '/':
|
|
sitedirs = [makepath(prefix,
|
|
"lib",
|
|
"python" + sys.version[:3],
|
|
"site-packages"),
|
|
makepath(prefix, "lib", "site-python")]
|
|
else:
|
|
sitedirs = [prefix]
|
|
for sitedir in sitedirs:
|
|
if os.path.isdir(sitedir):
|
|
addsitedir(sitedir)
|
|
|
|
# Define new built-ins 'quit' and 'exit'.
|
|
# These are simply strings that display a hint on how to exit.
|
|
if os.sep == ':':
|
|
exit = 'Use Cmd-Q to quit.'
|
|
elif os.sep == '\\':
|
|
exit = 'Use Ctrl-Z plus Return to exit.'
|
|
else:
|
|
exit = 'Use Ctrl-D (i.e. EOF) to exit.'
|
|
import __builtin__
|
|
__builtin__.quit = __builtin__.exit = exit
|
|
del exit
|
|
|
|
# interactive prompt objects for printing the license text, a list of
|
|
# contributors and the copyright notice.
|
|
class _Printer:
|
|
MAXLINES = 23
|
|
|
|
def __init__(self, s):
|
|
self.__lines = s.split('\n')
|
|
self.__linecnt = len(self.__lines)
|
|
|
|
def __repr__(self):
|
|
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:
|
|
key = raw_input(prompt)
|
|
if key not in ('', 'q'):
|
|
key = None
|
|
if key == 'q':
|
|
break
|
|
return ''
|
|
|
|
__builtin__.copyright = _Printer(sys.copyright)
|
|
__builtin__.credits = _Printer(
|
|
'''Python development is led by BeOpen PythonLabs (www.pythonlabs.com).''')
|
|
|
|
def make_license(filename):
|
|
try:
|
|
return _Printer(open(filename).read())
|
|
except IOError:
|
|
return None
|
|
|
|
here = os.path.dirname(os.__file__)
|
|
for dir in here, os.path.join(here, os.pardir), os.curdir:
|
|
for file in "LICENSE.txt", "LICENSE":
|
|
lic = make_license(os.path.join(dir, file))
|
|
if lic:
|
|
break
|
|
if lic:
|
|
__builtin__.license = lic
|
|
break
|
|
else:
|
|
__builtin__.license = _Printer('See http://hdl.handle.net/1895.22/1012')
|
|
|
|
|
|
# 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":
|
|
sys.setdefaultencoding(encoding)
|
|
|
|
#
|
|
# Run custom site specific code, if available.
|
|
#
|
|
try:
|
|
import sitecustomize
|
|
except ImportError:
|
|
pass
|
|
|
|
#
|
|
# 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, becuase this code is executed twice.
|
|
#
|
|
if hasattr(sys, "setdefaultencoding"):
|
|
del sys.setdefaultencoding
|
|
|
|
def _test():
|
|
print "sys.path = ["
|
|
for dir in sys.path:
|
|
print " %s," % `dir`
|
|
print "]"
|
|
|
|
if __name__ == '__main__':
|
|
_test()
|