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
|
from typing import Callable, Optional, TypeVar
from reactivex import Observable, abc
_T = TypeVar("_T")
def default_if_empty_(
default_value: Optional[_T] = None,
) -> Callable[[Observable[_T]], Observable[Optional[_T]]]:
def default_if_empty(source: Observable[_T]) -> Observable[Optional[_T]]:
"""Returns the elements of the specified sequence or the
specified value in a singleton sequence if the sequence is
empty.
Examples:
>>> obs = default_if_empty(source)
Args:
source: Source observable.
Returns:
An observable sequence that contains the specified default
value if the source is empty otherwise, the elements of the
source.
"""
def subscribe(
observer: abc.ObserverBase[Optional[_T]],
scheduler: Optional[abc.SchedulerBase] = None,
) -> abc.DisposableBase:
found = [False]
def on_next(x: _T):
found[0] = True
observer.on_next(x)
def on_completed():
if not found[0]:
observer.on_next(default_value)
observer.on_completed()
return source.subscribe(
on_next, observer.on_error, on_completed, scheduler=scheduler
)
return Observable(subscribe)
return default_if_empty
__all__ = ["default_if_empty_"]
|