sascha/devlib
1
0
Fork 0
mirror of https://github.com/ARM-software/devlib.git synced 2025-01-30 17:50:46 +00:00
Code Issues Releases Wiki Activity

utils/asyn: Factor out the calls to asyncio.run

Browse Source 
Prepare for providing our own implementation of asyncio.run() to work
without nest_asyncio package.
This commit is contained in:
Douglas Raillard 2024-04-26 13:41:40 +01:00 committed by Marc Bonnici
parent 165b87f248
commit b2e19d333b
2 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
devlib/module/cgroups.py
View File

@ -19,13 +19,12 @@ from collections import namedtuple
from shlex import quote
import itertools
import warnings
import asyncio
from devlib.module import Module
from devlib.exception import TargetStableError
from devlib.utils.misc import list_to_ranges, isiterable
from devlib.utils.types import boolean
from devlib.utils.asyn import asyncf
from devlib.utils.asyn import asyncf, run
class Controller(object):
@ -410,7 +409,7 @@ class CgroupsModule(Module):
controller.mount_point)
self.controllers[ss.name] = controller
asyncio.run(
run(
target.async_manager.map_concurrently(
register_controller,
subsys,

16
devlib/utils/asyn.py
View File

@ -292,6 +292,14 @@ class memoized_method:
self._name = name
def run(coro):
"""
Similar to :func:`asyncio.run` but can be called while an event loop is
running.
"""
return asyncio.run(coro)
def asyncf(f):
"""
Decorator used to turn a coroutine into a blocking function, with an
@ -328,14 +336,14 @@ def asyncf(f):
asyncgen = x.__aiter__()
while True:
try:
yield asyncio.run(asyncgen.__anext__())
yield run(asyncgen.__anext__())
except StopAsyncIteration:
return
return genf()
else:
return await x
return asyncio.run(wrapper())
return run(wrapper())
return _AsyncPolymorphicFunction(
asyn=f,
@ -358,10 +366,10 @@ class _AsyncPolymorphicCM:
return self.cm.__aexit__(*args, **kwargs)
def __enter__(self, *args, **kwargs):
return asyncio.run(self.cm.__aenter__(*args, **kwargs))
return run(self.cm.__aenter__(*args, **kwargs))
def __exit__(self, *args, **kwargs):
return asyncio.run(self.cm.__aexit__(*args, **kwargs))
return run(self.cm.__aexit__(*args, **kwargs))
def asynccontextmanager(f):