146 lines
5.3 KiB
Python
146 lines
5.3 KiB
Python
"""Support for MySensors switches."""
|
|
import voluptuous as vol
|
|
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.components import mysensors
|
|
from homeassistant.components.switch import DOMAIN, SwitchDevice
|
|
from homeassistant.const import ATTR_ENTITY_ID, STATE_OFF, STATE_ON
|
|
|
|
ATTR_IR_CODE = 'V_IR_SEND'
|
|
SERVICE_SEND_IR_CODE = 'mysensors_send_ir_code'
|
|
|
|
SEND_IR_CODE_SERVICE_SCHEMA = vol.Schema({
|
|
vol.Optional(ATTR_ENTITY_ID): cv.entity_ids,
|
|
vol.Required(ATTR_IR_CODE): cv.string,
|
|
})
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass, config, async_add_entities, discovery_info=None):
|
|
"""Set up the mysensors platform for switches."""
|
|
device_class_map = {
|
|
'S_DOOR': MySensorsSwitch,
|
|
'S_MOTION': MySensorsSwitch,
|
|
'S_SMOKE': MySensorsSwitch,
|
|
'S_LIGHT': MySensorsSwitch,
|
|
'S_LOCK': MySensorsSwitch,
|
|
'S_IR': MySensorsIRSwitch,
|
|
'S_BINARY': MySensorsSwitch,
|
|
'S_SPRINKLER': MySensorsSwitch,
|
|
'S_WATER_LEAK': MySensorsSwitch,
|
|
'S_SOUND': MySensorsSwitch,
|
|
'S_VIBRATION': MySensorsSwitch,
|
|
'S_MOISTURE': MySensorsSwitch,
|
|
'S_WATER_QUALITY': MySensorsSwitch,
|
|
}
|
|
mysensors.setup_mysensors_platform(
|
|
hass, DOMAIN, discovery_info, device_class_map,
|
|
async_add_entities=async_add_entities)
|
|
|
|
async def async_send_ir_code_service(service):
|
|
"""Set IR code as device state attribute."""
|
|
entity_ids = service.data.get(ATTR_ENTITY_ID)
|
|
ir_code = service.data.get(ATTR_IR_CODE)
|
|
devices = mysensors.get_mysensors_devices(hass, DOMAIN)
|
|
|
|
if entity_ids:
|
|
_devices = [device for device in devices.values()
|
|
if isinstance(device, MySensorsIRSwitch) and
|
|
device.entity_id in entity_ids]
|
|
else:
|
|
_devices = [device for device in devices.values()
|
|
if isinstance(device, MySensorsIRSwitch)]
|
|
|
|
kwargs = {ATTR_IR_CODE: ir_code}
|
|
for device in _devices:
|
|
await device.async_turn_on(**kwargs)
|
|
|
|
hass.services.async_register(
|
|
DOMAIN, SERVICE_SEND_IR_CODE, async_send_ir_code_service,
|
|
schema=SEND_IR_CODE_SERVICE_SCHEMA)
|
|
|
|
|
|
class MySensorsSwitch(mysensors.device.MySensorsEntity, SwitchDevice):
|
|
"""Representation of the value of a MySensors Switch child node."""
|
|
|
|
@property
|
|
def assumed_state(self):
|
|
"""Return True if unable to access real state of entity."""
|
|
return self.gateway.optimistic
|
|
|
|
@property
|
|
def current_power_w(self):
|
|
"""Return the current power usage in W."""
|
|
set_req = self.gateway.const.SetReq
|
|
return self._values.get(set_req.V_WATT)
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return True if switch is on."""
|
|
return self._values.get(self.value_type) == STATE_ON
|
|
|
|
async def async_turn_on(self, **kwargs):
|
|
"""Turn the switch on."""
|
|
self.gateway.set_child_value(
|
|
self.node_id, self.child_id, self.value_type, 1)
|
|
if self.gateway.optimistic:
|
|
# Optimistically assume that switch has changed state
|
|
self._values[self.value_type] = STATE_ON
|
|
self.async_schedule_update_ha_state()
|
|
|
|
async def async_turn_off(self, **kwargs):
|
|
"""Turn the switch off."""
|
|
self.gateway.set_child_value(
|
|
self.node_id, self.child_id, self.value_type, 0)
|
|
if self.gateway.optimistic:
|
|
# Optimistically assume that switch has changed state
|
|
self._values[self.value_type] = STATE_OFF
|
|
self.async_schedule_update_ha_state()
|
|
|
|
|
|
class MySensorsIRSwitch(MySensorsSwitch):
|
|
"""IR switch child class to MySensorsSwitch."""
|
|
|
|
def __init__(self, *args):
|
|
"""Set up instance attributes."""
|
|
super().__init__(*args)
|
|
self._ir_code = None
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return True if switch is on."""
|
|
set_req = self.gateway.const.SetReq
|
|
return self._values.get(set_req.V_LIGHT) == STATE_ON
|
|
|
|
async def async_turn_on(self, **kwargs):
|
|
"""Turn the IR switch on."""
|
|
set_req = self.gateway.const.SetReq
|
|
if ATTR_IR_CODE in kwargs:
|
|
self._ir_code = kwargs[ATTR_IR_CODE]
|
|
self.gateway.set_child_value(
|
|
self.node_id, self.child_id, self.value_type, self._ir_code)
|
|
self.gateway.set_child_value(
|
|
self.node_id, self.child_id, set_req.V_LIGHT, 1)
|
|
if self.gateway.optimistic:
|
|
# Optimistically assume that switch has changed state
|
|
self._values[self.value_type] = self._ir_code
|
|
self._values[set_req.V_LIGHT] = STATE_ON
|
|
self.async_schedule_update_ha_state()
|
|
# Turn off switch after switch was turned on
|
|
await self.async_turn_off()
|
|
|
|
async def async_turn_off(self, **kwargs):
|
|
"""Turn the IR switch off."""
|
|
set_req = self.gateway.const.SetReq
|
|
self.gateway.set_child_value(
|
|
self.node_id, self.child_id, set_req.V_LIGHT, 0)
|
|
if self.gateway.optimistic:
|
|
# Optimistically assume that switch has changed state
|
|
self._values[set_req.V_LIGHT] = STATE_OFF
|
|
self.async_schedule_update_ha_state()
|
|
|
|
async def async_update(self):
|
|
"""Update the controller with the latest value from a sensor."""
|
|
await super().async_update()
|
|
self._ir_code = self._values.get(self.value_type)
|