venv added, updated

myenv/lib/python3.12/site-packages/reactivex/scheduler/eventloop/asynciothreadsafescheduler.py

@@ -0,0 +1,157 @@
+import asyncio
+import logging
+from concurrent.futures import Future
+from typing import List, Optional, TypeVar
+
+from reactivex import abc, typing
+from reactivex.disposable import (
+    CompositeDisposable,
+    Disposable,
+    SingleAssignmentDisposable,
+)
+
+from .asyncioscheduler import AsyncIOScheduler
+
+_TState = TypeVar("_TState")
+
+log = logging.getLogger("Rx")
+
+
+class AsyncIOThreadSafeScheduler(AsyncIOScheduler):
+    """A scheduler that schedules work via the asyncio mainloop. This is a
+    subclass of AsyncIOScheduler which uses the threadsafe asyncio methods.
+    """
+
+    def schedule(
+        self, action: typing.ScheduledAction[_TState], state: Optional[_TState] = None
+    ) -> abc.DisposableBase:
+        """Schedules an action to be executed.
+
+        Args:
+            action: Action to be executed.
+            state: [Optional] state to be given to the action function.
+
+        Returns:
+            The disposable object used to cancel the scheduled action
+            (best effort).
+        """
+        sad = SingleAssignmentDisposable()
+
+        def interval() -> None:
+            sad.disposable = self.invoke_action(action, state=state)
+
+        handle = self._loop.call_soon_threadsafe(interval)
+
+        def dispose() -> None:
+            if self._on_self_loop_or_not_running():
+                handle.cancel()
+                return
+
+            future: "Future[int]" = Future()
+
+            def cancel_handle() -> None:
+                handle.cancel()
+                future.set_result(0)
+
+            self._loop.call_soon_threadsafe(cancel_handle)
+            future.result()
+
+        return CompositeDisposable(sad, Disposable(dispose))
+
+    def schedule_relative(
+        self,
+        duetime: typing.RelativeTime,
+        action: typing.ScheduledAction[_TState],
+        state: Optional[_TState] = None,
+    ) -> abc.DisposableBase:
+        """Schedules an action to be executed after duetime.
+
+        Args:
+            duetime: Relative time after which to execute the action.
+            action: Action to be executed.
+            state: [Optional] state to be given to the action function.
+
+        Returns:
+            The disposable object used to cancel the scheduled action
+            (best effort).
+        """
+        seconds = self.to_seconds(duetime)
+        if seconds <= 0:
+            return self.schedule(action, state=state)
+
+        sad = SingleAssignmentDisposable()
+
+        def interval() -> None:
+            sad.disposable = self.invoke_action(action, state=state)
+
+        # the operations on the list used here are atomic, so there is no
+        # need to protect its access with a lock
+        handle: List[asyncio.Handle] = []
+
+        def stage2() -> None:
+            handle.append(self._loop.call_later(seconds, interval))
+
+        handle.append(self._loop.call_soon_threadsafe(stage2))
+
+        def dispose() -> None:
+            def do_cancel_handles() -> None:
+                try:
+                    handle.pop().cancel()
+                    handle.pop().cancel()
+                except Exception:
+                    pass
+
+            if self._on_self_loop_or_not_running():
+                do_cancel_handles()
+                return
+
+            future: "Future[int]" = Future()
+
+            def cancel_handle() -> None:
+                do_cancel_handles()
+                future.set_result(0)
+
+            self._loop.call_soon_threadsafe(cancel_handle)
+            future.result()
+
+        return CompositeDisposable(sad, Disposable(dispose))
+
+    def schedule_absolute(
+        self,
+        duetime: typing.AbsoluteTime,
+        action: typing.ScheduledAction[_TState],
+        state: Optional[_TState] = None,
+    ) -> abc.DisposableBase:
+        """Schedules an action to be executed at duetime.
+
+        Args:
+            duetime: Absolute time at which to execute the action.
+            action: Action to be executed.
+            state: [Optional] state to be given to the action function.
+
+        Returns:
+            The disposable object used to cancel the scheduled action
+            (best effort).
+        """
+
+        duetime = self.to_datetime(duetime)
+        return self.schedule_relative(duetime - self.now, action, state=state)
+
+    def _on_self_loop_or_not_running(self) -> bool:
+        """
+        Returns True if either self._loop is not running, or we're currently
+        executing on self._loop. In both cases, waiting for a future to be
+        resolved on the loop would result in a deadlock.
+        """
+        if not self._loop.is_running():
+            return True
+        current_loop = None
+        try:
+            # In python 3.7 there asyncio.get_running_loop() is prefered.
+            current_loop = asyncio.get_event_loop()
+        except RuntimeError:
+            # If there is no loop in current thread at all, and it is not main
+            # thread, we get error like:
+            # RuntimeError: There is no current event loop in thread 'Thread-1'
+            pass
+        return self._loop == current_loop