Buffer JsonDecodeError in Flo (#109767)

This commit is contained in:
David F. Mulcahey
2024-02-05 18:39:56 -05:00
committed by Franck Nijhof
parent ffd5e04a29
commit 5025c15165

View File

@@ -7,6 +7,7 @@ from typing import Any
from aioflo.api import API from aioflo.api import API
from aioflo.errors import RequestError from aioflo.errors import RequestError
from orjson import JSONDecodeError
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@@ -46,7 +47,7 @@ class FloDeviceDataUpdateCoordinator(DataUpdateCoordinator): # pylint: disable=
await self._update_device() await self._update_device()
await self._update_consumption_data() await self._update_consumption_data()
self._failure_count = 0 self._failure_count = 0
except (RequestError, TimeoutError) as error: except (RequestError, TimeoutError, JSONDecodeError) as error:
self._failure_count += 1 self._failure_count += 1
if self._failure_count > 3: if self._failure_count > 3:
raise UpdateFailed(error) from error raise UpdateFailed(error) from error