From 1726c4237bf03d1f419c8a1443b4024d8c75e9a6 Mon Sep 17 00:00:00 2001
From: Otto Winter <otto@otto-winter.com>
Date: Sat, 1 Dec 2018 13:19:34 +0100
Subject: [PATCH] CSE7766 update interval (#250)

* CSE7766 update interval

* PollingComponent
---
 esphomeyaml/components/sensor/cse7766.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/esphomeyaml/components/sensor/cse7766.py b/esphomeyaml/components/sensor/cse7766.py
index 091c146632..f956819d8c 100644
--- a/esphomeyaml/components/sensor/cse7766.py
+++ b/esphomeyaml/components/sensor/cse7766.py
@@ -4,12 +4,12 @@ from esphomeyaml.components import sensor, uart
 from esphomeyaml.components.uart import UARTComponent
 import esphomeyaml.config_validation as cv
 from esphomeyaml.const import CONF_CURRENT, CONF_ID, CONF_NAME, CONF_POWER, CONF_UART_ID, \
-    CONF_VOLTAGE
-from esphomeyaml.helpers import App, Pvariable, get_variable, setup_component, Component
+    CONF_UPDATE_INTERVAL, CONF_VOLTAGE
+from esphomeyaml.helpers import App, PollingComponent, Pvariable, get_variable, setup_component
 
 DEPENDENCIES = ['uart']
 
-CSE7766Component = sensor.sensor_ns.class_('CSE7766Component', Component, uart.UARTDevice)
+CSE7766Component = sensor.sensor_ns.class_('CSE7766Component', PollingComponent, uart.UARTDevice)
 CSE7766VoltageSensor = sensor.sensor_ns.class_('CSE7766VoltageSensor',
                                                sensor.EmptySensor)
 CSE7766CurrentSensor = sensor.sensor_ns.class_('CSE7766CurrentSensor',
@@ -30,6 +30,7 @@ PLATFORM_SCHEMA = vol.All(sensor.PLATFORM_SCHEMA.extend({
     vol.Optional(CONF_POWER): cv.nameable(sensor.SENSOR_SCHEMA.extend({
         cv.GenerateID(): cv.declare_variable_id(CSE7766PowerSensor),
     })),
+    vol.Optional(CONF_UPDATE_INTERVAL): cv.update_interval,
 }).extend(cv.COMPONENT_SCHEMA.schema), cv.has_at_least_one_key(CONF_VOLTAGE, CONF_CURRENT,
                                                                CONF_POWER))
 
@@ -38,7 +39,7 @@ def to_code(config):
     for uart_ in get_variable(config[CONF_UART_ID]):
         yield
 
-    rhs = App.make_cse7766(uart_)
+    rhs = App.make_cse7766(uart_, config.get(CONF_UPDATE_INTERVAL))
     cse = Pvariable(config[CONF_ID], rhs)
 
     if CONF_VOLTAGE in config: