From 107bd11a31b645467fc64094e9a6c7f8c7874d51 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Thu, 28 Oct 2021 23:58:33 +0200
Subject: [PATCH] Use DeviceInfo in wemo (#58638)
---
homeassistant/components/wemo/light.py | 17 +++++++++--------
homeassistant/components/wemo/wemo_device.py | 15 ++++++++-------
2 files changed, 17 insertions(+), 15 deletions(-)
diff --git a/homeassistant/components/wemo/light.py b/homeassistant/components/wemo/light.py
index 980675bd4ff..c46a4e78440 100644
--- a/homeassistant/components/wemo/light.py
+++ b/homeassistant/components/wemo/light.py
@@ -17,6 +17,7 @@ from homeassistant.components.light import (
from homeassistant.core import callback
from homeassistant.helpers.device_registry import CONNECTION_ZIGBEE
from homeassistant.helpers.dispatcher import async_dispatcher_connect
+from homeassistant.helpers.entity import DeviceInfo
import homeassistant.util.color as color_util
from .const import DOMAIN as WEMO_DOMAIN
@@ -99,15 +100,15 @@ class WemoLight(WemoEntity, LightEntity):
return self.light.uniqueID
@property
- def device_info(self):
+ def device_info(self) -> DeviceInfo:
"""Return the device info."""
- return {
- "name": self.name,
- "connections": {(CONNECTION_ZIGBEE, self._unique_id)},
- "identifiers": {(WEMO_DOMAIN, self._unique_id)},
- "model": self._model_name,
- "manufacturer": "Belkin",
- }
+ return DeviceInfo(
+ connections={(CONNECTION_ZIGBEE, self._unique_id)},
+ identifiers={(WEMO_DOMAIN, self._unique_id)},
+ manufacturer="Belkin",
+ model=self._model_name,
+ name=self.name,
+ )
@property
def brightness(self):
diff --git a/homeassistant/components/wemo/wemo_device.py b/homeassistant/components/wemo/wemo_device.py
index 1690d30e082..a4f20eb55f5 100644
--- a/homeassistant/components/wemo/wemo_device.py
+++ b/homeassistant/components/wemo/wemo_device.py
@@ -17,6 +17,7 @@ from homeassistant.const import (
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.device_registry import async_get as async_get_device_registry
+from homeassistant.helpers.entity import DeviceInfo
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
from .const import DOMAIN, WEMO_SUBSCRIPTION_EVENT
@@ -120,13 +121,13 @@ class DeviceCoordinator(DataUpdateCoordinator):
raise UpdateFailed("WeMo update failed") from err
-def _device_info(wemo: WeMoDevice):
- return {
- "name": wemo.name,
- "identifiers": {(DOMAIN, wemo.serialnumber)},
- "model": wemo.model_name,
- "manufacturer": "Belkin",
- }
+def _device_info(wemo: WeMoDevice) -> DeviceInfo:
+ return DeviceInfo(
+ identifiers={(DOMAIN, wemo.serialnumber)},
+ manufacturer="Belkin",
+ model=wemo.model_name,
+ name=wemo.name,
+ )
async def async_register_device(
--
GitLab