Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Implement PlatformNotReady to Linky + fix TypeError (#31768)

pull/31779/head
Quentame 2020-02-13 01:12:35 +01:00 committed by GitHub
parent be388a4797
commit e0a035ce35
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
homeassistant/components/linky/sensor.py
View File

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