jdejaegh/irm-kmi-ha
1
0
Fork 0
mirror of https://github.com/jdejaegh/irm-kmi-ha.git synced 2025-06-27 03:35:56 +02:00
Code Issues Projects Releases Packages Wiki Activity

Put config entry in coordinator

Browse source
This commit is contained in:
Jules 2024-12-29 18:04:24 +01:00
parent fe412dfec3
commit 91d46dcb6c
Signed by: jdejaegh
GPG key ID: 99D6D184CA66933A
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
custom_components/irm_kmi/coordinator.py
View file

@ -43,6 +43,7 @@ class IrmKmiCoordinator(TimestampDataUpdateCoordinator):
super().__init__( super().__init__(
hass, hass,
_LOGGER, _LOGGER,
config_entry=entry,
# Name of the data. For logging purposes. # Name of the data. For logging purposes.
name="IRM KMI weather", name="IRM KMI weather",
# Polling interval. Will only be polled if there are subscribers. # Polling interval. Will only be polled if there are subscribers.
@ -52,11 +53,10 @@ class IrmKmiCoordinator(TimestampDataUpdateCoordinator):
self._zone = get_config_value(entry, CONF_ZONE) self._zone = get_config_value(entry, CONF_ZONE)
self._dark_mode = get_config_value(entry, CONF_DARK_MODE) self._dark_mode = get_config_value(entry, CONF_DARK_MODE)
self._style = get_config_value(entry, CONF_STYLE) self._style = get_config_value(entry, CONF_STYLE)
self._config_entry = entry
self.shared_device_info = DeviceInfo( self.shared_device_info = DeviceInfo(
entry_type=DeviceEntryType.SERVICE, entry_type=DeviceEntryType.SERVICE,
identifiers={(DOMAIN, entry.entry_id)}, identifiers={(DOMAIN, entry.entry_id)},
manufacturer=IRM_KMI_NAME.get(preferred_language(self.hass, self._config_entry)), manufacturer=IRM_KMI_NAME.get(preferred_language(self.hass, self.config_entry)),
name=f"{entry.title}" name=f"{entry.title}"
) )
@ -92,7 +92,7 @@ class IrmKmiCoordinator(TimestampDataUpdateCoordinator):
_LOGGER.error(f"The zone {self._zone} is now out of Benelux and forecast is only available in Benelux." _LOGGER.error(f"The zone {self._zone} is now out of Benelux and forecast is only available in Benelux."
f"Associated device is now disabled. Move the zone back in Benelux and re-enable to fix " f"Associated device is now disabled. Move the zone back in Benelux and re-enable to fix "
f"this") f"this")
disable_from_config(self.hass, self._config_entry) disable_from_config(self.hass, self.config_entry)
issue_registry.async_create_issue( issue_registry.async_create_issue(
self.hass, self.hass,
@ -101,7 +101,7 @@ class IrmKmiCoordinator(TimestampDataUpdateCoordinator):
is_fixable=True, is_fixable=True,
severity=issue_registry.IssueSeverity.ERROR, severity=issue_registry.IssueSeverity.ERROR,
translation_key='zone_moved', translation_key='zone_moved',
data={'config_entry_id': self._config_entry.entry_id, 'zone': self._zone}, data={'config_entry_id': self.config_entry.entry_id, 'zone': self._zone},
translation_placeholders={'zone': self._zone} translation_placeholders={'zone': self._zone}
) )
return ProcessedCoordinatorData() return ProcessedCoordinatorData()
@ -131,7 +131,7 @@ class IrmKmiCoordinator(TimestampDataUpdateCoordinator):
localisation = images_from_api[0] localisation = images_from_api[0]
images_from_api = images_from_api[1:] images_from_api = images_from_api[1:]
lang = preferred_language(self.hass, self._config_entry) lang = preferred_language(self.hass, self.config_entry)
radar_animation = RadarAnimationData( radar_animation = RadarAnimationData(
hint=api_data.get('animation', {}).get('sequenceHint', {}).get(lang), hint=api_data.get('animation', {}).get('sequenceHint', {}).get(lang),
unit=api_data.get('animation', {}).get('unit', {}).get(lang), unit=api_data.get('animation', {}).get('unit', {}).get(lang),
@ -367,7 +367,7 @@ class IrmKmiCoordinator(TimestampDataUpdateCoordinator):
return None return None
forecasts = list() forecasts = list()
lang = preferred_language(self.hass, self._config_entry) lang = preferred_language(self.hass, self.config_entry)
tz = await dt.async_get_time_zone('Europe/Brussels') tz = await dt.async_get_time_zone('Europe/Brussels')
forecast_day = dt.now(tz) forecast_day = dt.now(tz)
@ -502,7 +502,7 @@ class IrmKmiCoordinator(TimestampDataUpdateCoordinator):
if warning_data is None or not isinstance(warning_data, list) or len(warning_data) == 0: if warning_data is None or not isinstance(warning_data, list) or len(warning_data) == 0:
return [] return []
lang = preferred_language(self.hass, self._config_entry) lang = preferred_language(self.hass, self.config_entry)
result = list() result = list()
for data in warning_data: for data in warning_data:
try: try: