diff options
Diffstat (limited to 'Lib/asyncio/runners.py')
| -rw-r--r-- | Lib/asyncio/runners.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/Lib/asyncio/runners.py b/Lib/asyncio/runners.py index f524c3b..7489a50 100644 --- a/Lib/asyncio/runners.py +++ b/Lib/asyncio/runners.py @@ -52,6 +52,7 @@ class Runner: self._loop = None self._context = None self._interrupt_count = 0 + self._set_event_loop = False def __enter__(self): self._lazy_init() @@ -70,6 +71,8 @@ class Runner: loop.run_until_complete(loop.shutdown_asyncgens()) loop.run_until_complete(loop.shutdown_default_executor()) finally: + if self._set_event_loop: + events.set_event_loop(None) loop.close() self._loop = None self._state = _State.CLOSED @@ -111,6 +114,8 @@ class Runner: self._interrupt_count = 0 try: + if self._set_event_loop: + events.set_event_loop(self._loop) return self._loop.run_until_complete(task) except exceptions.CancelledError: if self._interrupt_count > 0 and task.uncancel() == 0: @@ -130,6 +135,7 @@ class Runner: return if self._loop_factory is None: self._loop = events.new_event_loop() + self._set_event_loop = True else: self._loop = self._loop_factory() if self._debug is not None: |