72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib
|
|
import importlib.abc
|
|
import importlib.machinery
|
|
import importlib.util
|
|
import sys
|
|
from collections.abc import Sequence
|
|
from types import ModuleType
|
|
|
|
__all__ = ["alias_httpx"]
|
|
|
|
|
|
class _AliasLoader(importlib.abc.Loader):
|
|
_original_spec: importlib.machinery.ModuleSpec | None
|
|
|
|
def __init__(self, real_name: str) -> None:
|
|
self._real_name = real_name
|
|
|
|
def create_module(self, spec: importlib.machinery.ModuleSpec) -> ModuleType:
|
|
module = importlib.import_module(self._real_name)
|
|
self._original_spec = module.__spec__
|
|
return module
|
|
|
|
def exec_module(self, module: ModuleType) -> None:
|
|
module.__spec__ = self._original_spec
|
|
|
|
|
|
class _AliasFinder(importlib.abc.MetaPathFinder):
|
|
def __init__(self, alias: str, real: str) -> None:
|
|
self._alias = alias
|
|
self._real = real
|
|
|
|
def find_spec(
|
|
self,
|
|
fullname: str,
|
|
path: Sequence[str] | None = None,
|
|
target: ModuleType | None = None,
|
|
) -> importlib.machinery.ModuleSpec | None:
|
|
if fullname != self._alias and not fullname.startswith(self._alias + "."):
|
|
return None
|
|
real_name = self._real + fullname.removeprefix(self._alias)
|
|
if real_name not in sys.modules and importlib.util.find_spec(real_name) is None:
|
|
return None
|
|
return importlib.machinery.ModuleSpec(fullname, _AliasLoader(real_name))
|
|
|
|
|
|
def _alias(alias: str, module: ModuleType) -> None:
|
|
existing = sys.modules.get(alias)
|
|
if existing is not None and existing is not module:
|
|
raise RuntimeError(f"{alias} was already imported; call `alias_httpx()` before any `import {alias}`.")
|
|
|
|
if not any(isinstance(finder, _AliasFinder) and finder._alias == alias for finder in sys.meta_path):
|
|
sys.meta_path.insert(0, _AliasFinder(alias, module.__name__))
|
|
sys.modules[alias] = module
|
|
|
|
|
|
def alias_httpx() -> None:
|
|
"""
|
|
Make `import httpx` resolve to `httpx2`, and `import httpcore` to `httpcore2`, process-wide.
|
|
|
|
Intended for applications migrating from `httpx`, so that dependencies still
|
|
importing `httpx` or `httpcore` share the `httpx2` classes. Libraries should never call this.
|
|
|
|
Must be called before anything imports `httpx` or `httpcore`. Calling it again is a no-op.
|
|
"""
|
|
import httpcore2
|
|
import httpx2
|
|
|
|
_alias("httpx", httpx2)
|
|
_alias("httpcore", httpcore2)
|