mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-12-28 08:21:45 +01:00
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
# Copyright 2018
|
|
# Licensed under the MIT license
|
|
|
|
|
|
from functools import wraps
|
|
from typing import Any, Callable, Coroutine, Generator, TypeVar
|
|
|
|
from typing_extensions import AsyncContextManager
|
|
|
|
from .cursor import Cursor
|
|
|
|
_T = TypeVar("_T")
|
|
|
|
|
|
class Result(AsyncContextManager[_T], Coroutine[Any, Any, _T]):
|
|
__slots__ = ("_coro", "_obj")
|
|
|
|
def __init__(self, coro: Coroutine[Any, Any, _T]):
|
|
self._coro = coro
|
|
self._obj: _T
|
|
|
|
def send(self, value) -> None:
|
|
return self._coro.send(value)
|
|
|
|
def throw(self, typ, val=None, tb=None) -> None:
|
|
if val is None:
|
|
return self._coro.throw(typ)
|
|
|
|
if tb is None:
|
|
return self._coro.throw(typ, val)
|
|
|
|
return self._coro.throw(typ, val, tb)
|
|
|
|
def close(self) -> None:
|
|
return self._coro.close()
|
|
|
|
def __await__(self) -> Generator[Any, None, _T]:
|
|
return self._coro.__await__()
|
|
|
|
async def __aenter__(self) -> _T:
|
|
self._obj = await self._coro
|
|
return self._obj
|
|
|
|
async def __aexit__(self, exc_type, exc, tb) -> None:
|
|
if isinstance(self._obj, Cursor):
|
|
await self._obj.close()
|
|
|
|
|
|
def contextmanager(
|
|
method: Callable[..., Coroutine[Any, Any, _T]]
|
|
) -> Callable[..., Result[_T]]:
|
|
@wraps(method)
|
|
def wrapper(self, *args, **kwargs) -> Result[_T]:
|
|
return Result(method(self, *args, **kwargs))
|
|
|
|
return wrapper
|