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

Add DeviceInfo to OTP integration (#122392)

Browse Source
This commit is contained in:
Mr. Bubbles 2024-07-22 21:39:22 +02:00 committed by GitHub
parent c61efe931a
commit fed17a4905
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/components/otp/sensor.py
View File

@ -15,6 +15,7 @@ from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
from homeassistant.const import CONF_NAME, CONF_TOKEN from homeassistant.const import CONF_NAME, CONF_TOKEN
from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant, callback from homeassistant.core import DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant, callback
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue from homeassistant.helpers.issue_registry import IssueSeverity, async_create_issue
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, StateType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, StateType
@ -76,13 +77,20 @@ class TOTPSensor(SensorEntity):
_attr_should_poll = False _attr_should_poll = False
_attr_native_value: StateType = None _attr_native_value: StateType = None
_next_expiration: float | None = None _next_expiration: float | None = None
_attr_has_entity_name = True
_attr_name = None
def __init__(self, name: str, token: str, entry_id: str) -> None: def __init__(self, name: str, token: str, entry_id: str) -> None:
"""Initialize the sensor.""" """Initialize the sensor."""
self._attr_name = name
self._attr_unique_id = entry_id self._attr_unique_id = entry_id
self._otp = pyotp.TOTP(token) self._otp = pyotp.TOTP(token)
self.device_info = DeviceInfo(
name=name,
entry_type=DeviceEntryType.SERVICE,
identifiers={(DOMAIN, entry_id)},
)
async def async_added_to_hass(self) -> None: async def async_added_to_hass(self) -> None:
"""Handle when an entity is about to be added to Home Assistant.""" """Handle when an entity is about to be added to Home Assistant."""
self._call_loop() self._call_loop()