diff --git a/homeassistant/components/switch/transmission.py b/homeassistant/components/switch/transmission.py index 43d3adf1eee..b3fb04dbd82 100644 --- a/homeassistant/components/switch/transmission.py +++ b/homeassistant/components/switch/transmission.py @@ -92,7 +92,6 @@ class TransmissionSwitch(ToggleEntity): def __init__(self, transmission_client, name): self._name = name self.transmission_client = transmission_client - self.turtle_mode_active = False self._state = STATE_OFF @property @@ -118,24 +117,19 @@ class TransmissionSwitch(ToggleEntity): """ Turn the device on. """ _LOGGING.info("Turning on Turtle Mode") - self.toggle_turtle_mode() + self.transmission_client.set_session( + alt_speed_enabled=True) def turn_off(self, **kwargs): """ Turn the device off. """ _LOGGING.info("Turning off Turtle Mode ") - self.toggle_turtle_mode() - - def toggle_turtle_mode(self): - """ Toggle turtle mode. """ - self.transmission_client.set_session( - alt_speed_enabled=not self.turtle_mode_active) - self.update() + alt_speed_enabled=False) def update(self): """ Gets the latest data from Transmission and updates the state. """ - self.turtle_mode_active = self.transmission_client.get_session( + active = self.transmission_client.get_session( ).alt_speed_enabled - self._state = STATE_ON if self.turtle_mode_active else STATE_OFF + self._state = STATE_ON if active else STATE_OFF