From 6ef70c85ee03f9b0666368745d687f7f101e32cc Mon Sep 17 00:00:00 2001 From: Maciej Bieniek Date: Sat, 9 Oct 2021 23:42:20 +0200 Subject: [PATCH] Add -9999 error fix back in Xiaomi Miio (#57399) --- homeassistant/components/xiaomi_miio/__init__.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/xiaomi_miio/__init__.py b/homeassistant/components/xiaomi_miio/__init__.py index 11fe4cc1337..840212d3dd6 100644 --- a/homeassistant/components/xiaomi_miio/__init__.py +++ b/homeassistant/components/xiaomi_miio/__init__.py @@ -146,12 +146,23 @@ def get_platforms(config_entry): def _async_update_data_default(hass, device): async def update(): """Fetch data from the device using async_add_executor_job.""" - try: + + async def _async_fetch_data(): + """Fetch data from the device.""" async with async_timeout.timeout(10): state = await hass.async_add_executor_job(device.status) _LOGGER.debug("Got new state: %s", state) return state + try: + return await _async_fetch_data() + except DeviceException as ex: + if getattr(ex, "code", None) != -9999: + raise UpdateFailed(ex) from ex + _LOGGER.info("Got exception while fetching the state, trying again: %s", ex) + # Try to fetch the data a second time after error code -9999 + try: + return await _async_fetch_data() except DeviceException as ex: raise UpdateFailed(ex) from ex