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

Use shorthand attributes in yandex transport sensor (#145225)

Browse Source
This commit is contained in:
epenet 2025-05-19 15:58:34 +02:00 committed by GitHub
parent 05795d0ad8
commit e3d2f917e2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
homeassistant/components/yandex_transport/sensor.py
View File

@ -4,6 +4,7 @@ from __future__ import annotations
from datetime import timedelta from datetime import timedelta
import logging import logging
from typing import Any
from aioymaps import CaptchaError, NoSessionError, YandexMapsRequester from aioymaps import CaptchaError, NoSessionError, YandexMapsRequester
import voluptuous as vol import voluptuous as vol
@ -71,6 +72,7 @@ class DiscoverYandexTransport(SensorEntity):
"""Implementation of yandex_transport sensor.""" """Implementation of yandex_transport sensor."""
_attr_attribution = "Data provided by maps.yandex.ru" _attr_attribution = "Data provided by maps.yandex.ru"
_attr_device_class = SensorDeviceClass.TIMESTAMP
_attr_icon = "mdi:bus" _attr_icon = "mdi:bus"
def __init__(self, requester: YandexMapsRequester, stop_id, routes, name) -> None: def __init__(self, requester: YandexMapsRequester, stop_id, routes, name) -> None:
@ -78,13 +80,15 @@ class DiscoverYandexTransport(SensorEntity):
self.requester = requester self.requester = requester
self._stop_id = stop_id self._stop_id = stop_id
self._routes = routes self._routes = routes
self._state = None self._attr_name = name
self._name = name
self._attrs = None
async def async_update(self, *, tries=0): async def async_update(self) -> None:
"""Get the latest data from maps.yandex.ru and update the states.""" """Get the latest data from maps.yandex.ru and update the states."""
attrs = {} await self._try_update(tries=0)
async def _try_update(self, *, tries: int) -> None:
"""Get the latest data from maps.yandex.ru and update the states."""
attrs: dict[str, Any] = {}
closer_time = None closer_time = None
try: try:
yandex_reply = await self.requester.get_stop_info(self._stop_id) yandex_reply = await self.requester.get_stop_info(self._stop_id)
@ -108,7 +112,7 @@ class DiscoverYandexTransport(SensorEntity):
if tries > 0: if tries > 0:
return return
await self.requester.set_new_session() await self.requester.set_new_session()
await self.async_update(tries=tries + 1) await self._try_update(tries=tries + 1)
return return
stop_name = data["name"] stop_name = data["name"]
@ -146,27 +150,9 @@ class DiscoverYandexTransport(SensorEntity):
attrs[STOP_NAME] = stop_name attrs[STOP_NAME] = stop_name
if closer_time is None: if closer_time is None:
self._state = None self._attr_native_value = None
else: else:
self._state = dt_util.utc_from_timestamp(closer_time).replace(microsecond=0) self._attr_native_value = dt_util.utc_from_timestamp(closer_time).replace(
self._attrs = attrs microsecond=0
)
@property self._attr_extra_state_attributes = attrs
def native_value(self):
"""Return the state of the sensor."""
return self._state
@property
def device_class(self):
"""Return the device class."""
return SensorDeviceClass.TIMESTAMP
@property
def name(self):
"""Return the name of the sensor."""
return self._name
@property
def extra_state_attributes(self):
"""Return the state attributes."""
return self._attrs