Accept transition for light.toggle (#8466)

This commit is contained in:
Anders Melchiorsen 2017-07-14 05:04:23 +02:00 committed by Paulus Schoutsen
parent 5829cdfdf1
commit 87b83f3602

View File

@ -378,18 +378,18 @@ class ToggleEntity(Entity):
return self.hass.async_add_job( return self.hass.async_add_job(
ft.partial(self.turn_off, **kwargs)) ft.partial(self.turn_off, **kwargs))
def toggle(self) -> None: def toggle(self, **kwargs) -> None:
"""Toggle the entity.""" """Toggle the entity."""
if self.is_on: if self.is_on:
self.turn_off() self.turn_off(**kwargs)
else: else:
self.turn_on() self.turn_on(**kwargs)
def async_toggle(self): def async_toggle(self, **kwargs):
"""Toggle the entity. """Toggle the entity.
This method must be run in the event loop and returns a coroutine. This method must be run in the event loop and returns a coroutine.
""" """
if self.is_on: if self.is_on:
return self.async_turn_off() return self.async_turn_off(**kwargs)
return self.async_turn_on() return self.async_turn_on(**kwargs)