Merge pull request #84 from davidusb-geek/davidusb-geek-patch-2

Update retrieve_hass.py
davidusb-geek · May 31, 2023 · 5f9c8d2 · 5f9c8d2
2 parents add21ae + d500b92
commit 5f9c8d2
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)