sh_cpl/src/cpl_reactive_extensions/observer.py

from typing import Callable

from cpl_core.type import T


class Observer:
    def __init__(self, on_next: Callable, on_error: Callable = None, on_complete: Callable = None):
        self._on_next = on_next
        self._on_error = on_error
        self._on_complete = on_complete

        self._closed = False

    @property
    def closed(self) -> bool:
        return self._closed

    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()