Move imports to top for mediaroom (#29509)

pull/29223/head^2
springstan 2019-12-05 16:56:53 +01:00 committed by Martin Hjelmare
parent 2a92eb1962
commit 76f455cea9
1 changed files with 2 additions and 15 deletions

View File

@ -1,9 +1,10 @@
"""Support for the Mediaroom Set-up-box.""" """Support for the Mediaroom Set-up-box."""
import logging import logging
from pymediaroom import PyMediaroomError, Remote, State, install_mediaroom_protocol
import voluptuous as vol import voluptuous as vol
from homeassistant.components.media_player import MediaPlayerDevice, PLATFORM_SCHEMA from homeassistant.components.media_player import PLATFORM_SCHEMA, MediaPlayerDevice
from homeassistant.components.media_player.const import ( from homeassistant.components.media_player.const import (
MEDIA_TYPE_CHANNEL, MEDIA_TYPE_CHANNEL,
SUPPORT_NEXT_TRACK, SUPPORT_NEXT_TRACK,
@ -99,7 +100,6 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
async_add_entities([new_stb]) async_add_entities([new_stb])
if not config[CONF_OPTIMISTIC]: if not config[CONF_OPTIMISTIC]:
from pymediaroom import install_mediaroom_protocol
already_installed = hass.data.get(DISCOVERY_MEDIAROOM, None) already_installed = hass.data.get(DISCOVERY_MEDIAROOM, None)
if not already_installed: if not already_installed:
@ -123,7 +123,6 @@ class MediaroomDevice(MediaPlayerDevice):
def set_state(self, mediaroom_state): def set_state(self, mediaroom_state):
"""Map pymediaroom state to HA state.""" """Map pymediaroom state to HA state."""
from pymediaroom import State
state_map = { state_map = {
State.OFF: STATE_OFF, State.OFF: STATE_OFF,
@ -139,7 +138,6 @@ class MediaroomDevice(MediaPlayerDevice):
def __init__(self, host, device_id, optimistic=False, timeout=DEFAULT_TIMEOUT): def __init__(self, host, device_id, optimistic=False, timeout=DEFAULT_TIMEOUT):
"""Initialize the device.""" """Initialize the device."""
from pymediaroom import Remote
self.host = host self.host = host
self.stb = Remote(host) self.stb = Remote(host)
@ -184,7 +182,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_play_media(self, media_type, media_id, **kwargs): async def async_play_media(self, media_type, media_id, **kwargs):
"""Play media.""" """Play media."""
from pymediaroom import PyMediaroomError
_LOGGER.debug( _LOGGER.debug(
"STB(%s) Play media: %s (%s)", self.stb.stb_ip, media_id, media_type "STB(%s) Play media: %s (%s)", self.stb.stb_ip, media_id, media_type
@ -237,7 +234,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_turn_on(self): async def async_turn_on(self):
"""Turn on the receiver.""" """Turn on the receiver."""
from pymediaroom import PyMediaroomError
try: try:
self.set_state(await self.stb.turn_on()) self.set_state(await self.stb.turn_on())
@ -250,7 +246,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_turn_off(self): async def async_turn_off(self):
"""Turn off the receiver.""" """Turn off the receiver."""
from pymediaroom import PyMediaroomError
try: try:
self.set_state(await self.stb.turn_off()) self.set_state(await self.stb.turn_off())
@ -263,7 +258,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_media_play(self): async def async_media_play(self):
"""Send play command.""" """Send play command."""
from pymediaroom import PyMediaroomError
try: try:
_LOGGER.debug("media_play()") _LOGGER.debug("media_play()")
@ -277,7 +271,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_media_pause(self): async def async_media_pause(self):
"""Send pause command.""" """Send pause command."""
from pymediaroom import PyMediaroomError
try: try:
await self.stb.send_cmd("PlayPause") await self.stb.send_cmd("PlayPause")
@ -290,7 +283,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_media_stop(self): async def async_media_stop(self):
"""Send stop command.""" """Send stop command."""
from pymediaroom import PyMediaroomError
try: try:
await self.stb.send_cmd("Stop") await self.stb.send_cmd("Stop")
@ -303,7 +295,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_media_previous_track(self): async def async_media_previous_track(self):
"""Send Program Down command.""" """Send Program Down command."""
from pymediaroom import PyMediaroomError
try: try:
await self.stb.send_cmd("ProgDown") await self.stb.send_cmd("ProgDown")
@ -316,7 +307,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_media_next_track(self): async def async_media_next_track(self):
"""Send Program Up command.""" """Send Program Up command."""
from pymediaroom import PyMediaroomError
try: try:
await self.stb.send_cmd("ProgUp") await self.stb.send_cmd("ProgUp")
@ -329,7 +319,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_volume_up(self): async def async_volume_up(self):
"""Send volume up command.""" """Send volume up command."""
from pymediaroom import PyMediaroomError
try: try:
await self.stb.send_cmd("VolUp") await self.stb.send_cmd("VolUp")
@ -340,7 +329,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_volume_down(self): async def async_volume_down(self):
"""Send volume up command.""" """Send volume up command."""
from pymediaroom import PyMediaroomError
try: try:
await self.stb.send_cmd("VolDown") await self.stb.send_cmd("VolDown")
@ -350,7 +338,6 @@ class MediaroomDevice(MediaPlayerDevice):
async def async_mute_volume(self, mute): async def async_mute_volume(self, mute):
"""Send mute command.""" """Send mute command."""
from pymediaroom import PyMediaroomError
try: try:
await self.stb.send_cmd("Mute") await self.stb.send_cmd("Mute")