123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120 |
- #
- # Initialization for the win32com package
- #
- import win32api, sys, os
- import pythoncom
- # flag if we are in a "frozen" build.
- _frozen = getattr(sys, "frozen", 1==0)
- # pythoncom dumbly defaults this to zero - we believe sys.frozen over it.
- if _frozen and not getattr(pythoncom, "frozen", 0):
- pythoncom.frozen = sys.frozen
- # Add support for an external "COM Extensions" path.
- # Concept is that you can register a seperate path to be used for
- # COM extensions, outside of the win32com directory. These modules, however,
- # look identical to win32com built-in modules.
- # This is the technique that we use for the "standard" COM extensions.
- # eg "win32com.mapi" or "win32com.axscript" both work, even though they do not
- # live under the main win32com directory.
- __gen_path__ = ''
- __build_path__ = None
- ### TODO - Load _all_ \\Extensions subkeys - for now, we only read the default
- ### Modules will work if loaded into "win32comext" path.
- def SetupEnvironment():
- HKEY_LOCAL_MACHINE = -2147483646 # Avoid pulling in win32con for just these...
- KEY_QUERY_VALUE = 0x1
- # Open the root key once, as this is quite slow on NT.
- try:
- keyName = "SOFTWARE\\Python\\PythonCore\\%s\\PythonPath\\win32com" % sys.winver
- key = win32api.RegOpenKey(HKEY_LOCAL_MACHINE , keyName, 0, KEY_QUERY_VALUE)
- except (win32api.error, AttributeError):
- key = None
-
- try:
- found = 0
- if key is not None:
- try:
- __path__.append( win32api.RegQueryValue(key, "Extensions" ))
- found = 1
- except win32api.error:
- # Nothing registered
- pass
- if not found:
- try:
- __path__.append( win32api.GetFullPathName( __path__[0] + "\\..\\win32comext") )
- except win32api.error:
- # Give up in disgust!
- pass
-
- # For the sake of developers, we also look up a "BuildPath" key
- # If extension modules add support, we can load their .pyd's from a completely
- # different directory (see the comments below)
- try:
- if key is not None:
- global __build_path__
- __build_path__ = win32api.RegQueryValue(key, "BuildPath")
- __path__.append(__build_path__)
- except win32api.error:
- # __build_path__ neednt be defined.
- pass
- global __gen_path__
- if key is not None:
- try:
- __gen_path__ = win32api.RegQueryValue(key, "GenPath")
- except win32api.error:
- pass
- finally:
- if key is not None:
- key.Close()
- # A Helper for developers. A sub-package's __init__ can call this help function,
- # which allows the .pyd files for the extension to live in a special "Build" directory
- # (which the win32com developers do!)
- def __PackageSupportBuildPath__(package_path):
- # See if we have a special directory for the binaries (for developers)
- if not _frozen and __build_path__:
- package_path.append(__build_path__)
- if not _frozen:
- SetupEnvironment()
- # If we don't have a special __gen_path__, see if we have a gen_py as a
- # normal module and use that (ie, "win32com.gen_py" may already exist as
- # a package.
- if not __gen_path__:
- try:
- import win32com.gen_py
- # hrmph - 3.3 throws: TypeError: '_NamespacePath' object does not support indexing
- # attempting to get __path__[0] - but I can't quickly repro this stand-alone.
- # Work around it by using an iterator.
- __gen_path__ = next(iter(sys.modules["win32com.gen_py"].__path__))
- except ImportError:
- # If a win32com\gen_py directory already exists, then we use it
- # (gencache doesn't insist it have an __init__, but our __import__
- # above does!
- __gen_path__ = os.path.abspath(os.path.join(__path__[0], "gen_py"))
- if not os.path.isdir(__gen_path__):
- # We used to dynamically create a directory under win32com -
- # but this sucks. If the dir doesn't already exist, we we
- # create a version specific directory under the user temp
- # directory.
- __gen_path__ = os.path.join(
- win32api.GetTempPath(), "gen_py",
- "%d.%d" % (sys.version_info[0], sys.version_info[1]))
- # we must have a __gen_path__, but may not have a gen_py module -
- # set that up.
- if "win32com.gen_py" not in sys.modules:
- # Create a "win32com.gen_py", but with a custom __path__
- import types
- gen_py = types.ModuleType("win32com.gen_py")
- gen_py.__path__ = [ __gen_path__ ]
- sys.modules[gen_py.__name__] = gen_py
- del types
- gen_py = sys.modules["win32com.gen_py"]
- # get rid of these for module users
- del os, sys, win32api, pythoncom
|