blob: 2832d54e27a335e5e7f3b864ebcbe06f7562ed71 [file] [log] [blame]
Brett Cannon6f44d662012-04-15 16:08:47 -04001"""This module provides the components needed to build your own __import__
2function. Undocumented functions are obsolete.
3
4In most cases it is preferred you consider using the importlib module's
5functionality over this module.
6
7"""
8# (Probably) need to stay in _imp
9from _imp import (lock_held, acquire_lock, release_lock, reload,
Brett Cannon2fef4d22012-04-15 19:06:23 -040010 load_dynamic, get_frozen_object, is_frozen_package,
11 init_builtin, init_frozen, is_builtin, is_frozen,
12 _fix_co_filename)
Brett Cannon6f44d662012-04-15 16:08:47 -040013# Can (probably) move to importlib
14from _imp import (get_magic, get_tag, get_suffixes, cache_from_source,
15 source_from_cache)
16# Should be re-implemented here (and mostly deprecated)
Brett Cannon2ee61422012-04-15 22:28:28 -040017from _imp import (find_module, load_compiled, load_source, NullImporter,
Brett Cannon6f44d662012-04-15 16:08:47 -040018 SEARCH_ERROR, PY_SOURCE, PY_COMPILED, C_EXTENSION,
19 PY_RESOURCE, PKG_DIRECTORY, C_BUILTIN, PY_FROZEN,
20 PY_CODERESOURCE, IMP_HOOK)
21
22from importlib._bootstrap import _new_module as new_module
Brett Cannon01a76172012-04-15 20:25:23 -040023
Brett Cannon2ee61422012-04-15 22:28:28 -040024from importlib import _bootstrap
25import os
26
27
28def load_package(name, path):
29 if os.path.isdir(path):
30 extensions = _bootstrap._suffix_list(PY_SOURCE)
31 extensions += _bootstrap._suffix_list(PY_COMPILED)
32 for extension in extensions:
33 path = os.path.join(path, '__init__'+extension)
34 if os.path.exists(path):
35 break
36 else:
37 raise ValueError('{!r} is not a package'.format(path))
38 return _bootstrap._SourceFileLoader(name, path).load_module(name)
39
Brett Cannon01a76172012-04-15 20:25:23 -040040
41def load_module(name, file, filename, details):
42 """Load a module, given information returned by find_module().
43
44 The module name must include the full package name, if any.
45
46 """
47 suffix, mode, type_ = details
Brett Cannonde10bf42012-04-16 20:44:21 -040048 if mode and (not mode.startswith(('r', 'U')) or '+' in mode):
Brett Cannon01a76172012-04-15 20:25:23 -040049 raise ValueError('invalid file open mode {!r}'.format(mode))
50 elif file is None and type_ in {PY_SOURCE, PY_COMPILED}:
51 msg = 'file object required for import (type code {})'.format(type_)
52 raise ValueError(msg)
53 elif type_ == PY_SOURCE:
54 return load_source(name, filename, file)
55 elif type_ == PY_COMPILED:
56 return load_compiled(name, filename, file)
57 elif type_ == PKG_DIRECTORY:
58 return load_package(name, filename)
59 elif type_ == C_BUILTIN:
60 return init_builtin(name)
61 elif type_ == PY_FROZEN:
62 return init_frozen(name)
63 else:
64 msg = "Don't know how to import {} (type code {}".format(name, type_)
65 raise ImportError(msg, name=name)