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 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
|
from collections.abc import Callable
from typing import TYPE_CHECKING, TypeVar
from returns.interfaces.specific.reader_future_result import (
ReaderFutureResultLikeN,
)
from returns.primitives.hkt import Kinded, KindN, kinded
if TYPE_CHECKING:
from returns.context import ReaderFutureResult # noqa: WPS433
_FirstType = TypeVar('_FirstType')
_SecondType = TypeVar('_SecondType')
_ThirdType = TypeVar('_ThirdType')
_UpdatedType = TypeVar('_UpdatedType')
_ReaderFutureResultLikeKind = TypeVar(
'_ReaderFutureResultLikeKind',
bound=ReaderFutureResultLikeN,
)
def bind_context_future_result(
function: Callable[
[_FirstType],
'ReaderFutureResult[_UpdatedType, _SecondType, _ThirdType]',
],
) -> Kinded[
Callable[
[
KindN[
_ReaderFutureResultLikeKind, _FirstType, _SecondType, _ThirdType
]
],
KindN[
_ReaderFutureResultLikeKind, _UpdatedType, _SecondType, _ThirdType
],
]
]:
"""
Lifts function from ``RequiresContextFutureResult`` for better composition.
In other words, it modifies the function's
signature from:
``a -> RequiresContextFutureResult[env, b, c]``
to:
``Container[env, a, c]`` -> ``Container[env, b, c]``
.. code:: python
>>> import anyio
>>> from returns.context import ReaderFutureResult
>>> from returns.io import IOSuccess, IOFailure
>>> from returns.future import FutureResult
>>> from returns.pointfree import bind_context_future_result
>>> def function(arg: int) -> ReaderFutureResult[str, int, str]:
... return ReaderFutureResult(
... lambda deps: FutureResult.from_value(len(deps) + arg),
... )
>>> assert anyio.run(bind_context_future_result(function)(
... ReaderFutureResult.from_value(2),
... )('abc').awaitable) == IOSuccess(5)
>>> assert anyio.run(bind_context_future_result(function)(
... ReaderFutureResult.from_failure(0),
... )('abc').awaitable) == IOFailure(0)
"""
@kinded
def factory(
container: KindN[
_ReaderFutureResultLikeKind,
_FirstType,
_SecondType,
_ThirdType,
],
) -> KindN[
_ReaderFutureResultLikeKind,
_UpdatedType,
_SecondType,
_ThirdType,
]:
return container.bind_context_future_result(function)
return factory
|