mirror of
https://github.com/home-assistant/core.git
synced 2026-08-17 02:34:48 +01:00
25 lines
813 B
Python
25 lines
813 B
Python
"""Helpers for the zone integration."""
|
|
|
|
from homeassistant.components.device_tracker import (
|
|
DOMAIN as DEVICE_TRACKER_DOMAIN,
|
|
DeviceTrackerEntityStateAttribute,
|
|
)
|
|
from homeassistant.components.person import (
|
|
DOMAIN as PERSON_DOMAIN,
|
|
PersonEntityStateAttribute,
|
|
)
|
|
from homeassistant.core import State
|
|
|
|
|
|
def get_in_zones_attribute(state: State) -> str | None:
|
|
"""Return the in_zones attribute for the tracked entity, or None.
|
|
|
|
Only person and device_tracker entities report zone membership; each
|
|
exposes it under its own platform enum. Any other domain returns None.
|
|
"""
|
|
if state.domain == PERSON_DOMAIN:
|
|
return PersonEntityStateAttribute.IN_ZONES
|
|
if state.domain == DEVICE_TRACKER_DOMAIN:
|
|
return DeviceTrackerEntityStateAttribute.IN_ZONES
|
|
return None
|