xmlstream: add a wrap() method for ensure_future

This commit is contained in:
mathieui 2021-02-14 11:36:01 +01:00
parent 5f9ab45a5e
commit 8da5310ea6

View file

@ -9,6 +9,7 @@
# :license: MIT, see LICENSE for more details
from typing import (
Any,
Coroutine,
Callable,
Iterable,
Iterator,
@ -1251,3 +1252,13 @@ class XMLStream(asyncio.BaseProtocol):
raise
finally:
self.del_event_handler(event, handler)
def wrap(self, coroutine: Coroutine[Any, Any, Any]) -> Future:
"""Make a Future out of a coroutine with the current loop.
:param coroutine: The coroutine to wrap.
"""
return asyncio.ensure_future(
coroutine,
loop=self.loop,
)