djavu-asus/elpy/rpc-venv/lib/python3.11/site-packages/setuptools/_imp.py

   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)