sh_cpl/src/cpl_reactive_extensions/subscriber.py

from typing import Callable

from cpl_core.type import T
from cpl_reactive_extensions.abc.observer import Observer
from cpl_reactive_extensions.subscription import Subscription
from cpl_reactive_extensions.type import ObserverOrCallable


class Subscriber(Subscription, Observer):
    def __init__(
        self, on_next_or_observer: ObserverOrCallable, on_error: Callable = None, on_complete: Callable = None
    ):
        Subscription.__init__(self)
        if isinstance(on_next_or_observer, Observer):
            self._on_next = on_next_or_observer.next
            self._on_error = on_next_or_observer.error
            self._on_complete = on_next_or_observer.complete
        else:
            self._on_next = on_next_or_observer
            self._on_error = on_error
            self._on_complete = on_complete

    def next(self, value: T):
        if self._closed:
            raise Exception("Observer is closed")

        self._on_next(value)

    def error(self, ex: Exception):
        if self._on_error is None:
            return
        self._on_error(ex)

    def complete(self):
        self._closed = True
        if self._on_complete is None:
            return

        self._on_complete()

    def unsubscribe(self):
        if self._closed:
            return

        super().unsubscribe()