From 5c3af1d3f66c73c0a0ea01df55fc3a065340bb38 Mon Sep 17 00:00:00 2001
From: Jesse Hills <3060199+jesserockz@users.noreply.github.com>
Date: Wed, 12 May 2021 03:36:14 +1200
Subject: [PATCH] Allow duration for deep_sleep.enter to be templateable
 (#1765)

---
 esphome/components/deep_sleep/__init__.py | 4 +++-
 tests/test2.yaml                          | 2 ++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/esphome/components/deep_sleep/__init__.py b/esphome/components/deep_sleep/__init__.py
index 793d6b2ebb..c0d035987d 100644
--- a/esphome/components/deep_sleep/__init__.py
+++ b/esphome/components/deep_sleep/__init__.py
@@ -108,7 +108,9 @@ def to_code(config):
 DEEP_SLEEP_ENTER_SCHEMA = automation.maybe_simple_id(
     {
         cv.GenerateID(): cv.use_id(DeepSleepComponent),
-        cv.Optional(CONF_SLEEP_DURATION): cv.positive_time_period_milliseconds,
+        cv.Optional(CONF_SLEEP_DURATION): cv.templatable(
+            cv.positive_time_period_milliseconds
+        ),
     }
 )
 
diff --git a/tests/test2.yaml b/tests/test2.yaml
index 34724ee955..9b4db1477f 100644
--- a/tests/test2.yaml
+++ b/tests/test2.yaml
@@ -346,6 +346,8 @@ text_sensor:
       - homeassistant.tag_scanned: 1234-abcd
       - deep_sleep.enter:
           sleep_duration: 30min
+      - deep_sleep.enter:
+          sleep_duration: !lambda "return 30 * 60 * 1000;"
   - platform: template
     name: 'Template Text Sensor'
     lambda: |-