ditus/ha_pstryk
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
balgerion
2025-04-30 14:38:12 +02:00
committed by GitHub
parent c68df1c21f
commit 332da37f85
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
custom_components/pstryk/__init__.py
View File

@ -23,17 +23,27 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
_LOGGER.debug("Pstryk entry setup: %s", entry.entry_id) _LOGGER.debug("Pstryk entry setup: %s", entry.entry_id)
return True return True
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def _cleanup_coordinators(hass: HomeAssistant, entry: ConfigEntry) -> None:
"""Unload sensor platform and clear data.""" """Clean up coordinators and cancel scheduled tasks."""
# First cancel coordinators' scheduled updates
for price_type in ("buy", "sell"): for price_type in ("buy", "sell"):
key = f"{entry.entry_id}_{price_type}" key = f"{entry.entry_id}_{price_type}"
coordinator = hass.data[DOMAIN].get(key) coordinator = hass.data[DOMAIN].get(key)
if coordinator: if coordinator:
_LOGGER.debug("Cleaning up %s coordinator for entry %s", price_type, entry.entry_id)
# Cancel scheduled updates
if hasattr(coordinator, '_unsub_hourly') and coordinator._unsub_hourly: if hasattr(coordinator, '_unsub_hourly') and coordinator._unsub_hourly:
coordinator._unsub_hourly() coordinator._unsub_hourly()
coordinator._unsub_hourly = None
if hasattr(coordinator, '_unsub_midnight') and coordinator._unsub_midnight: if hasattr(coordinator, '_unsub_midnight') and coordinator._unsub_midnight:
coordinator._unsub_midnight() coordinator._unsub_midnight()
coordinator._unsub_midnight = None
# Remove from hass data
hass.data[DOMAIN].pop(key, None)
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload sensor platform and clear data."""
# First cancel coordinators' scheduled updates
await _cleanup_coordinators(hass, entry)
# Then unload the platform # Then unload the platform
unload_ok = await hass.config_entries.async_forward_entry_unload(entry, "sensor") unload_ok = await hass.config_entries.async_forward_entry_unload(entry, "sensor")
@ -43,7 +53,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
if entry.entry_id in hass.data[DOMAIN]: if entry.entry_id in hass.data[DOMAIN]:
hass.data[DOMAIN].pop(entry.entry_id) hass.data[DOMAIN].pop(entry.entry_id)
# Clean up any coordinators # Clean up any remaining components
for key in list(hass.data[DOMAIN].keys()): for key in list(hass.data[DOMAIN].keys()):
if key.startswith(f"{entry.entry_id}_"): if key.startswith(f"{entry.entry_id}_"):
hass.data[DOMAIN].pop(key, None) hass.data[DOMAIN].pop(key, None)