Use DeviceInfo in zwave-js (#58649)

This commit is contained in:
epenet 2021-10-29 02:00:12 +02:00 committed by GitHub
parent e10bd39827
commit c8cbd0070d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 7 deletions

View file

@ -10,7 +10,7 @@ from zwave_js_server.model.value import Value as ZwaveValue, get_value_id
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import callback from homeassistant.core import callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import DeviceInfo, Entity
from .const import DOMAIN from .const import DOMAIN
from .discovery import ZwaveDiscoveryInfo from .discovery import ZwaveDiscoveryInfo
@ -54,9 +54,9 @@ class ZWaveBaseEntity(Entity):
) )
self._attr_assumed_state = self.info.assumed_state self._attr_assumed_state = self.info.assumed_state
# device is precreated in main handler # device is precreated in main handler
self._attr_device_info = { self._attr_device_info = DeviceInfo(
"identifiers": {get_device_id(self.client, self.info.node)}, identifiers={get_device_id(self.client, self.info.node)},
} )
@callback @callback
def on_value_update(self) -> None: def on_value_update(self) -> None:

View file

@ -47,6 +47,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_platform from homeassistant.helpers import entity_platform
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import DeviceInfo
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import ( from .const import (
@ -476,9 +477,9 @@ class ZWaveNodeStatusSensor(SensorEntity):
f"{self.client.driver.controller.home_id}.{node.node_id}.node_status" f"{self.client.driver.controller.home_id}.{node.node_id}.node_status"
) )
# device is precreated in main handler # device is precreated in main handler
self._attr_device_info = { self._attr_device_info = DeviceInfo(
"identifiers": {get_device_id(self.client, self.node)}, identifiers={get_device_id(self.client, self.node)},
} )
self._attr_native_value: str = node.status.name.lower() self._attr_native_value: str = node.status.name.lower()
async def async_poll_value(self, _: bool) -> None: async def async_poll_value(self, _: bool) -> None: