evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve rainbird error handling (#98239)

Browse source
This commit is contained in:
Allen Porter 2023-08-14 04:32:08 -07:00 • committed by GitHub
parent 6f97270cd2
commit 9ddf11f6cd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
7 changed files with 170 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

10
homeassistant/components/rainbird/coordinator.py
View file

@ -8,7 +8,11 @@ import logging
from typing import TypeVar
import async_timeout
from pyrainbird.async_client import AsyncRainbirdController, RainbirdApiException
from pyrainbird.async_client import (
AsyncRainbirdController,
RainbirdApiException,
RainbirdDeviceBusyException,
)
from pyrainbird.data import ModelAndVersion
from homeassistant.core import HomeAssistant
@ -84,8 +88,10 @@ class RainbirdUpdateCoordinator(DataUpdateCoordinator[RainbirdDeviceState]):
try:
async with async_timeout.timeout(TIMEOUT_SECONDS):
return await self._fetch_data()
except RainbirdDeviceBusyException as err:
raise UpdateFailed("Rain Bird device is busy") from err
except RainbirdApiException as err:
raise UpdateFailed(f"Error communicating with Device: {err}") from err
raise UpdateFailed("Rain Bird device failure") from err
async def _fetch_data(self) -> RainbirdDeviceState:
"""Fetch data from the Rain Bird device.