Update retrieve_hass.py

Feat - Added support to publish complete forecast data as attributes for load cost and prod price
davidusb-geek · May 31, 2023 · d500b92 · d500b92
1 parent add21ae
commit d500b92
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/emhass/retrieve_hass.py b/src/emhass/retrieve_hass.py
@@ -285,6 +285,12 @@ def post_data(self, data_df: pd.DataFrame, idx: int, entity_id: str,
         elif 'SOC' in entity_id:
             data = retrieve_hass.get_attr_data_dict(data_df, idx, entity_id, unit_of_measurement, 
                                                     friendly_name, "battery_scheduled_soc", state)
+        elif 'unit_load_cost' in entity_id:
+            data = retrieve_hass.get_attr_data_dict(data_df, idx, entity_id, unit_of_measurement, 
+                                                    friendly_name, "unit_load_cost_forecasts", state)
+        elif 'unit_prod_price' in entity_id:
+            data = retrieve_hass.get_attr_data_dict(data_df, idx, entity_id, unit_of_measurement, 
+                                                    friendly_name, "unit_load_cost_forecasts", state)
         elif from_mlforecaster:
             data = retrieve_hass.get_attr_data_dict(data_df, idx, entity_id, unit_of_measurement, 
                                                     friendly_name, "scheduled_forecast", state)