Buffer JsonDecodeError in Flo (#109767)

This commit is contained in:
David F. Mulcahey 2024-02-05 18:39:56 -05:00 committed by GitHub
parent b7284b92ac
commit 13bc018e23
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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