Skip to content

Commit

Permalink
Cleanup on exit (#13918)
Browse files Browse the repository at this point in the history
* Cleanup on exit

* lint

* version bump

* pymediaroom version bump

* address @kellerza comment

* avoid None in the _name
  • Loading branch information
dgomes authored and balloob committed Apr 17, 2018
1 parent d0d61d1 commit 8fdeebc
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 7 deletions.
22 changes: 16 additions & 6 deletions homeassistant/components/media_player/mediaroom.py
Expand Up @@ -8,6 +8,7 @@

import voluptuous as vol

from homeassistant.core import callback
from homeassistant.components.media_player import (
MEDIA_TYPE_CHANNEL, SUPPORT_PAUSE, SUPPORT_PLAY_MEDIA, SUPPORT_TURN_OFF,
SUPPORT_TURN_ON, SUPPORT_STOP, PLATFORM_SCHEMA, SUPPORT_NEXT_TRACK,
Expand All @@ -20,11 +21,11 @@
from homeassistant.const import (
CONF_HOST, CONF_NAME, CONF_OPTIMISTIC, STATE_OFF,
CONF_TIMEOUT, STATE_PAUSED, STATE_PLAYING, STATE_STANDBY,
STATE_UNAVAILABLE
STATE_UNAVAILABLE, EVENT_HOMEASSISTANT_STOP
)
import homeassistant.helpers.config_validation as cv

REQUIREMENTS = ['pymediaroom==0.6']
REQUIREMENTS = ['pymediaroom==0.6.3']

_LOGGER = logging.getLogger(__name__)

Expand Down Expand Up @@ -81,12 +82,21 @@ def callback_notify(notify):
if not config[CONF_OPTIMISTIC]:
from pymediaroom import install_mediaroom_protocol

already_installed = hass.data.get(DISCOVERY_MEDIAROOM, False)
already_installed = hass.data.get(DISCOVERY_MEDIAROOM, None)
if not already_installed:
await install_mediaroom_protocol(
hass.data[DISCOVERY_MEDIAROOM] = await install_mediaroom_protocol(
responses_callback=callback_notify)

@callback
def stop_discovery(event):
"""Stop discovery of new mediaroom STB's."""
_LOGGER.debug("Stopping internal pymediaroom discovery.")
hass.data[DISCOVERY_MEDIAROOM].close()

hass.bus.async_listen_once(EVENT_HOMEASSISTANT_STOP,
stop_discovery)

_LOGGER.debug("Auto discovery installed")
hass.data[DISCOVERY_MEDIAROOM] = True


class MediaroomDevice(MediaPlayerDevice):
Expand Down Expand Up @@ -120,7 +130,7 @@ def __init__(self, host, device_id, optimistic=False,
self._channel = None
self._optimistic = optimistic
self._state = STATE_PLAYING if optimistic else STATE_STANDBY
self._name = 'Mediaroom {}'.format(device_id)
self._name = 'Mediaroom {}'.format(device_id if device_id else host)
self._available = True
if device_id:
self._unique_id = device_id
Expand Down
2 changes: 1 addition & 1 deletion requirements_all.txt
Expand Up @@ -832,7 +832,7 @@ pylutron==0.1.0
pymailgunner==1.4

# homeassistant.components.media_player.mediaroom
pymediaroom==0.6
pymediaroom==0.6.3

# homeassistant.components.media_player.xiaomi_tv
pymitv==1.0.0
Expand Down

0 comments on commit 8fdeebc

Please sign in to comment.