cf90e49b50
* Make reproduce state use platform instead of rely on function * Fix types * address comment Martin.
29 lines
1 KiB
Python
29 lines
1 KiB
Python
"""Module that groups code required to handle state restore for component."""
|
|
from typing import Iterable, Optional
|
|
|
|
from homeassistant.core import Context, State
|
|
from homeassistant.helpers.typing import HomeAssistantType
|
|
|
|
|
|
async def async_reproduce_states(
|
|
hass: HomeAssistantType, states: Iterable[State], context: Optional[Context] = None
|
|
) -> None:
|
|
"""Reproduce component states."""
|
|
from . import get_entity_ids
|
|
from homeassistant.helpers.state import async_reproduce_state
|
|
|
|
states_copy = []
|
|
for state in states:
|
|
members = get_entity_ids(hass, state.entity_id)
|
|
for member in members:
|
|
states_copy.append(
|
|
State(
|
|
member,
|
|
state.state,
|
|
state.attributes,
|
|
last_changed=state.last_changed,
|
|
last_updated=state.last_updated,
|
|
context=state.context,
|
|
)
|
|
)
|
|
await async_reproduce_state(hass, states_copy, blocking=True, context=context)
|