feat: Added data source attribute to sensors

Author: BottlecapDave

custom_components/target_timeframes/entities/rolling_target_timeframe.py

@@ -68,6 +68,7 @@ def __init__(self, hass: HomeAssistant, data_source_id: str, config_entry, confi
     self._attributes = self._config.copy()
     self._last_evaluated = None
     self._data_source_id = data_source_id
+    self._attributes["data_source_id"] = self._data_source_id
 
     is_rolling_target = True
     if CONFIG_TARGET_ROLLING_TARGET in self._config:
@@ -205,6 +206,7 @@ async def async_update(self):
     self._attributes["next_average_value"] = active_result["next_average_value"]
     self._attributes["next_min_value"] = active_result["next_min_value"]
     self._attributes["next_max_value"] = active_result["next_max_value"]
+    self._attributes["data_source_id"] = self._data_source_id
 
     self._state = active_result["is_active"]
 

custom_components/target_timeframes/entities/target_timeframe.py

@@ -73,6 +73,7 @@ def __init__(self, hass: HomeAssistant, data_source_id: str, config_entry, confi
     self._attributes = self._config.copy()
     self._last_evaluated = None
     self._data_source_id = data_source_id
+    self._attributes["data_source_id"] = self._data_source_id
 
     is_rolling_target = True
     if CONFIG_TARGET_ROLLING_TARGET in self._config:
@@ -86,6 +87,7 @@ def __init__(self, hass: HomeAssistant, data_source_id: str, config_entry, confi
 
     self._data_source_data = initial_data if initial_data is not None else []
     self._target_timeframes = []
+
 
     self._hass = hass
     self.entity_id = generate_entity_id("binary_sensor.{}", self.unique_id, hass=hass)
@@ -230,6 +232,7 @@ async def async_update(self):
     self._attributes["next_average_value"] = active_result["next_average_value"]
     self._attributes["next_min_value"] = active_result["next_min_value"]
     self._attributes["next_max_value"] = active_result["next_max_value"]
+    self._attributes["data_source_id"] = self._data_source_id
 
     self._state = active_result["is_active"]