firestar5683
25 lines
619 B
Python
25 lines
619 B
Python
from typing import Callable, List, Optional, TypeVar
|
|
|
|
from reactivex import Observable, abc, compose
|
|
from reactivex import operators as ops
|
|
from reactivex import typing
|
|
|
|
_T = TypeVar("_T")
|
|
|
|
|
|
def buffer_with_time_(
|
|
timespan: typing.RelativeTime,
|
|
timeshift: Optional[typing.RelativeTime] = None,
|
|
scheduler: Optional[abc.SchedulerBase] = None,
|
|
) -> Callable[[Observable[_T]], Observable[List[_T]]]:
|
|
if not timeshift:
|
|
timeshift = timespan
|
|
|
|
return compose(
|
|
ops.window_with_time(timespan, timeshift, scheduler),
|
|
ops.flat_map(ops.to_list()),
|
|
)
|
|
|
|
|
|
__all__ = ["buffer_with_time_"]
|