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
|
from typing import Callable, Optional, TypeVar
from reactivex import Observable, abc, typing
from reactivex.disposable import Disposable
_T = TypeVar("_T")
def finally_action_(
action: typing.Action,
) -> Callable[[Observable[_T]], Observable[_T]]:
def finally_action(source: Observable[_T]) -> Observable[_T]:
"""Invokes a specified action after the source observable
sequence terminates gracefully or exceptionally.
Example:
res = finally(source)
Args:
source: Observable sequence.
Returns:
An observable sequence with the action-invoking termination
behavior applied.
"""
def subscribe(
observer: abc.ObserverBase[_T],
scheduler: Optional[abc.SchedulerBase] = None,
) -> abc.DisposableBase:
try:
subscription = source.subscribe(observer, scheduler=scheduler)
except Exception:
action()
raise
def dispose():
try:
subscription.dispose()
finally:
action()
return Disposable(dispose)
return Observable(subscribe)
return finally_action
__all__ = ["finally_action_"]
|