Use asyncio.timeout [s-z] (#98452)

This commit is contained in:
Marc Mueller
2023-08-15 14:34:18 +02:00
committed by GitHub
parent 71d985e4d6
commit 8b0fdd6fd2
38 changed files with 62 additions and 97 deletions

View File

@@ -11,7 +11,6 @@ from aiohttp import CookieJar
import aiounifi
from aiounifi.interfaces.api_handlers import ItemEvent
from aiounifi.websocket import WebsocketState
import async_timeout
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
@@ -375,7 +374,7 @@ class UniFiController:
async def async_reconnect(self) -> None:
"""Try to reconnect UniFi Network session."""
try:
async with async_timeout.timeout(5):
async with asyncio.timeout(5):
await self.api.login()
self.api.start_websocket()
@@ -444,7 +443,7 @@ async def get_unifi_controller(
)
try:
async with async_timeout.timeout(10):
async with asyncio.timeout(10):
await controller.check_unifi_os()
await controller.login()
return controller