jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 11:17:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add battery support to energy (#54432)

Browse Source
This commit is contained in:
Bram Kragten 2021-08-11 18:49:56 +02:00 committed by GitHub
parent 98a4e6a7e8
commit f020d65416
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
homeassistant/components/energy/data.py
View File

@ -79,7 +79,16 @@ class SolarSourceType(TypedDict):
config_entry_solar_forecast: list[str] | None
SourceType = Union[GridSourceType, SolarSourceType]
class BatterySourceType(TypedDict):
"""Dictionary holding the source of battery storage."""
type: Literal["battery"]
stat_energy_from: str
stat_energy_to: str
SourceType = Union[GridSourceType, SolarSourceType, BatterySourceType]
class DeviceConsumption(TypedDict):
@ -177,6 +186,13 @@ SOLAR_SOURCE_SCHEMA = vol.Schema(
vol.Optional("config_entry_solar_forecast"): vol.Any([str], None),
}
)
BATTERY_SOURCE_SCHEMA = vol.Schema(
{
vol.Required("type"): "battery",
vol.Required("stat_energy_from"): str,
vol.Required("stat_energy_to"): str,
}
)
def check_type_limits(value: list[SourceType]) -> list[SourceType]:
@ -197,6 +213,7 @@ ENERGY_SOURCE_SCHEMA = vol.All(
{
"grid": GRID_SOURCE_SCHEMA,
"solar": SOLAR_SOURCE_SCHEMA,
"battery": BATTERY_SOURCE_SCHEMA,
},
)
]

5
tests/components/energy/test_websocket_api.py
View File

@ -104,6 +104,11 @@ async def test_save_preferences(hass, hass_ws_client, hass_storage) -> None:
"stat_energy_from": "my_solar_production",
"config_entry_solar_forecast": ["predicted_config_entry"],
},
{
"type": "battery",
"stat_energy_from": "my_battery_draining",
"stat_energy_to": "my_battery_charging",
},
],
"device_consumption": [{"stat_consumption": "some_device_usage"}],
}