| | 1 | """ |
| | 2 | Re-implementation of find_module and get_frozen_object |
| | 3 | from the deprecated imp module. |
| | 4 | """ |
| | 5 | |
| | 6 | import os |
| | 7 | import importlib.util |
| | 8 | import importlib.machinery |
| | 9 | |
| | 10 | from .py34compat import module_from_spec |
| | 11 | |
| | 12 | |
| | 13 | PY_SOURCE = 1 |
| | 14 | PY_COMPILED = 2 |
| | 15 | C_EXTENSION = 3 |
| | 16 | C_BUILTIN = 6 |
| | 17 | PY_FROZEN = 7 |
| | 18 | |
| | 19 | |
| | 20 | def find_spec(module, paths): |
| | 21 | finder = ( |
| | 22 | importlib.machinery.PathFinder().find_spec |
| | 23 | if isinstance(paths, list) else |
| | 24 | importlib.util.find_spec |
| | 25 | ) |
| | 26 | return finder(module, paths) |
| | 27 | |
| | 28 | |
| | 29 | def find_module(module, paths=None): |
| | 30 | """Just like 'imp.find_module()', but with package support""" |
| | 31 | spec = find_spec(module, paths) |
| | 32 | if spec is None: |
| | 33 | raise ImportError("Can't find %s" % module) |
| | 34 | if not spec.has_location and hasattr(spec, 'submodule_search_locations'): |
| | 35 | spec = importlib.util.spec_from_loader('__init__.py', spec.loader) |
| | 36 | |
| | 37 | kind = -1 |
| | 38 | file = None |
| | 39 | static = isinstance(spec.loader, type) |
| | 40 | if spec.origin == 'frozen' or static and issubclass( |
| | 41 | spec.loader, importlib.machinery.FrozenImporter): |
| | 42 | kind = PY_FROZEN |
| | 43 | path = None # imp compabilty |
| | 44 | suffix = mode = '' # imp compatibility |
| | 45 | elif spec.origin == 'built-in' or static and issubclass( |
| | 46 | spec.loader, importlib.machinery.BuiltinImporter): |
| | 47 | kind = C_BUILTIN |
| | 48 | path = None # imp compabilty |
| | 49 | suffix = mode = '' # imp compatibility |
| | 50 | elif spec.has_location: |
| | 51 | path = spec.origin |
| | 52 | suffix = os.path.splitext(path)[1] |
| | 53 | mode = 'r' if suffix in importlib.machinery.SOURCE_SUFFIXES else 'rb' |
| | 54 | |
| | 55 | if suffix in importlib.machinery.SOURCE_SUFFIXES: |
| | 56 | kind = PY_SOURCE |
| | 57 | elif suffix in importlib.machinery.BYTECODE_SUFFIXES: |
| | 58 | kind = PY_COMPILED |
| | 59 | elif suffix in importlib.machinery.EXTENSION_SUFFIXES: |
| | 60 | kind = C_EXTENSION |
| | 61 | |
| | 62 | if kind in {PY_SOURCE, PY_COMPILED}: |
| | 63 | file = open(path, mode) |
| | 64 | else: |
| | 65 | path = None |
| | 66 | suffix = mode = '' |
| | 67 | |
| | 68 | return file, path, (suffix, mode, kind) |
| | 69 | |
| | 70 | |
| | 71 | def get_frozen_object(module, paths=None): |
| | 72 | spec = find_spec(module, paths) |
| | 73 | if not spec: |
| | 74 | raise ImportError("Can't find %s" % module) |
| | 75 | return spec.loader.get_code(module) |
| | 76 | |
| | 77 | |
| | 78 | def get_module(module, paths, info): |
| | 79 | spec = find_spec(module, paths) |
| | 80 | if not spec: |
| | 81 | raise ImportError("Can't find %s" % module) |
| | 82 | return module_from_spec(spec) |
projects / config.git / blame_incremental