jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-21 12:17:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small cleanup to IQVIA DataUpdateCoordinator logic (#42642)

Browse Source
This commit is contained in:
Aaron Bach 2020-10-30 14:10:32 -06:00 committed by GitHub
parent 2b24b9658f
commit 036a8b5667
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
homeassistant/components/iqvia/__init__.py
View File

@ -1,6 +1,7 @@
"""Support for IQVIA.""" """Support for IQVIA."""
import asyncio import asyncio
from datetime import timedelta from datetime import timedelta
from functools import partial
from pyiqvia import Client from pyiqvia import Client
from pyiqvia.errors import IQVIAError from pyiqvia.errors import IQVIAError
@ -77,7 +78,7 @@ async def async_setup_entry(hass, entry):
LOGGER, LOGGER,
name=f"{entry.data[CONF_ZIP_CODE]} {sensor_type}", name=f"{entry.data[CONF_ZIP_CODE]} {sensor_type}",
update_interval=DEFAULT_SCAN_INTERVAL, update_interval=DEFAULT_SCAN_INTERVAL,
update_method=lambda coro=api_coro: async_get_data_from_api(coro), update_method=partial(async_get_data_from_api, api_coro),
) )
init_data_update_tasks.append(coordinator.async_refresh()) init_data_update_tasks.append(coordinator.async_refresh())
@ -151,22 +152,22 @@ class IQVIAEntity(CoordinatorEntity):
"""Return the unit the value is expressed in.""" """Return the unit the value is expressed in."""
return "index" return "index"
@callback
def _handle_coordinator_update(self) -> None:
"""Handle updated data from the coordinator."""
self.update_from_latest_data()
self.async_write_ha_state()
async def async_added_to_hass(self): async def async_added_to_hass(self):
"""Register callbacks.""" """Register callbacks."""
await super().async_added_to_hass()
@callback
def update():
"""Update the state."""
self.update_from_latest_data()
self.async_write_ha_state()
self.async_on_remove(self.coordinator.async_add_listener(update))
if self._type == TYPE_ALLERGY_FORECAST: if self._type == TYPE_ALLERGY_FORECAST:
outlook_coordinator = self.hass.data[DOMAIN][DATA_COORDINATOR][ self.async_on_remove(
self._entry.entry_id self.hass.data[DOMAIN][DATA_COORDINATOR][self._entry.entry_id][
][TYPE_ALLERGY_OUTLOOK] TYPE_ALLERGY_OUTLOOK
self.async_on_remove(outlook_coordinator.async_add_listener(update)) ].async_add_listener(self._handle_coordinator_update)
)
self.update_from_latest_data() self.update_from_latest_data()