Update nest-asyncio/README.md

This commit is contained in:
2024-11-25 20:08:33 -05:00
parent 3ba8d7db0e
commit 744b0e7c29

View File

@@ -1,19 +1,24 @@
```python ```python
import asyncio def sync(coro):
def _run_once(loop): import asyncio, functools
if not asyncio.iscoroutinefunction(coro): return coro
@functools.wraps(coro)
def wrapper(*args, **kwargs):
loop, future = asyncio.get_event_loop(), asyncio.ensure_future(coro(*args, **kwargs))
while not future.done():
loop._process_events(loop._selector.select(0)) loop._process_events(loop._selector.select(0))
if (ready := loop._ready) and (handle := ready.popleft())._cancelled is False: if (ready := loop._ready) and (handle := ready.popleft())._cancelled is False:
task = (tasks := asyncio.tasks._current_tasks).pop(loop, None) task = (tasks := asyncio.tasks._current_tasks).pop(loop, None)
handle._run(); tasks[loop] = task handle._run(); tasks[loop] = task
(loop := asyncio.get_event_loop()).__class__._run_once = _run_once return future.result()
asyncio.run = loop.run_until_complete; loop.__class__._check_running = lambda _: 0 return wrapper
@sync
async def hello(): async def hello():
print('hello') @sync
async def world(): async def world(): print('hello')
print('world') world() or print('world')
asyncio.run(world()) hello()
asyncio.run(hello())
""" """
hello hello
world world