Support for hierarchy of individual energy devices (#132616)

* Support for hierarchy of individual energy devices

* update DeviceConsumption dict

* change name parent to 'included_in'

* Break comment

---------

Co-authored-by: Martin Hjelmare <[email protected]>

Author: karwosts

homeassistant/components/energy/data.py

@@ -139,6 +139,10 @@ class DeviceConsumption(TypedDict):
     # An optional custom name for display in energy graphs
     name: str | None
 
+    # An optional statistic_id identifying a device
+    # that includes this device's consumption in its total
+    included_in_stat: str | None
+
 
 class EnergyPreferences(TypedDict):
     """Dictionary holding the energy data."""
@@ -291,6 +295,7 @@ def check_type_limits(value: list[SourceType]) -> list[SourceType]:
     {
         vol.Required("stat_consumption"): str,
         vol.Optional("name"): str,
+        vol.Optional("included_in_stat"): str,
     }
 )
 

tests/components/energy/test_websocket_api.py

@@ -149,7 +149,13 @@ async def test_save_preferences(
                 "stat_energy_to": "my_battery_charging",
             },
         ],
-        "device_consumption": [{"stat_consumption": "some_device_usage"}],
+        "device_consumption": [
+            {
+                "stat_consumption": "some_device_usage",
+                "name": "My Device",
+                "included_in_stat": "sensor.some_other_device",
+            }
+        ],
     }
 
     await client.send_json({"id": 6, "type": "energy/save_prefs", **new_prefs})