69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
import logging
|
||
|
|
from homeassistant.helpers.entity import ToggleEntity
|
||
|
|
from homeassistant.const import STATE_ON, STATE_OFF, DEVICE_DEFAULT_NAME
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
# pylint: disable=unused-argument
|
||
|
|
def setup_platform(hass, config, add_devices_callback, discovery_info=None):
|
||
|
|
""" Find and return switches controlled by shell commands. """
|
||
|
|
|
||
|
|
switches = config.get('switches', {})
|
||
|
|
devices = []
|
||
|
|
|
||
|
|
for k, v in switches.items():
|
||
|
|
devices.append(
|
||
|
|
CommandSwitch(
|
||
|
|
k,
|
||
|
|
v.get('oncmd', 'true'),
|
||
|
|
v.get('offcmd', 'true')))
|
||
|
|
|
||
|
|
add_devices_callback(devices)
|
||
|
|
|
||
|
|
|
||
|
|
class CommandSwitch(ToggleEntity):
|
||
|
|
def __init__(self, name, command_on, command_off):
|
||
|
|
self._name = name or DEVICE_DEFAULT_NAME
|
||
|
|
self._state = STATE_OFF
|
||
|
|
self._command_on = command_on
|
||
|
|
self._command_off = command_off
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def _switch(command):
|
||
|
|
_LOGGER.info('Running command: {}'.format(command))
|
||
|
|
|
||
|
|
success = (subprocess.call(command, shell=True) == 0)
|
||
|
|
|
||
|
|
if not success:
|
||
|
|
_LOGGER.error('Command failed: {}'.format(command))
|
||
|
|
|
||
|
|
return success
|
||
|
|
|
||
|
|
@property
|
||
|
|
def should_poll(self):
|
||
|
|
return False
|
||
|
|
|
||
|
|
@property
|
||
|
|
def name(self):
|
||
|
|
return self._name
|
||
|
|
|
||
|
|
@property
|
||
|
|
def state(self):
|
||
|
|
return self._state
|
||
|
|
|
||
|
|
@property
|
||
|
|
def is_on(self):
|
||
|
|
return self._state == STATE_ON
|
||
|
|
|
||
|
|
def turn_on(self, **kwargs):
|
||
|
|
if CommandSwitch._switch(self._command_on):
|
||
|
|
self._state = STATE_ON
|
||
|
|
|
||
|
|
def turn_off(self, **kwargs):
|
||
|
|
if CommandSwitch._switch(self._command_off):
|
||
|
|
self._state = STATE_OFF
|
||
|
|
|