blob: 1d3c39788c3bdafff67d310689413e87270e0aba (
about) (
plain)
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
|
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()
|
