Yamaha: added mapping and exclude lists (#1880)

* added mapping and exclude lists

* reworked pr

* made code more pythonic

homeassistant/components/media_player/yamaha.py

@@ -18,11 +18,19 @@ _LOGGER = logging.getLogger(__name__)
 SUPPORT_YAMAHA = SUPPORT_VOLUME_SET | SUPPORT_VOLUME_MUTE | \
                  SUPPORT_TURN_ON | SUPPORT_TURN_OFF | SUPPORT_SELECT_SOURCE
 
+CONF_SOURCE_NAMES = 'source_names'
+CONF_SOURCE_IGNORE = 'source_ignore'
+
 
 def setup_platform(hass, config, add_devices, discovery_info=None):
     """Setup the Yamaha platform."""
     import rxv
-    add_devices(YamahaDevice(config.get("name"), receiver)
+
+    source_ignore = config.get(CONF_SOURCE_IGNORE, [])
+    source_names = config.get(CONF_SOURCE_NAMES, {})
+
+    add_devices(
+        YamahaDevice(config.get("name"), receiver, source_ignore, source_names)
         for receiver in rxv.find())
 
 
@@ -30,7 +38,8 @@ class YamahaDevice(MediaPlayerDevice):
     """Representation of a Yamaha device."""
 
     # pylint: disable=too-many-public-methods, abstract-method
-    def __init__(self, name, receiver):
+    # pylint: disable=too-many-instance-attributes
+    def __init__(self, name, receiver, source_ignore, source_names):
         """Initialize the Yamaha Receiver."""
         self._receiver = receiver
         self._muted = False
@@ -38,6 +47,9 @@ class YamahaDevice(MediaPlayerDevice):
         self._pwstate = STATE_OFF
         self._current_source = None
         self._source_list = None
+        self._source_ignore = source_ignore
+        self._source_names = source_names
+        self._reverse_mapping = None
         self.update()
         self._name = name
 
@@ -49,8 +61,23 @@ class YamahaDevice(MediaPlayerDevice):
             self._pwstate = STATE_OFF
         self._muted = self._receiver.mute
         self._volume = (self._receiver.volume / 100) + 1
-        self._current_source = self._receiver.input
-        self._source_list = list(self._receiver.inputs().keys())
+
+        if self.source_list is None:
+            self.build_source_list()
+
+        current_source = self._receiver.input
+        self._current_source = self._source_names.get(current_source,
+                                                      current_source)
+
+    def build_source_list(self):
+        """Build the source list."""
+        self._reverse_mapping = {alias: source for source, alias in
+                                 self._source_names.items()}
+
+        self._source_list = sorted(
+            self._source_names.get(source, source) for source in
+            self._receiver.inputs()
+            if source not in self._source_ignore)
 
     @property
     def name(self):
@@ -108,4 +135,5 @@ class YamahaDevice(MediaPlayerDevice):
 
     def select_source(self, source):
         """Select input source."""
-        self._receiver.input = source
+        self._receiver.input = self._reverse_mapping.get(source,
+                                                         source)