Migrate to voluptuous (#3162)

🐬 and 🍪 for fixing quotes!
pull/3176/head
Fabian Affolter 2016-09-04 04:20:45 +02:00 committed by Teagan Glenn
parent 6a2f0fc456
commit 3b1c0a7502
1 changed files with 35 additions and 33 deletions

View File

@ -7,32 +7,34 @@ https://home-assistant.io/components/media_player.denon/
import logging
import telnetlib
import voluptuous as vol
from homeassistant.components.media_player import (
DOMAIN, SUPPORT_NEXT_TRACK, SUPPORT_PAUSE, SUPPORT_PREVIOUS_TRACK,
PLATFORM_SCHEMA, SUPPORT_NEXT_TRACK, SUPPORT_PAUSE, SUPPORT_PREVIOUS_TRACK,
SUPPORT_TURN_OFF, SUPPORT_TURN_ON, SUPPORT_VOLUME_MUTE, SUPPORT_VOLUME_SET,
MediaPlayerDevice)
from homeassistant.const import CONF_HOST, STATE_OFF, STATE_ON, STATE_UNKNOWN
from homeassistant.const import (
CONF_HOST, CONF_NAME, STATE_OFF, STATE_ON, STATE_UNKNOWN)
import homeassistant.helpers.config_validation as cv
_LOGGER = logging.getLogger(__name__)
DEFAULT_NAME = 'Music station'
SUPPORT_DENON = SUPPORT_PAUSE | SUPPORT_VOLUME_SET | SUPPORT_VOLUME_MUTE | \
SUPPORT_PREVIOUS_TRACK | SUPPORT_NEXT_TRACK | \
SUPPORT_TURN_ON | SUPPORT_TURN_OFF
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_HOST): cv.string,
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
})
def setup_platform(hass, config, add_devices, discovery_info=None):
"""Setup the Denon platform."""
if not config.get(CONF_HOST):
_LOGGER.error(
"Missing required configuration items in %s: %s",
DOMAIN,
CONF_HOST)
return False
denon = DenonDevice(config.get(CONF_NAME), config.get(CONF_HOST))
denon = DenonDevice(
config.get("name", "Music station"),
config.get("host")
)
if denon.update():
add_devices([denon])
return True
@ -48,21 +50,21 @@ class DenonDevice(MediaPlayerDevice):
"""Initialize the Denon device."""
self._name = name
self._host = host
self._pwstate = "PWSTANDBY"
self._pwstate = 'PWSTANDBY'
self._volume = 0
self._muted = False
self._mediasource = ""
self._mediasource = ''
@classmethod
def telnet_request(cls, telnet, command):
"""Execute `command` and return the response."""
telnet.write(command.encode("ASCII") + b"\r")
return telnet.read_until(b"\r", timeout=0.2).decode("ASCII").strip()
telnet.write(command.encode('ASCII') + b'\r')
return telnet.read_until(b'\r', timeout=0.2).decode('ASCII').strip()
def telnet_command(self, command):
"""Establish a telnet connection and sends `command`."""
telnet = telnetlib.Telnet(self._host)
telnet.write(command.encode("ASCII") + b"\r")
telnet.write(command.encode('ASCII') + b'\r')
telnet.read_very_eager() # skip response
telnet.close()
@ -73,14 +75,14 @@ class DenonDevice(MediaPlayerDevice):
except ConnectionRefusedError:
return False
self._pwstate = self.telnet_request(telnet, "PW?")
self._pwstate = self.telnet_request(telnet, 'PW?')
# PW? sends also SISTATUS, which is not interesting
telnet.read_until(b"\r", timeout=0.2)
volume_str = self.telnet_request(telnet, "MV?")[len("MV"):]
volume_str = self.telnet_request(telnet, 'MV?')[len('MV'):]
self._volume = int(volume_str) / 60
self._muted = (self.telnet_request(telnet, "MU?") == "MUON")
self._mediasource = self.telnet_request(telnet, "SI?")[len("SI"):]
self._muted = (self.telnet_request(telnet, 'MU?') == 'MUON')
self._mediasource = self.telnet_request(telnet, 'SI?')[len('SI'):]
telnet.close()
return True
@ -93,9 +95,9 @@ class DenonDevice(MediaPlayerDevice):
@property
def state(self):
"""Return the state of the device."""
if self._pwstate == "PWSTANDBY":
if self._pwstate == 'PWSTANDBY':
return STATE_OFF
if self._pwstate == "PWON":
if self._pwstate == 'PWON':
return STATE_ON
return STATE_UNKNOWN
@ -122,41 +124,41 @@ class DenonDevice(MediaPlayerDevice):
def turn_off(self):
"""Turn off media player."""
self.telnet_command("PWSTANDBY")
self.telnet_command('PWSTANDBY')
def volume_up(self):
"""Volume up media player."""
self.telnet_command("MVUP")
self.telnet_command('MVUP')
def volume_down(self):
"""Volume down media player."""
self.telnet_command("MVDOWN")
self.telnet_command('MVDOWN')
def set_volume_level(self, volume):
"""Set volume level, range 0..1."""
# 60dB max
self.telnet_command("MV" + str(round(volume * 60)).zfill(2))
self.telnet_command('MV' + str(round(volume * 60)).zfill(2))
def mute_volume(self, mute):
"""Mute (true) or unmute (false) media player."""
self.telnet_command("MU" + ("ON" if mute else "OFF"))
self.telnet_command('MU' + ('ON' if mute else 'OFF'))
def media_play(self):
"""Play media media player."""
self.telnet_command("NS9A")
self.telnet_command('NS9A')
def media_pause(self):
"""Pause media player."""
self.telnet_command("NS9B")
self.telnet_command('NS9B')
def media_next_track(self):
"""Send the next track command."""
self.telnet_command("NS9D")
self.telnet_command('NS9D')
def media_previous_track(self):
"""Send the previous track command."""
self.telnet_command("NS9E")
self.telnet_command('NS9E')
def turn_on(self):
"""Turn the media player on."""
self.telnet_command("PWON")
self.telnet_command('PWON')