GPTCache

1
import importlib
2
from types import ModuleType
3

4

5
class LazyImport(ModuleType):
6
    """
7
    Lazily import a module.
8
    """
9

10
    def __init__(self, local_name, parent_module_globals, name):
11
        self._local_name = local_name
12
        self._parent_module_globals = parent_module_globals
13
        super().__init__(name)
14

15
    def _load(self):
16
        module = importlib.import_module(self.__name__)
17
        self._parent_module_globals[self._local_name] = module
18
        self.__dict__.update(module.__dict__)
19
        return module
20

21
    def __getattr__(self, item):
22
        module = self._load()
23
        return getattr(module, item)
24

25
    def __dir__(self):
26
        module = self._load()
27
        return dir(module)
28