Add webostv sound_output capability

homeassistant/components/webostv/__init__.py

@@ -16,6 +16,7 @@
 )
 import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.dispatcher import async_dispatcher_send
+from .media_player import ATTR_SOUND_OUTPUT
 
 DOMAIN = "webostv"
 
@@ -30,6 +31,8 @@
 SERVICE_COMMAND = "command"
 ATTR_COMMAND = "command"
 
+SERVICE_SELECT_SOUND_OUTPUT = "select_sound_output"
+
 CUSTOMIZE_SCHEMA = vol.Schema(
     {vol.Optional(CONF_SOURCES, default=[]): vol.All(cv.ensure_list, [cv.string])}
 )
@@ -60,9 +63,15 @@
 
 COMMAND_SCHEMA = CALL_SCHEMA.extend({vol.Required(ATTR_COMMAND): cv.string})
 
+SOUND_OUTPUT_SCHEMA = CALL_SCHEMA.extend({vol.Required(ATTR_SOUND_OUTPUT): cv.string})
+
 SERVICE_TO_METHOD = {
     SERVICE_BUTTON: {"method": "async_button", "schema": BUTTON_SCHEMA},
     SERVICE_COMMAND: {"method": "async_command", "schema": COMMAND_SCHEMA},
+    SERVICE_SELECT_SOUND_OUTPUT: {
+        "method": "async_select_sound_output",
+        "schema": SOUND_OUTPUT_SCHEMA,
+    },
 }
 
 _LOGGER = logging.getLogger(__name__)

homeassistant/components/webostv/media_player.py

@@ -61,6 +61,8 @@
 
 LIVE_TV_APP_ID = "com.webos.app.livetv"
 
+ATTR_SOUND_OUTPUT = "sound_output"
+
 
 async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
     """Set up the LG WebOS TV platform."""
@@ -290,6 +292,14 @@ def supported_features(self):
             return SUPPORT_WEBOSTV | SUPPORT_TURN_ON
         return SUPPORT_WEBOSTV
 
+    @property
+    def device_state_attributes(self):
+        """Return device specific state attributes."""
+        attributes = {}
+        if self._client.sound_output is not None and self.state != STATE_OFF:
+            attributes[ATTR_SOUND_OUTPUT] = self._client.sound_output
+        return attributes
+
     @cmd
     async def async_turn_off(self):
         """Turn off media player."""
@@ -321,6 +331,11 @@ async def async_mute_volume(self, mute):
         """Send mute command."""
         await self._client.set_mute(mute)
 
+    @cmd
+    async def async_select_sound_output(self, sound_output):
+        """Select the sound output."""
+        await self._client.change_sound_output(sound_output)
+
     @cmd
     async def async_media_play_pause(self):
         """Simulate play pause media player."""

homeassistant/components/webostv/services.yaml

@@ -24,3 +24,12 @@ command:
         https://github.com/TheRealLink/pylgtv/blob/master/pylgtv/endpoints.py
       example: 'media.controls/rewind'
 
+select_sound_output:
+  description: 'Send the TV the command to change sound output.'
+  fields:
+    entity_id:
+      description: Name(s) of the webostv entities to change sound output on.
+      example: 'media_player.living_room_tv'
+    sound_output:
+      description: Name of the sound output to switch to.
+      example: 'external_speaker'