evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Enforce MediaPlayerState in hdmi_cec media player (#78522)

Browse source
This commit is contained in:
epenet 2022-09-15 17:48:05 +02:00 • committed by GitHub
parent 7d11509011
commit b29605060a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 29 additions and 66 deletions
Download patch file Download diff file Expand all files Collapse all files

20
homeassistant/components/hdmi_cec/switch.py
View file

@ -4,8 +4,9 @@ from __future__ import annotations
import logging
from typing import Any
from pycec.const import POWER_OFF, POWER_ON
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN, SwitchEntity
from homeassistant.const import STATE_OFF, STATE_ON
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
@ -44,16 +45,21 @@ class CecSwitchEntity(CecEntity, SwitchEntity):
def turn_on(self, **kwargs: Any) -> None:
"""Turn device on."""
self._device.turn_on()
self._state = STATE_ON
self._attr_is_on = True
self.schedule_update_ha_state(force_refresh=False)
def turn_off(self, **kwargs: Any) -> None:
"""Turn device off."""
self._device.turn_off()
self._state = STATE_OFF
self._attr_is_on = False
self.schedule_update_ha_state(force_refresh=False)
@property
def is_on(self) -> bool:
"""Return True if entity is on."""
return self._state == STATE_ON
def update(self) -> None:
"""Update device status."""
device = self._device
if device.power_status in {POWER_OFF, 3}:
self._attr_is_on = False
elif device.power_status in {POWER_ON, 4}:
self._attr_is_on = True
else:
_LOGGER.warning("Unknown state: %d", device.power_status)