Use asyncio.timeout [s-z] (#98452)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user