Support buffering in media_player device conditions (#70863)

Co-authored-by: Franck Nijhof <git@frenck.dev>
This commit is contained in:
Erik Montnemery 2022-04-27 15:40:07 +02:00 committed by GitHub
parent 9daf3d8e72
commit c530bc823b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 72 additions and 9 deletions

View file

@ -10,6 +10,7 @@ from homeassistant.const import (
CONF_DOMAIN,
CONF_ENTITY_ID,
CONF_TYPE,
STATE_BUFFERING,
STATE_IDLE,
STATE_OFF,
STATE_ON,
@ -23,7 +24,14 @@ from homeassistant.helpers.typing import ConfigType, TemplateVarsType
from .const import DOMAIN
CONDITION_TYPES = {"is_on", "is_off", "is_idle", "is_paused", "is_playing"}
CONDITION_TYPES = {
"is_on",
"is_off",
"is_buffering",
"is_idle",
"is_paused",
"is_playing",
}
CONDITION_SCHEMA = DEVICE_CONDITION_BASE_SCHEMA.extend(
{
@ -63,16 +71,18 @@ def async_condition_from_config(
hass: HomeAssistant, config: ConfigType
) -> condition.ConditionCheckerType:
"""Create a function to test a device condition."""
if config[CONF_TYPE] == "is_playing":
state = STATE_PLAYING
if config[CONF_TYPE] == "is_buffering":
state = STATE_BUFFERING
elif config[CONF_TYPE] == "is_idle":
state = STATE_IDLE
elif config[CONF_TYPE] == "is_paused":
state = STATE_PAUSED
elif config[CONF_TYPE] == "is_off":
state = STATE_OFF
elif config[CONF_TYPE] == "is_on":
state = STATE_ON
else:
state = STATE_OFF
elif config[CONF_TYPE] == "is_paused":
state = STATE_PAUSED
else: # is_playing
state = STATE_PLAYING
def test_is_state(hass: HomeAssistant, variables: TemplateVarsType) -> bool:
"""Test if an entity is a certain state."""