1 files changed, 59 insertions, 0 deletions
diff --git a/components/template/sensor/__init__.py b/components/template/sensor/__init__.py
new file mode 100644
index 0000000..75fb505
--- /dev/null
+++ b/components/template/sensor/__init__.py
@@ -0,0 +1,59 @@
+import esphome.codegen as cg
+import esphome.config_validation as cv
+from esphome import automation
+from esphome.components import sensor
+from esphome.const import (
+    CONF_ID,
+    CONF_LAMBDA,
+    CONF_STATE,
+    STATE_CLASS_NONE,
+)
+from .. import template_ns
+
+TemplateSensor = template_ns.class_(
+    "TemplateSensor", sensor.Sensor, cg.PollingComponent
+)
+
+CONFIG_SCHEMA = (
+    sensor.sensor_schema(
+        accuracy_decimals=1,
+        state_class=STATE_CLASS_NONE,
+    )
+    .extend(
+        {
+            cv.GenerateID(): cv.declare_id(TemplateSensor),
+            cv.Optional(CONF_LAMBDA): cv.returning_lambda,
+        }
+    )
+    .extend(cv.polling_component_schema("60s"))
+)
+
+
+async def to_code(config):
+    var = cg.new_Pvariable(config[CONF_ID])
+    await cg.register_component(var, config)
+    await sensor.register_sensor(var, config)
+
+    if CONF_LAMBDA in config:
+        template_ = await cg.process_lambda(
+            config[CONF_LAMBDA], [], return_type=cg.optional.template(float)
+        )
+        cg.add(var.set_template(template_))
+
+
+[email protected]_action(
+    "sensor.template.publish",
+    sensor.SensorPublishAction,
+    cv.Schema(
+        {
+            cv.Required(CONF_ID): cv.use_id(sensor.Sensor),
+            cv.Required(CONF_STATE): cv.templatable(cv.float_),
+        }
+    ),
+)
+async def sensor_template_publish_to_code(config, action_id, template_arg, args):
+    paren = await cg.get_variable(config[CONF_ID])
+    var = cg.new_Pvariable(action_id, template_arg, paren)
+    template_ = await cg.templatable(config[CONF_STATE], args, float)
+    cg.add(var.set_state(template_))
+    return var