python 3.11 fixes

This commit is contained in:
Maxim Devaev
2023-05-04 12:37:05 +03:00
parent b5d6731497
commit b5353e63cd
8 changed files with 37 additions and 22 deletions

View File

@@ -202,7 +202,7 @@ async def wait_infinite() -> None:
await asyncio.sleep(3600)
async def wait_first(*aws: Awaitable) -> tuple[set[asyncio.Task], set[asyncio.Task]]:
async def wait_first(*aws: (asyncio.Future | asyncio.Task)) -> tuple[set[asyncio.Task], set[asyncio.Task]]:
return (await asyncio.wait(list(aws), return_when=asyncio.FIRST_COMPLETED))
@@ -242,7 +242,10 @@ class AioNotifier:
await self.__queue.get()
else:
try:
await asyncio.wait_for(self.__queue.get(), timeout=timeout)
await asyncio.wait_for(
asyncio.ensure_future(self.__queue.get()),
timeout=timeout,
)
except asyncio.TimeoutError:
return # False
while not self.__queue.empty():

View File

@@ -329,7 +329,10 @@ class Streamer: # pylint: disable=too-many-instance-attributes
if waiter_task is None:
waiter_task = asyncio.create_task(self.__notifier.wait())
if waiter_task in (await aiotools.wait_first(asyncio.sleep(self.__state_poll), waiter_task))[0]:
if waiter_task in (await aiotools.wait_first(
asyncio.ensure_future(asyncio.sleep(self.__state_poll)),
waiter_task,
))[0]:
waiter_task = None
# =====

View File

@@ -215,7 +215,10 @@ class Inotify:
async def get_event(self, timeout: float) -> (InotifyEvent | None):
assert timeout > 0
try:
return (await asyncio.wait_for(self.__events_queue.get(), timeout=timeout))
return (await asyncio.wait_for(
asyncio.ensure_future(self.__events_queue.get()),
timeout=timeout,
))
except asyncio.TimeoutError:
return None

View File

@@ -137,8 +137,14 @@ class Plugin(BaseUserGpioDriver): # pylint: disable=too-many-instance-attribute
assert self.__writer is not None
try:
self.__writer.write(b"\xAA\xBB\x03%s\xEE" % (cmd))
await asyncio.wait_for(self.__writer.drain(), timeout=self.__timeout)
return (await asyncio.wait_for(self.__reader.readexactly(6), timeout=self.__timeout))[4]
await asyncio.wait_for(
asyncio.ensure_future(self.__writer.drain()),
timeout=self.__timeout,
)
return (await asyncio.wait_for(
asyncio.ensure_future(self.__reader.readexactly(6)),
timeout=self.__timeout,
))[4]
except Exception as err:
get_logger(0).error("Can't send command to TESmart KVM [%s]:%d: %s",
self.__host, self.__port, tools.efmt(err))
@@ -155,7 +161,7 @@ class Plugin(BaseUserGpioDriver): # pylint: disable=too-many-instance-attribute
async def __ensure_device_net(self) -> None:
try:
(self.__reader, self.__writer) = await asyncio.wait_for(
asyncio.open_connection(self.__host, self.__port),
asyncio.ensure_future(asyncio.open_connection(self.__host, self.__port)),
timeout=self.__timeout,
)
except Exception as err: