70d4b4d20d
* Add device tracker option * Update const name to be more understandable * Clean up sensor code * Clean up init and coordinator * Add tests and update util function and it's usage * Switch to using the registry entry and add tests * Clean up code * Consolidate duplicate code and adjust tests * Fix runtime error * Fix blocking of the event loop * Adjust API object handling * Update homeassistant/components/dwd_weather_warnings/exceptions.py * Optimize coordinator data update --------- Co-authored-by: Erik Montnemery <erik@montnemery.com>
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
"""Util functions for the dwd_weather_warnings integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.const import ATTR_LATITUDE, ATTR_LONGITUDE
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import entity_registry as er
|
|
|
|
from .exceptions import EntityNotFoundError
|
|
|
|
|
|
def get_position_data(
|
|
hass: HomeAssistant, registry_id: str
|
|
) -> tuple[float, float] | None:
|
|
"""Extract longitude and latitude from a device tracker."""
|
|
registry = er.async_get(hass)
|
|
registry_entry = registry.async_get(registry_id)
|
|
if registry_entry is None:
|
|
raise EntityNotFoundError(f"Failed to find registry entry {registry_id}")
|
|
|
|
entity = hass.states.get(registry_entry.entity_id)
|
|
if entity is None:
|
|
raise EntityNotFoundError(f"Failed to find entity {registry_entry.entity_id}")
|
|
|
|
latitude = entity.attributes.get(ATTR_LATITUDE)
|
|
if not latitude:
|
|
raise AttributeError(
|
|
f"Failed to find attribute '{ATTR_LATITUDE}' in {registry_entry.entity_id}",
|
|
ATTR_LATITUDE,
|
|
)
|
|
|
|
longitude = entity.attributes.get(ATTR_LONGITUDE)
|
|
if not longitude:
|
|
raise AttributeError(
|
|
f"Failed to find attribute '{ATTR_LONGITUDE}' in {registry_entry.entity_id}",
|
|
ATTR_LONGITUDE,
|
|
)
|
|
|
|
return (latitude, longitude)
|