62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
""" Support for Tellstick switches. """
|
|
import logging
|
|
|
|
from homeassistant.helpers import ToggleDevice
|
|
from homeassistant.const import ATTR_FRIENDLY_NAME
|
|
|
|
try:
|
|
import tellcore.constants as tc_constants
|
|
except ImportError:
|
|
# Don't care for now. Warning will come when get_switches is called.
|
|
pass
|
|
|
|
|
|
# pylint: disable=unused-argument
|
|
def get_devices(hass, config):
|
|
""" Find and return Tellstick switches. """
|
|
try:
|
|
import tellcore.telldus as telldus
|
|
except ImportError:
|
|
logging.getLogger(__name__).exception(
|
|
"Failed to import tellcore")
|
|
return []
|
|
|
|
core = telldus.TelldusCore()
|
|
switches = core.devices()
|
|
|
|
return [TellstickSwitch(switch) for switch in switches]
|
|
|
|
|
|
class TellstickSwitch(ToggleDevice):
|
|
""" represents a Tellstick switch within home assistant. """
|
|
last_sent_command_mask = (tc_constants.TELLSTICK_TURNON |
|
|
tc_constants.TELLSTICK_TURNOFF)
|
|
|
|
def __init__(self, tellstick):
|
|
self.tellstick = tellstick
|
|
self.state_attr = {ATTR_FRIENDLY_NAME: tellstick.name}
|
|
|
|
def get_name(self):
|
|
""" Returns the name of the switch if any. """
|
|
return self.tellstick.name
|
|
|
|
# pylint: disable=unused-argument
|
|
def turn_on(self, **kwargs):
|
|
""" Turns the switch on. """
|
|
self.tellstick.turn_on()
|
|
|
|
# pylint: disable=unused-argument
|
|
def turn_off(self, **kwargs):
|
|
""" Turns the switch off. """
|
|
self.tellstick.turn_off()
|
|
|
|
def is_on(self):
|
|
""" True if switch is on. """
|
|
last_command = self.tellstick.last_sent_command(
|
|
self.last_sent_command_mask)
|
|
|
|
return last_command == tc_constants.TELLSTICK_TURNON
|
|
|
|
def get_state_attributes(self):
|
|
""" Returns optional state attributes. """
|
|
return self.state_attr
|