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
|
from collections.abc import Callable
from typing import TypeVar
from returns.interfaces.specific.maybe import MaybeLikeN
from returns.primitives.hkt import Kinded, KindN, kinded
_FirstType = TypeVar('_FirstType')
_SecondType = TypeVar('_SecondType')
_ThirdType = TypeVar('_ThirdType')
_UpdatedType = TypeVar('_UpdatedType')
_MaybeLikeKind = TypeVar('_MaybeLikeKind', bound=MaybeLikeN)
def bind_optional(
function: Callable[[_FirstType], _UpdatedType | None],
) -> Kinded[
Callable[
[KindN[_MaybeLikeKind, _FirstType, _SecondType, _ThirdType]],
KindN[_MaybeLikeKind, _UpdatedType, _SecondType, _ThirdType],
]
]:
"""
Binds a function returning optional value over a container.
In other words, it modifies the function's
signature from:
``a -> Optional[b]``
to:
``Container[a] -> Container[b]``
.. code:: python
>>> from typing import Optional
>>> from returns.pointfree import bind_optional
>>> from returns.maybe import Some, Nothing
>>> def example(argument: int) -> Optional[int]:
... return argument + 1 if argument > 0 else None
>>> assert bind_optional(example)(Some(1)) == Some(2)
>>> assert bind_optional(example)(Some(0)) == Nothing
>>> assert bind_optional(example)(Nothing) == Nothing
Note, that this function works
for all containers with ``.bind_optional`` method.
See :class:`returns.primitives.interfaces.specific.maybe._MaybeLikeKind`
for more info.
"""
@kinded
def factory(
container: KindN[_MaybeLikeKind, _FirstType, _SecondType, _ThirdType],
) -> KindN[_MaybeLikeKind, _UpdatedType, _SecondType, _ThirdType]:
return container.bind_optional(function)
return factory
|