Add new power utility (#78867)

* Add power utility

* Fix tests
This commit is contained in:
epenet 2022-09-21 14:48:38 +02:00 committed by GitHub
parent 664a576113
commit d7382aadfe
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 102 additions and 20 deletions

View file

@ -13,17 +13,18 @@ from homeassistant.const import (
ENERGY_KILO_WATT_HOUR,
ENERGY_MEGA_WATT_HOUR,
ENERGY_WATT_HOUR,
POWER_KILO_WATT,
POWER_WATT,
VOLUME_CUBIC_FEET,
VOLUME_CUBIC_METERS,
)
from homeassistant.core import HomeAssistant, callback, valid_entity_id
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.json import JSON_DUMP
from homeassistant.util import dt as dt_util
import homeassistant.util.pressure as pressure_util
import homeassistant.util.temperature as temperature_util
from homeassistant.util import (
dt as dt_util,
power as power_util,
pressure as pressure_util,
temperature as temperature_util,
)
from .const import MAX_QUEUE_BACKLOG
from .statistics import (
@ -122,7 +123,7 @@ async def ws_handle_get_statistics_during_period(
vol.Optional("energy"): vol.Any(
ENERGY_WATT_HOUR, ENERGY_KILO_WATT_HOUR, ENERGY_MEGA_WATT_HOUR
),
vol.Optional("power"): vol.Any(POWER_WATT, POWER_KILO_WATT),
vol.Optional("power"): vol.In(power_util.VALID_UNITS),
vol.Optional("pressure"): vol.In(pressure_util.VALID_UNITS),
vol.Optional("temperature"): vol.In(temperature_util.VALID_UNITS),
vol.Optional("volume"): vol.Any(VOLUME_CUBIC_FEET, VOLUME_CUBIC_METERS),