Implement PlatformNotReady to Linky + fix TypeError (#31768)

This commit is contained in:
Quentame
2020-02-13 01:12:35 +01:00
committed by GitHub
parent be388a4797
commit e0a035ce35

View File

@@ -13,6 +13,7 @@ from homeassistant.const import (
CONF_USERNAME,
ENERGY_KILO_WATT_HOUR,
)
from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.event import async_track_time_interval
from homeassistant.helpers.typing import HomeAssistantType
@@ -73,6 +74,7 @@ class LinkyAccount:
_LOGGER.debug(json.dumps(self._data, indent=2))
except PyLinkyException as exp:
_LOGGER.error(exp)
raise PlatformNotReady
finally:
client.close_session()
@@ -146,6 +148,9 @@ class LinkySensor(Entity):
async def async_update(self) -> None:
"""Retrieve the new data for the sensor."""
if self._account.data is None:
return
data = self._account.data[self._scale][self._when]
self._consumption = data[CONSUMPTION]
self._time = data[TIME]