jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-15 09:17:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add support for tracking entity attributes in ESPHome (#50528)

Browse Source
This commit is contained in:
Franck Nijhof 2021-05-14 00:39:53 +02:00 committed by GitHub
parent 35f304450c
commit a16629601a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 45 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
homeassistant/components/esphome/__init__.py
View File

@ -140,34 +140,65 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
) )
) )
async def send_home_assistant_state_event(event: Event) -> None:
"""Forward Home Assistant states updates to ESPHome."""
new_state = event.data.get("new_state")
if new_state is None:
return
entity_id = event.data.get("entity_id")
await cli.send_home_assistant_state(entity_id, None, new_state.state)
async def _send_home_assistant_state( async def _send_home_assistant_state(
entity_id: str, new_state: State | None entity_id: str, attribute: str | None, state: State | None
) -> None: ) -> None:
"""Forward Home Assistant states to ESPHome.""" """Forward Home Assistant states to ESPHome."""
await cli.send_home_assistant_state(entity_id, None, new_state.state) if state is None or (attribute and attribute not in state.attributes):
return
send_state = state.state
if attribute:
send_state = state.attributes[attribute]
# ESPHome only handles "on"/"off" for boolean values
if isinstance(send_state, bool):
send_state = "on" if send_state else "off"
await cli.send_home_assistant_state(entity_id, attribute, str(send_state))
@callback @callback
def async_on_state_subscription( def async_on_state_subscription(
entity_id: str, attribute: str | None = None entity_id: str, attribute: str | None = None
) -> None: ) -> None:
"""Subscribe and forward states for requested entities.""" """Subscribe and forward states for requested entities."""
async def send_home_assistant_state_event(event: Event) -> None:
"""Forward Home Assistant states updates to ESPHome."""
# Only communicate changes to the state or attribute tracked
if (
"old_state" in event.data
and "new_state" in event.data
and (
(
not attribute
and event.data["old_state"].state
== event.data["new_state"].state
)
or (
attribute
and attribute in event.data["old_state"].attributes
and attribute in event.data["new_state"].attributes
and event.data["old_state"].attributes[attribute]
== event.data["new_state"].attributes[attribute]
)
)
):
return
await _send_home_assistant_state(
event.data["entity_id"], attribute, event.data.get("new_state")
)
unsub = async_track_state_change_event( unsub = async_track_state_change_event(
hass, [entity_id], send_home_assistant_state_event hass, [entity_id], send_home_assistant_state_event
) )
entry_data.disconnect_callbacks.append(unsub) entry_data.disconnect_callbacks.append(unsub)
new_state = hass.states.get(entity_id)
if new_state is None:
return
# Send initial state # Send initial state
hass.async_create_task(_send_home_assistant_state(entity_id, new_state)) hass.async_create_task(
_send_home_assistant_state(entity_id, attribute, hass.states.get(entity_id))
)
async def on_login() -> None: async def on_login() -> None:
"""Subscribe to states and list entities on successful API login.""" """Subscribe to states and list entities on successful API login."""