def test_minmax_stepper_step(minmax, value, steps, direction, expected_value,
expected_exceeded):
stepper = MinMaxStepper(*minmax, steps)
# SUT
new_value, exceeded = stepper.step(value, direction)
# Checks
assert new_value == expected_value
assert exceeded == expected_exceeded
class MediaPlayerController(TypeController, ReleaseHoldController):
async def initialize(self) -> None:
self.media_player = self.args["media_player"]
await self.check_domain(self.media_player)
volume_steps = self.args.get("volume_steps", DEFAULT_VOLUME_STEPS)
self.volume_stepper = MinMaxStepper(0, 1, volume_steps)
self.volume_level = 0.0
self.supported_features = MediaPlayerSupport(self.media_player, self)
await super().initialize()
def get_domain(self) -> str:
return "media_player"
def get_type_actions_mapping(self) -> TypeActionsMapping:
return {
MediaPlayer.HOLD_VOLUME_DOWN: (self.hold, Stepper.DOWN),
MediaPlayer.HOLD_VOLUME_UP: (self.hold, Stepper.UP),
MediaPlayer.CLICK_VOLUME_DOWN: self.volume_down,
MediaPlayer.CLICK_VOLUME_UP: self.volume_up,
MediaPlayer.RELEASE: self.release,
MediaPlayer.PLAY_PAUSE: self.play_pause,
MediaPlayer.NEXT_TRACK: self.next_track,
MediaPlayer.PREVIOUS_TRACK: self.previous_track,
MediaPlayer.NEXT_SOURCE: (self.change_source_list, Stepper.UP),
MediaPlayer.PREVIOUS_SOURCE:
(self.change_source_list, Stepper.DOWN),
}
@action
async def change_source_list(self, direction: str) -> None:
entity_states = await self.get_entity_state(self.media_player,
attribute="all")
entity_attributes = entity_states["attributes"]
source_list = entity_attributes.get("source_list")
if len(source_list) == 0 or source_list is None:
self.log(
f"⚠️ There is no `source_list` parameter in `{self.media_player}`",
level="WARNING",
ascii_encode=False,
)
return
source = entity_attributes.get("source")
if source is None:
new_index_source = 0
else:
index_source = source_list.index(source)
source_stepper = CircularStepper(0,
len(source_list) - 1,
len(source_list))
new_index_source, _ = source_stepper.step(index_source, direction)
await self.call_service(
"media_player/select_source",
entity_id=self.media_player,
source=source_list[new_index_source],
)
@action
async def play_pause(self) -> None:
await self.call_service("media_player/media_play_pause",
entity_id=self.media_player)
@action
async def previous_track(self) -> None:
await self.call_service("media_player/media_previous_track",
entity_id=self.media_player)
@action
async def next_track(self) -> None:
await self.call_service("media_player/media_next_track",
entity_id=self.media_player)
@action
async def volume_up(self) -> None:
await self.prepare_volume_change()
await self.volume_change(Stepper.UP)
@action
async def volume_down(self) -> None:
await self.prepare_volume_change()
await self.volume_change(Stepper.DOWN)
@action
async def hold(self, direction: str) -> None:
await self.prepare_volume_change()
await super().hold(direction)
async def prepare_volume_change(self) -> None:
volume_level = await self.get_entity_state(self.media_player,
attribute="volume_level")
if volume_level is not None:
self.volume_level = volume_level
async def volume_change(self, direction: str) -> bool:
if await self.supported_features.is_supported(
MediaPlayerSupport.VOLUME_SET):
self.volume_level, exceeded = self.volume_stepper.step(
self.volume_level, direction)
await self.call_service(
"media_player/volume_set",
entity_id=self.media_player,
volume_level=self.volume_level,
)
return exceeded
else:
if direction == Stepper.UP:
await self.call_service("media_player/volume_up",
entity_id=self.media_player)
else:
await self.call_service("media_player/volume_down",
entity_id=self.media_player)
return False
async def hold_loop(self, direction: str) -> bool: # type: ignore
return await self.volume_change(direction)
def default_delay(self) -> int:
return 500
class MediaPlayerController(ReleaseHoldController):
def initialize(self):
super().initialize()
self.media_player = self.args["media_player"]
volume_steps = self.args.get("volume_steps", DEFAULT_VOLUME_STEPS)
self.volume_stepper = MinMaxStepper(0, 1, volume_steps)
self.volume_level = 0
def get_type_actions_mapping(self):
return {
MediaPlayer.HOLD_VOLUME_DOWN: (self.hold, Stepper.DOWN),
MediaPlayer.HOLD_VOLUME_UP: (self.hold, Stepper.UP),
MediaPlayer.CLICK_VOLUME_DOWN: self.volume_down,
MediaPlayer.CLICK_VOLUME_UP: self.volume_up,
MediaPlayer.RELEASE: self.release,
MediaPlayer.PLAY_PAUSE: self.play_pause,
MediaPlayer.NEXT_TRACK: self.next_track,
MediaPlayer.PREVIOUS_TRACK: self.previous_track,
MediaPlayer.NEXT_SOURCE: (self.change_source_list, Stepper.UP),
MediaPlayer.PREVIOUS_SOURCE:
(self.change_source_list, Stepper.DOWN),
}
@action
async def change_source_list(self, direction):
entity_states = await self.get_entity_state(self.media_player,
attribute="all")
entity_attributes = entity_states["attributes"]
source_list = entity_attributes.get("source_list")
if len(source_list) == 0 or source_list is None:
self.log(
"There is no 'source_list' parameter in this media player",
level="WARNING",
)
return
source = entity_attributes.get("source")
if source is None:
new_index_source = 0
else:
index_source = source_list.index(source)
source_stepper = CircularStepper(0,
len(source_list) - 1,
len(source_list))
new_index_source, _ = source_stepper.step(index_source, direction)
self.call_service(
"media_player/select_source",
entity_id=self.media_player,
source=source_list[new_index_source],
)
@action
async def play_pause(self):
self.call_service("media_player/media_play_pause",
entity_id=self.media_player)
@action
async def previous_track(self):
self.call_service("media_player/media_previous_track",
entity_id=self.media_player)
@action
async def next_track(self):
self.call_service("media_player/media_next_track",
entity_id=self.media_player)
@action
async def volume_up(self):
await self.prepare_volume_change()
await self.volume_change(Stepper.UP)
@action
async def volume_down(self):
await self.prepare_volume_change()
await self.volume_change(Stepper.DOWN)
@action
async def hold(self, direction):
await self.prepare_volume_change()
await super().hold(direction)
async def prepare_volume_change(self):
volume_level = await self.get_entity_state(self.media_player,
attribute="volume_level")
if volume_level is not None:
self.volume_level = volume_level
async def volume_change(self, direction):
self.volume_level, exceeded = self.volume_stepper.step(
self.volume_level, direction)
self.call_service(
"media_player/volume_set",
entity_id=self.media_player,
volume_level=self.volume_level,
)
return exceeded
async def hold_loop(self, direction):
return await self.volume_change(direction)
def default_delay(self):
return 500
class MediaPlayerController(ReleaseHoldController):
def initialize(self):
super().initialize()
self.media_player = self.args["media_player"]
volume_steps = self.args.get("volume_steps", DEFAULT_VOLUME_STEPS)
self.volume_stepper = MinMaxStepper(0, 1, volume_steps)
self.volume_level = 0
def get_type_actions_mapping(self):
return {
MediaPlayer.HOLD_VOLUME_DOWN: (self.hold, Stepper.DOWN),
MediaPlayer.HOLD_VOLUME_UP: (self.hold, Stepper.UP),
MediaPlayer.CLICK_VOLUME_DOWN: self.volume_down,
MediaPlayer.CLICK_VOLUME_UP: self.volume_up,
MediaPlayer.RELEASE: self.release,
MediaPlayer.PLAY_PAUSE: self.play_pause,
MediaPlayer.NEXT_TRACK: self.next_track,
MediaPlayer.PREVIOUS_TRACK: self.previous_track,
}
@action
async def play_pause(self):
self.call_service("media_player/media_play_pause",
entity_id=self.media_player)
@action
async def previous_track(self):
self.call_service("media_player/media_previous_track",
entity_id=self.media_player)
@action
async def next_track(self):
self.call_service("media_player/media_next_track",
entity_id=self.media_player)
@action
async def volume_up(self):
await self.prepare_volume_change()
await self.volume_change(Stepper.UP)
@action
async def volume_down(self):
await self.prepare_volume_change()
await self.volume_change(Stepper.DOWN)
@action
async def hold(self, direction):
await self.prepare_volume_change()
await super().hold(direction)
async def prepare_volume_change(self):
volume_level = await self.get_entity_state(self.media_player,
attribute="volume_level")
if volume_level is not None:
self.volume_level = volume_level
async def volume_change(self, direction):
self.volume_level, exceeded = self.volume_stepper.step(
self.volume_level, direction)
self.call_service(
"media_player/volume_set",
entity_id=self.media_player,
volume_level=self.volume_level,
)
return exceeded
async def hold_loop(self, direction):
return await self.volume_change(direction)
def default_delay(self):
return 500