two version of R2R are hereHEADmaster

1 files changed, 28 insertions, 0 deletions

diff --git a/.venv/lib/python3.12/site-packages/hatchet_sdk/clients/event_ts.py b/.venv/lib/python3.12/site-packages/hatchet_sdk/clients/event_ts.py
new file mode 100644
index 00000000..1d3c3978
--- /dev/null
+++ b/.venv/lib/python3.12/site-packages/hatchet_sdk/clients/event_ts.py
@@ -0,0 +1,28 @@
+import asyncio
+from typing import Any
+
+
+class Event_ts(asyncio.Event):
+    """
+    Event_ts is a subclass of asyncio.Event that allows for thread-safe setting and clearing of the event.
+    """
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        if self._loop is None:
+            self._loop = asyncio.get_event_loop()
+
+    def set(self):
+        if not self._loop.is_closed():
+            self._loop.call_soon_threadsafe(super().set)
+
+    def clear(self):
+        self._loop.call_soon_threadsafe(super().clear)
+
+
+async def read_with_interrupt(listener: Any, interrupt: Event_ts):
+    try:
+        result = await listener.read()
+        return result
+    finally:
+        interrupt.set()