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

Update sensor.py

Browse Source
This commit is contained in:
balgerion
2025-04-29 16:37:20 +02:00
committed by GitHub
parent 483846cf9b
commit af052853bf
1 changed files with 19 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
custom_components/pstryk/sensor.py
View File

@ -1,3 +1,4 @@
"""Sensor platform for Pstryk Energy integration."""
import logging import logging
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@ -19,11 +20,22 @@ async def async_setup_entry(
buy_top = entry.options.get("buy_top", entry.data.get("buy_top", 5)) buy_top = entry.options.get("buy_top", entry.data.get("buy_top", 5))
sell_top = entry.options.get("sell_top", entry.data.get("sell_top", 5)) sell_top = entry.options.get("sell_top", entry.data.get("sell_top", 5))
# Cleanup old coordinators if they exist
for price_type in ("buy", "sell"):
key = f"{entry.entry_id}_{price_type}"
coordinator = hass.data[DOMAIN].get(key)
if coordinator:
# Cancel scheduled updates
if hasattr(coordinator, '_unsub_hourly') and coordinator._unsub_hourly:
coordinator._unsub_hourly()
if hasattr(coordinator, '_unsub_midnight') and coordinator._unsub_midnight:
coordinator._unsub_midnight()
# Remove from hass data
hass.data[DOMAIN].pop(key, None)
entities = [] entities = []
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: PstrykDataUpdateCoordinator = hass.data[DOMAIN].get(key)
if not coordinator:
coordinator = PstrykDataUpdateCoordinator(hass, api_key, price_type) coordinator = PstrykDataUpdateCoordinator(hass, api_key, price_type)
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()
coordinator.schedule_hourly_update() coordinator.schedule_hourly_update()