File size: 2,431 Bytes
c12ef43 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | 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)
|