File: _toset.py

package info (click to toggle)
python-rx 4.1.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 4,204 kB
  • sloc: python: 39,525; javascript: 77; makefile: 24
file content (37 lines) | stat: -rw-r--r-- 996 bytes parent folder | download | duplicates (2)
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
from typing import Callable, Optional, Set, TypeVar

from reactivex import Observable, abc

_T = TypeVar("_T")


def to_set_() -> Callable[[Observable[_T]], Observable[Set[_T]]]:
    """Converts the observable sequence to a set.

    Returns an observable sequence with a single value of a set
    containing the values from the observable sequence.
    """

    def to_set(source: Observable[_T]) -> Observable[Set[_T]]:
        def subscribe(
            observer: abc.ObserverBase[Set[_T]],
            scheduler: Optional[abc.SchedulerBase] = None,
        ) -> abc.DisposableBase:
            s: Set[_T] = set()

            def on_completed() -> None:
                nonlocal s
                observer.on_next(s)
                s = set()
                observer.on_completed()

            return source.subscribe(
                s.add, observer.on_error, on_completed, scheduler=scheduler
            )

        return Observable(subscribe)

    return to_set


__all__ = ["to_set_"]