22
33import logging
44
5- from homeassistant .components .media_player .const import (
6- MEDIA_TYPE_CHANNEL ,
7- MEDIA_TYPE_MUSIC ,
8- SUPPORT_NEXT_TRACK ,
9- SUPPORT_SELECT_SOURCE ,
10- SUPPORT_TURN_OFF ,
11- SUPPORT_TURN_ON ,
12- SUPPORT_VOLUME_MUTE ,
13- SUPPORT_VOLUME_SET ,
14- SUPPORT_VOLUME_STEP ,
15- )
165from homeassistant .const import STATE_OFF , STATE_PLAYING # STATE_IDLE,
17-
6+ from homeassistant .components .media_player import (
7+ MediaPlayerEntityFeature ,
8+ MediaPlayerEntity ,
9+ MediaType )
1810from .vimar_entity import VimarEntity , vimar_setup_entry
19-
20- try :
21- from homeassistant .components .media_player import MediaPlayerEntity
22- except ImportError :
23- from homeassistant .components .media_player import MediaPlayerDevice as MediaPlayerEntity
24-
2511from .const import DEVICE_TYPE_MEDIA_PLAYERS as CURR_PLATFORM
2612
2713_LOGGER = logging .getLogger (__name__ )
@@ -128,9 +114,9 @@ def media_content_type(self):
128114 """Content type of current playing media."""
129115 # if self.has_state('source') and self.get_state('source') == self._channel_source_id:
130116 if self .has_state ("channel" ) and self .get_state ("channel" ) == self ._channel_source_id :
131- return MEDIA_TYPE_CHANNEL
117+ return MediaType . CHANNEL
132118 else :
133- return MEDIA_TYPE_MUSIC
119+ return MediaType . MUSIC
134120
135121 @property
136122 def media_title (self ):
@@ -158,12 +144,12 @@ def supported_features(self):
158144 """Flag supported features."""
159145 flags = 0
160146 if self .has_state ("on/off" ):
161- flags |= SUPPORT_TURN_ON | SUPPORT_TURN_OFF
147+ flags |= MediaPlayerEntityFeature . TURN_ON | MediaPlayerEntityFeature . TURN_OFF
162148 if self .has_state ("volume" ):
163- flags |= SUPPORT_VOLUME_SET | SUPPORT_VOLUME_MUTE | SUPPORT_VOLUME_STEP
149+ flags |= MediaPlayerEntityFeature . VOLUME_SET | MediaPlayerEntityFeature . VOLUME_MUTE | MediaPlayerEntityFeature . VOLUME_STEP
164150 # if self.has_state('source'):
165151 if self .has_state ("channel" ):
166- flags |= SUPPORT_SELECT_SOURCE
152+ flags |= MediaPlayerEntityFeature . SELECT_SOURCE
167153 # channel only available on source == 0 /
168154 # if self.get_state('source') == 0 and self.has_state('channel'):
169155 # if self.get_state('channel') == self._channel_source_id and self.has_state('source'):
@@ -172,7 +158,7 @@ def supported_features(self):
172158 if self .has_state ("source" ):
173159 # we can only do next track
174160 # flags |= SUPPORT_NEXT_TRACK | SUPPORT_PREVIOUS_TRACK
175- flags |= SUPPORT_NEXT_TRACK
161+ flags |= MediaPlayerEntityFeature . NEXT_TRACK
176162
177163 # FIXED FIX ME - remove me in live
178164 # flags |= SUPPORT_VOLUME_SET | SUPPORT_VOLUME_MUTE | SUPPORT_SELECT_SOURCE | SUPPORT_NEXT_TRACK | SUPPORT_PREVIOUS_TRACK
0 commit comments