home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-05 21:55:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

add-template: use yield from

Browse Source
This commit is contained in:
Giang Vo
2017-11-05 19:13:07 +11:00
parent 494b7a4403
commit 1e053714a7
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
homeassistant/components/light/template.py
View File

@@ -21,7 +21,6 @@ from homeassistant.helpers.config_validation import PLATFORM_SCHEMA
from homeassistant.exceptions import TemplateError
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity import async_generate_entity_id
from homeassistant.helpers.event import async_track_state_change
from homeassistant.helpers.script import Script
_LOGGER = logging.getLogger(__name__)
@@ -194,7 +193,7 @@ class LightTemplate(Light):
"""Update template on startup."""
if (self._template is not None or
self._level_template is not None):
async_track_state_change(
self.hass.helpers.event.async_track_state_change(
self.hass, self._entities, template_light_state_listener)
self.async_schedule_update_ha_state(True)
@@ -218,10 +217,10 @@ class LightTemplate(Light):
optimistic_set = True
if ATTR_BRIGHTNESS in kwargs and self._level_script:
self.hass.async_add_job(self._level_script.async_run(
self.hass.async_add_job(yield from self._level_script.async_run(
{"brightness": kwargs[ATTR_BRIGHTNESS]}))
else:
self.hass.async_add_job(self._on_script.async_run())
self.hass.async_add_job(yield from self._on_script.async_run())
if optimistic_set:
self.async_schedule_update_ha_state()
@@ -229,7 +228,7 @@ class LightTemplate(Light):
@asyncio.coroutine
def async_turn_off(self, **kwargs):
"""Turn the light off."""
self.hass.async_add_job(self._off_script.async_run())
self.hass.async_add_job(yield from self._off_script.async_run())
if self._template is None:
self._state = False
self.async_schedule_update_ha_state()