evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix #7026 adding a new wol parameter (#7144)

Browse source
This commit is contained in:
Thibault Cohen 2017-04-18 12:09:06 -04:00 • committed by Paulus Schoutsen
parent c7a11277ac
commit de71fee0a0
1 changed files with 10 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

14
homeassistant/components/switch/wake_on_lan.py
View file

@ -21,6 +21,7 @@ _LOGGER = logging.getLogger(__name__)
CONF_MAC_ADDRESS = 'mac_address'
CONF_OFF_ACTION = 'turn_off'
CONF_BROADCAST_ADDRESS = 'broadcast_address'
DEFAULT_NAME = 'Wake on LAN'
DEFAULT_PING_TIMEOUT = 1
@ -28,6 +29,7 @@ DEFAULT_PING_TIMEOUT = 1
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_MAC_ADDRESS): cv.string,
vol.Optional(CONF_HOST): cv.string,
vol.Optional(CONF_BROADCAST_ADDRESS): cv.string,
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
vol.Optional(CONF_OFF_ACTION): cv.SCRIPT_SCHEMA,
})
@ -38,21 +40,25 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
name = config.get(CONF_NAME)
host = config.get(CONF_HOST)
mac_address = config.get(CONF_MAC_ADDRESS)
broadcast_address = config.get(CONF_BROADCAST_ADDRESS)
off_action = config.get(CONF_OFF_ACTION)
add_devices([WOLSwitch(hass, name, host, mac_address, off_action)])
add_devices([WOLSwitch(hass, name, host, mac_address,
off_action, broadcast_address)])
class WOLSwitch(SwitchDevice):
"""Representation of a wake on lan switch."""
def __init__(self, hass, name, host, mac_address, off_action):
def __init__(self, hass, name, host, mac_address,
off_action, broadcast_address):
"""Initialize the WOL switch."""
from wakeonlan import wol
self._hass = hass
self._name = name
self._host = host
self._mac_address = mac_address
self._broadcast_address = broadcast_address
self._off_script = Script(hass, off_action) if off_action else None
self._state = False
self._wol = wol
@ -75,9 +81,9 @@ class WOLSwitch(SwitchDevice):
def turn_on(self):
"""Turn the device on."""
if self._host:
if self._broadcast_address:
self._wol.send_magic_packet(self._mac_address,
ip_address=self._host)
ip_address=self._broadcast_address)
else:
self._wol.send_magic_packet(self._mac_address)