Fetch the data a second time when -9999 error occurs in Xiaomi Miio integration (#56288)

pull/56366/head
Maciej Bieniek 2021-09-16 18:19:41 +02:00 committed by Paulus Schoutsen
parent f8d62866e5
commit 0a5948dc8b
1 changed files with 12 additions and 1 deletions

View File

@ -171,12 +171,23 @@ async def async_create_miio_device_and_coordinator(
async def async_update_data():
"""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