From de71fee0a095ea043c66be92ad8bd6685b7fc74f Mon Sep 17 00:00:00 2001 From: Thibault Cohen Date: Tue, 18 Apr 2017 12:09:06 -0400 Subject: [PATCH] Fix #7026 adding a new wol parameter (#7144) --- homeassistant/components/switch/wake_on_lan.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/switch/wake_on_lan.py b/homeassistant/components/switch/wake_on_lan.py index c86a4674bb6..4413da191d3 100644 --- a/homeassistant/components/switch/wake_on_lan.py +++ b/homeassistant/components/switch/wake_on_lan.py @@ -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)