firestar5683
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
from typing import Callable, Optional, TypeVar
|
|
|
|
import reactivex
|
|
from reactivex import Observable
|
|
from reactivex.internal.utils import infinite
|
|
|
|
_T = TypeVar("_T")
|
|
|
|
|
|
def retry_(
|
|
retry_count: Optional[int] = None,
|
|
) -> Callable[[Observable[_T]], Observable[_T]]:
|
|
"""Repeats the source observable sequence the specified number of
|
|
times or until it successfully terminates. If the retry count is
|
|
not specified, it retries indefinitely.
|
|
|
|
Examples:
|
|
>>> retried = retry()
|
|
>>> retried = retry(42)
|
|
|
|
Args:
|
|
retry_count: [Optional] Number of times to retry the sequence.
|
|
If not provided, retry the sequence indefinitely.
|
|
|
|
Returns:
|
|
An observable sequence producing the elements of the given
|
|
sequence repeatedly until it terminates successfully.
|
|
"""
|
|
|
|
if retry_count is None:
|
|
gen = infinite()
|
|
else:
|
|
gen = range(retry_count)
|
|
|
|
def retry(source: Observable[_T]) -> Observable[_T]:
|
|
return reactivex.catch_with_iterable(source for _ in gen)
|
|
|
|
return retry
|
|
|
|
|
|
__all__ = ["retry_"]
|