Archived
80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
"""
|
|
Re-implementation of find_module and get_frozen_object
|
|
from the deprecated imp module.
|
|
"""
|
|
|
|
import os
|
|
import importlib.util
|
|
import importlib.machinery
|
|
|
|
from .py34compat import module_from_spec
|
|
|
|
|
|
PY_SOURCE = 1
|
|
PY_COMPILED = 2
|
|
C_EXTENSION = 3
|
|
C_BUILTIN = 6
|
|
PY_FROZEN = 7
|
|
|
|
|
|
def find_module(module, paths=None):
|
|
"""Just like 'imp.find_module()', but with package support"""
|
|
spec = importlib.util.find_spec(module, paths)
|
|
if spec is None:
|
|
raise ImportError("Can't find %s" % module)
|
|
if not spec.has_location and hasattr(spec, "submodule_search_locations"):
|
|
spec = importlib.util.spec_from_loader("__init__.py", spec.loader)
|
|
|
|
kind = -1
|
|
file = None
|
|
static = isinstance(spec.loader, type)
|
|
if (
|
|
spec.origin == "frozen"
|
|
or static
|
|
and issubclass(spec.loader, importlib.machinery.FrozenImporter)
|
|
):
|
|
kind = PY_FROZEN
|
|
path = None # imp compabilty
|
|
suffix = mode = "" # imp compability
|
|
elif (
|
|
spec.origin == "built-in"
|
|
or static
|
|
and issubclass(spec.loader, importlib.machinery.BuiltinImporter)
|
|
):
|
|
kind = C_BUILTIN
|
|
path = None # imp compabilty
|
|
suffix = mode = "" # imp compability
|
|
elif spec.has_location:
|
|
path = spec.origin
|
|
suffix = os.path.splitext(path)[1]
|
|
mode = "r" if suffix in importlib.machinery.SOURCE_SUFFIXES else "rb"
|
|
|
|
if suffix in importlib.machinery.SOURCE_SUFFIXES:
|
|
kind = PY_SOURCE
|
|
elif suffix in importlib.machinery.BYTECODE_SUFFIXES:
|
|
kind = PY_COMPILED
|
|
elif suffix in importlib.machinery.EXTENSION_SUFFIXES:
|
|
kind = C_EXTENSION
|
|
|
|
if kind in {PY_SOURCE, PY_COMPILED}:
|
|
file = open(path, mode)
|
|
else:
|
|
path = None
|
|
suffix = mode = ""
|
|
|
|
return file, path, (suffix, mode, kind)
|
|
|
|
|
|
def get_frozen_object(module, paths=None):
|
|
spec = importlib.util.find_spec(module, paths)
|
|
if not spec:
|
|
raise ImportError("Can't find %s" % module)
|
|
return spec.loader.get_code(module)
|
|
|
|
|
|
def get_module(module, paths, info):
|
|
spec = importlib.util.find_spec(module, paths)
|
|
if not spec:
|
|
raise ImportError("Can't find %s" % module)
|
|
return module_from_spec(spec)
|