mirror of
https://github.com/kovidgoyal/calibre.git
synced 2025-10-30 10:12:25 -04:00
107 lines
2.9 KiB
Python
107 lines
2.9 KiB
Python
"""
|
|
Append module search paths for third-party packages to sys.path.
|
|
|
|
This is stripped down and customized for use in py2app applications
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
|
|
def makepath(*paths):
|
|
dir = os.path.abspath(os.path.join(*paths))
|
|
return dir, os.path.normcase(dir)
|
|
|
|
for m in sys.modules.values():
|
|
f = getattr(m, '__file__', None)
|
|
if isinstance(f, basestring) and os.path.exists(f):
|
|
m.__file__ = os.path.abspath(m.__file__)
|
|
del m
|
|
|
|
# This ensures that the initial path provided by the interpreter contains
|
|
# only absolute pathnames, even if we're running from the build directory.
|
|
L = []
|
|
_dirs_in_sys_path = {}
|
|
dir = dircase = None # sys.path may be empty at this point
|
|
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 _dirs_in_sys_path:
|
|
L.append(dir)
|
|
_dirs_in_sys_path[dircase] = 1
|
|
sys.path[:] = L
|
|
del dir, dircase, L
|
|
_dirs_in_sys_path = None
|
|
|
|
def _init_pathinfo():
|
|
global _dirs_in_sys_path
|
|
_dirs_in_sys_path = d = {}
|
|
for dir in sys.path:
|
|
if dir and not os.path.isdir(dir):
|
|
continue
|
|
dir, dircase = makepath(dir)
|
|
d[dircase] = 1
|
|
|
|
def addsitedir(sitedir):
|
|
global _dirs_in_sys_path
|
|
if _dirs_in_sys_path is None:
|
|
_init_pathinfo()
|
|
reset = 1
|
|
else:
|
|
reset = 0
|
|
sitedir, sitedircase = makepath(sitedir)
|
|
if not sitedircase in _dirs_in_sys_path:
|
|
sys.path.append(sitedir) # Add path component
|
|
try:
|
|
names = os.listdir(sitedir)
|
|
except os.error:
|
|
return
|
|
names.sort()
|
|
for name in names:
|
|
if name[-4:] == os.extsep + "pth":
|
|
addpackage(sitedir, name)
|
|
if reset:
|
|
_dirs_in_sys_path = None
|
|
|
|
def addpackage(sitedir, name):
|
|
global _dirs_in_sys_path
|
|
if _dirs_in_sys_path is None:
|
|
_init_pathinfo()
|
|
reset = 1
|
|
else:
|
|
reset = 0
|
|
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.startswith("import"):
|
|
exec dir
|
|
continue
|
|
if dir[-1] == '\n':
|
|
dir = dir[:-1]
|
|
dir, dircase = makepath(sitedir, dir)
|
|
if not dircase in _dirs_in_sys_path and os.path.exists(dir):
|
|
sys.path.append(dir)
|
|
_dirs_in_sys_path[dircase] = 1
|
|
if reset:
|
|
_dirs_in_sys_path = None
|
|
|
|
|
|
sys.setdefaultencoding('utf-8')
|
|
|
|
#
|
|
# 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
|