60 lines
1.2 KiB
Python
60 lines
1.2 KiB
Python
from collections.abc import Awaitable
|
|
from inspect import CO_ITERABLE_COROUTINE
|
|
from types import CoroutineType
|
|
from types import GeneratorType
|
|
|
|
|
|
async def do_await(obj):
|
|
return await obj
|
|
|
|
|
|
def do_yield_from(gen):
|
|
return (yield from gen)
|
|
|
|
|
|
def await_(obj):
|
|
obj_type = type(obj)
|
|
if (
|
|
obj_type is CoroutineType
|
|
or (obj_type is GeneratorType and bool(obj.gi_code.co_flags & CO_ITERABLE_COROUTINE))
|
|
or isinstance(obj, Awaitable)
|
|
):
|
|
return do_await(obj).__await__()
|
|
else:
|
|
return do_yield_from(obj)
|
|
|
|
|
|
def __aiter__(self):
|
|
return self.__wrapped__.__aiter__()
|
|
|
|
|
|
async def __anext__(self):
|
|
return await self.__wrapped__.__anext__()
|
|
|
|
|
|
def __await__(self):
|
|
return await_(self.__wrapped__)
|
|
|
|
|
|
def __aenter__(self):
|
|
return self.__wrapped__.__aenter__()
|
|
|
|
|
|
def __aexit__(self, *args, **kwargs):
|
|
return self.__wrapped__.__aexit__(*args, **kwargs)
|
|
|
|
|
|
def identity(obj):
|
|
return obj
|
|
|
|
|
|
class cached_property:
|
|
def __init__(self, func):
|
|
self.func = func
|
|
|
|
def __get__(self, obj, cls):
|
|
if obj is None:
|
|
return self
|
|
value = obj.__dict__[self.func.__name__] = self.func(obj)
|
|
return value
|