wmantly
70c71161f3
- FastAPI backend with full Proxmox VE API integration - ISO builder using proxmox-auto-install-assistant - Premium dark-mode SPA frontend with glassmorphism design - VM lifecycle management (create, start, stop, destroy) - Build pipeline tracking with real-time logs - Deployment automation for custom auto-installer ISOs - Production deployment script (setup.sh + systemd) - Comprehensive README with API documentation
112 lines
2.9 KiB
Python
112 lines
2.9 KiB
Python
from __future__ import annotations
|
|
|
|
import functools
|
|
import sys
|
|
from collections.abc import AsyncGenerator, Awaitable, Callable, Generator
|
|
from contextlib import AbstractAsyncContextManager, asynccontextmanager
|
|
from typing import Any, Generic, Protocol, TypeVar, overload
|
|
|
|
import anyio.abc
|
|
|
|
from starlette.types import Scope
|
|
|
|
if sys.version_info >= (3, 13): # pragma: no cover
|
|
from inspect import iscoroutinefunction
|
|
from typing import TypeIs
|
|
else: # pragma: no cover
|
|
from asyncio import iscoroutinefunction
|
|
|
|
from typing_extensions import TypeIs
|
|
|
|
if sys.version_info < (3, 11): # pragma: no cover
|
|
try:
|
|
from exceptiongroup import BaseExceptionGroup
|
|
except ImportError:
|
|
|
|
class BaseExceptionGroup(BaseException): # type: ignore[no-redef]
|
|
pass
|
|
|
|
|
|
T = TypeVar("T")
|
|
AwaitableCallable = Callable[..., Awaitable[T]]
|
|
|
|
|
|
@overload
|
|
def is_async_callable(obj: AwaitableCallable[T]) -> TypeIs[AwaitableCallable[T]]: ...
|
|
|
|
|
|
@overload
|
|
def is_async_callable(obj: Any) -> TypeIs[AwaitableCallable[Any]]: ...
|
|
|
|
|
|
def is_async_callable(obj: Any) -> Any:
|
|
while isinstance(obj, functools.partial):
|
|
obj = obj.func
|
|
|
|
return iscoroutinefunction(obj) or (callable(obj) and iscoroutinefunction(obj.__call__))
|
|
|
|
|
|
T_co = TypeVar("T_co", covariant=True)
|
|
|
|
|
|
class AwaitableOrContextManager(
|
|
Awaitable[T_co], AbstractAsyncContextManager[T_co], Protocol[T_co]
|
|
): ... # pragma: no branch
|
|
|
|
|
|
class SupportsAsyncClose(Protocol):
|
|
async def close(self) -> None: ... # pragma: no cover
|
|
|
|
|
|
SupportsAsyncCloseType = TypeVar("SupportsAsyncCloseType", bound=SupportsAsyncClose, covariant=False)
|
|
|
|
|
|
class AwaitableOrContextManagerWrapper(Generic[SupportsAsyncCloseType]):
|
|
__slots__ = ("aw", "entered")
|
|
|
|
def __init__(self, aw: Awaitable[SupportsAsyncCloseType]) -> None:
|
|
self.aw = aw
|
|
|
|
def __await__(self) -> Generator[Any, None, SupportsAsyncCloseType]:
|
|
return self.aw.__await__()
|
|
|
|
async def __aenter__(self) -> SupportsAsyncCloseType:
|
|
self.entered = await self.aw
|
|
return self.entered
|
|
|
|
async def __aexit__(self, *args: Any) -> None | bool:
|
|
await self.entered.close()
|
|
return None
|
|
|
|
|
|
@asynccontextmanager
|
|
async def create_collapsing_task_group() -> AsyncGenerator[anyio.abc.TaskGroup, None]:
|
|
try:
|
|
async with anyio.create_task_group() as tg:
|
|
yield tg
|
|
except BaseExceptionGroup as excs:
|
|
if len(excs.exceptions) != 1:
|
|
raise
|
|
|
|
exc = excs.exceptions[0]
|
|
context = None if exc.__suppress_context__ else exc.__context__
|
|
raise exc from exc.__cause__ or context
|
|
|
|
|
|
def get_route_path(scope: Scope) -> str:
|
|
path: str = scope["path"]
|
|
root_path = scope.get("root_path", "")
|
|
if not root_path:
|
|
return path
|
|
|
|
if not path.startswith(root_path):
|
|
return path
|
|
|
|
if path == root_path:
|
|
return ""
|
|
|
|
if path[len(root_path)] == "/":
|
|
return path[len(root_path) :]
|
|
|
|
return path
|