mirror of
https://github.com/python/cpython.git
synced 2025-08-04 00:48:58 +00:00
Add asyncio.get_running_loop() function. (#4782)
This commit is contained in:
parent
3e9751819a
commit
abae67ebc2
4 changed files with 27 additions and 1 deletions
|
@ -25,6 +25,13 @@ the execution of the process.
|
|||
|
||||
Equivalent to calling ``get_event_loop_policy().new_event_loop()``.
|
||||
|
||||
.. function:: get_running_loop()
|
||||
|
||||
Return the running event loop in the current OS thread. If there
|
||||
is no running event loop a :exc:`RuntimeError` is raised.
|
||||
|
||||
.. versionadded:: 3.7
|
||||
|
||||
|
||||
.. _asyncio-event-loops:
|
||||
|
||||
|
|
|
@ -7,7 +7,8 @@ __all__ = (
|
|||
'get_event_loop_policy', 'set_event_loop_policy',
|
||||
'get_event_loop', 'set_event_loop', 'new_event_loop',
|
||||
'get_child_watcher', 'set_child_watcher',
|
||||
'_set_running_loop', '_get_running_loop',
|
||||
'_set_running_loop', 'get_running_loop',
|
||||
'_get_running_loop',
|
||||
)
|
||||
|
||||
import functools
|
||||
|
@ -646,6 +647,17 @@ class _RunningLoop(threading.local):
|
|||
_running_loop = _RunningLoop()
|
||||
|
||||
|
||||
def get_running_loop():
|
||||
"""Return the running event loop. Raise a RuntimeError if there is none.
|
||||
|
||||
This function is thread-specific.
|
||||
"""
|
||||
loop = _get_running_loop()
|
||||
if loop is None:
|
||||
raise RuntimeError('no running event loop')
|
||||
return loop
|
||||
|
||||
|
||||
def _get_running_loop():
|
||||
"""Return the running event loop or None.
|
||||
|
||||
|
|
|
@ -2733,10 +2733,13 @@ class PolicyTests(unittest.TestCase):
|
|||
try:
|
||||
asyncio.set_event_loop_policy(Policy())
|
||||
loop = asyncio.new_event_loop()
|
||||
with self.assertRaisesRegex(RuntimeError, 'no running'):
|
||||
self.assertIs(asyncio.get_running_loop(), None)
|
||||
self.assertIs(asyncio._get_running_loop(), None)
|
||||
|
||||
async def func():
|
||||
self.assertIs(asyncio.get_event_loop(), loop)
|
||||
self.assertIs(asyncio.get_running_loop(), loop)
|
||||
self.assertIs(asyncio._get_running_loop(), loop)
|
||||
|
||||
loop.run_until_complete(func())
|
||||
|
@ -2745,6 +2748,9 @@ class PolicyTests(unittest.TestCase):
|
|||
if loop is not None:
|
||||
loop.close()
|
||||
|
||||
with self.assertRaisesRegex(RuntimeError, 'no running'):
|
||||
self.assertIs(asyncio.get_running_loop(), None)
|
||||
|
||||
self.assertIs(asyncio._get_running_loop(), None)
|
||||
|
||||
|
||||
|
|
|
@ -0,0 +1 @@
|
|||
Add asyncio.get_running_loop() function.
|
Loading…
Add table
Add a link
Reference in a new issue