total_daily_energy: allow to disable restore mode (#2795)

This commit is contained in:
Adrián Panella
2021-11-25 15:35:36 -06:00
committed by GitHub
parent 2347e043a9
commit e7827a6997
3 changed files with 11 additions and 6 deletions
@@ -4,6 +4,7 @@ from esphome.components import sensor, time
from esphome.const import (
CONF_ICON,
CONF_ID,
CONF_RESTORE,
CONF_TIME_ID,
DEVICE_CLASS_ENERGY,
CONF_METHOD,
@@ -36,6 +37,7 @@ CONFIG_SCHEMA = (
cv.GenerateID(): cv.declare_id(TotalDailyEnergy),
cv.GenerateID(CONF_TIME_ID): cv.use_id(time.RealTimeClock),
cv.Required(CONF_POWER_ID): cv.use_id(sensor.Sensor),
cv.Optional(CONF_RESTORE, default=True): cv.boolean,
cv.Optional(
CONF_MIN_SAVE_INTERVAL, default="0s"
): cv.positive_time_period_milliseconds,
@@ -70,5 +72,6 @@ async def to_code(config):
cg.add(var.set_parent(sens))
time_ = await cg.get_variable(config[CONF_TIME_ID])
cg.add(var.set_time(time_))
cg.add(var.set_restore(config[CONF_RESTORE]))
cg.add(var.set_min_save_interval(config[CONF_MIN_SAVE_INTERVAL]))
cg.add(var.set_method(config[CONF_METHOD]))