| # SPDX-License-Identifier: MIT | |
| # Copyright (c) 2019 Martijn Pieters | |
| # Licensed under the MIT license as detailed in LICENSE.txt | |
| # compatibility across Python versions | |
| import asyncio | |
| import sys | |
| if sys.version_info < (3, 8): # pragma: no cover | |
| wait_for = asyncio.wait_for | |
| else: | |
| from typing import Any, Awaitable, Coroutine, Generator, TypeVar, Union | |
| _T = TypeVar("_T") | |
| _FutureLike = Union["asyncio.Future[_T]", Generator[Any, None, _T], Awaitable[_T]] | |
| def wait_for( | |
| fut: _FutureLike[_T], *args: Any, loop: Any, **kwargs: Any | |
| ) -> Coroutine[Any, Any, _T]: | |
| # loop argument is deprecated, drop it automatically | |
| return asyncio.wait_for(fut, *args, **kwargs) | |