Allow customizable turn on action for LG WebOS tv (#9206)

* allow customizable action for webos tv turn on as not all models allow for WOL

* trying to fix the houndci-bot

* last few fixes hopefully

* I guess not

* last time!

* This is a breaking change. I have removed the build-in wake-on-lan functionality and have opted for a script which can be a wake-on-lan switch. I have also removed any reference to wol.

* hoping to fix formatting

* linter errors

homeassistant/components/media_player/webostv.py

@@ -19,10 +19,11 @@ from homeassistant.components.media_player import (
     SUPPORT_SELECT_SOURCE, SUPPORT_PLAY_MEDIA, MEDIA_TYPE_CHANNEL,
     MediaPlayerDevice, PLATFORM_SCHEMA)
 from homeassistant.const import (
-    CONF_HOST, CONF_MAC, CONF_CUSTOMIZE, CONF_TIMEOUT, STATE_OFF,
+    CONF_HOST, CONF_CUSTOMIZE, CONF_TIMEOUT, STATE_OFF,
     STATE_PLAYING, STATE_PAUSED,
     STATE_UNKNOWN, CONF_NAME, CONF_FILENAME)
 import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.script import Script
 
 REQUIREMENTS = ['pylgtv==0.1.7',
                 'websockets==3.2',
@@ -32,6 +33,7 @@ _CONFIGURING = {}  # type: Dict[str, str]
 _LOGGER = logging.getLogger(__name__)
 
 CONF_SOURCES = 'sources'
+CONF_ON_ACTION = 'turn_on_action'
 
 DEFAULT_NAME = 'LG webOS Smart TV'
 
@@ -53,10 +55,10 @@ CUSTOMIZE_SCHEMA = vol.Schema({
 PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
     vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
     vol.Optional(CONF_HOST): cv.string,
-    vol.Optional(CONF_MAC): cv.string,
     vol.Optional(CONF_CUSTOMIZE, default={}): CUSTOMIZE_SCHEMA,
     vol.Optional(CONF_FILENAME, default=WEBOSTV_CONFIG_FILE): cv.string,
     vol.Optional(CONF_TIMEOUT, default=10): cv.positive_int,
+    vol.Optional(CONF_ON_ACTION): cv.SCRIPT_SCHEMA,
 })
 
 
@@ -76,15 +78,19 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
     if host in _CONFIGURING:
         return
 
-    mac = config.get(CONF_MAC)
     name = config.get(CONF_NAME)
     customize = config.get(CONF_CUSTOMIZE)
     timeout = config.get(CONF_TIMEOUT)
+    turn_on_action = config.get(CONF_ON_ACTION)
+
     config = hass.config.path(config.get(CONF_FILENAME))
-    setup_tv(host, mac, name, customize, config, timeout, hass, add_devices)
+
+    setup_tv(host, name, customize, config, timeout, hass,
+             add_devices, turn_on_action)
 
 
-def setup_tv(host, mac, name, customize, config, timeout, hass, add_devices):
+def setup_tv(host, name, customize, config, timeout, hass,
+             add_devices, turn_on_action):
     """Set up a LG WebOS TV based on host parameter."""
     from pylgtv import WebOsClient
     from pylgtv import PyLGTVPairException
@@ -108,7 +114,8 @@ def setup_tv(host, mac, name, customize, config, timeout, hass, add_devices):
             # Not registered, request configuration.
             _LOGGER.warning("LG webOS TV %s needs to be paired", host)
             request_configuration(
-                host, mac, name, customize, config, timeout, hass, add_devices)
+                host, name, customize, config, timeout, hass,
+                add_devices, turn_on_action)
             return
 
     # If we came here and configuring this host, mark as done.
@@ -117,12 +124,13 @@ def setup_tv(host, mac, name, customize, config, timeout, hass, add_devices):
         configurator = hass.components.configurator
         configurator.request_done(request_id)
 
-    add_devices([LgWebOSDevice(host, mac, name, customize, config, timeout)],
-                True)
+    add_devices([LgWebOSDevice(host, name, customize, config, timeout,
+                               hass, turn_on_action)], True)
 
 
 def request_configuration(
-        host, mac, name, customize, config, timeout, hass, add_devices):
+        host, name, customize, config, timeout, hass,
+        add_devices, turn_on_action):
     """Request configuration steps from the user."""
     configurator = hass.components.configurator
 
@@ -135,8 +143,8 @@ def request_configuration(
     # pylint: disable=unused-argument
     def lgtv_configuration_callback(data):
         """The actions to do when our configuration callback is called."""
-        setup_tv(host, mac, name, customize, config, timeout, hass,
-                 add_devices)
+        setup_tv(host, name, customize, config, timeout, hass,
+                 add_devices, turn_on_action)
 
     _CONFIGURING[host] = configurator.request_config(
         name, lgtv_configuration_callback,
@@ -149,13 +157,12 @@ def request_configuration(
 class LgWebOSDevice(MediaPlayerDevice):
     """Representation of a LG WebOS TV."""
 
-    def __init__(self, host, mac, name, customize, config, timeout):
+    def __init__(self, host, name, customize, config, timeout,
+                 hass, on_action):
         """Initialize the webos device."""
         from pylgtv import WebOsClient
-        from wakeonlan import wol
         self._client = WebOsClient(host, config, timeout)
-        self._wol = wol
-        self._mac = mac
+        self._on_script = Script(hass, on_action) if on_action else None
         self._customize = customize
 
         self._name = name
@@ -273,7 +280,7 @@ class LgWebOSDevice(MediaPlayerDevice):
     @property
     def supported_features(self):
         """Flag media player features that are supported."""
-        if self._mac:
+        if self._on_script:
             return SUPPORT_WEBOSTV | SUPPORT_TURN_ON
         return SUPPORT_WEBOSTV
 
@@ -289,8 +296,8 @@ class LgWebOSDevice(MediaPlayerDevice):
 
     def turn_on(self):
         """Turn on the media player."""
-        if self._mac:
-            self._wol.send_magic_packet(self._mac)
+        if self._on_script:
+            self._on_script.run()
 
     def volume_up(self):
         """Volume up the media player."""