From 79b3865b6014fe1fb45dea5d792cdc3d1e44ed91 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Tue, 28 Jun 2022 13:15:38 +0200
Subject: [PATCH] Use attributes in ialarm alarm (#74099)

---
 homeassistant/components/ialarm/__init__.py   | 12 ++++---
 .../components/ialarm/alarm_control_panel.py  | 35 +++++++++----------
 2 files changed, 23 insertions(+), 24 deletions(-)

diff --git a/homeassistant/components/ialarm/__init__.py b/homeassistant/components/ialarm/__init__.py
index db9aa000066..254bf6f685f 100644
--- a/homeassistant/components/ialarm/__init__.py
+++ b/homeassistant/components/ialarm/__init__.py
@@ -1,4 +1,6 @@
 """iAlarm integration."""
+from __future__ import annotations
+
 import asyncio
 import logging
 
@@ -20,8 +22,8 @@ _LOGGER = logging.getLogger(__name__)
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up iAlarm config."""
-    host = entry.data[CONF_HOST]
-    port = entry.data[CONF_PORT]
+    host: str = entry.data[CONF_HOST]
+    port: int = entry.data[CONF_PORT]
     ialarm = IAlarm(host, port)
 
     try:
@@ -55,11 +57,11 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
 class IAlarmDataUpdateCoordinator(DataUpdateCoordinator):
     """Class to manage fetching iAlarm data."""
 
-    def __init__(self, hass, ialarm, mac):
+    def __init__(self, hass: HomeAssistant, ialarm: IAlarm, mac: str) -> None:
         """Initialize global iAlarm data updater."""
         self.ialarm = ialarm
-        self.state = None
-        self.host = ialarm.host
+        self.state: str | None = None
+        self.host: str = ialarm.host
         self.mac = mac
 
         super().__init__(
diff --git a/homeassistant/components/ialarm/alarm_control_panel.py b/homeassistant/components/ialarm/alarm_control_panel.py
index 74310d940e7..6a4e3d191eb 100644
--- a/homeassistant/components/ialarm/alarm_control_panel.py
+++ b/homeassistant/components/ialarm/alarm_control_panel.py
@@ -11,6 +11,7 @@ from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
+from . import IAlarmDataUpdateCoordinator
 from .const import DATA_COORDINATOR, DOMAIN
 
 
@@ -18,39 +19,35 @@ async def async_setup_entry(
     hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
 ) -> None:
     """Set up a iAlarm alarm control panel based on a config entry."""
-    coordinator = hass.data[DOMAIN][entry.entry_id][DATA_COORDINATOR]
+    coordinator: IAlarmDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][
+        DATA_COORDINATOR
+    ]
     async_add_entities([IAlarmPanel(coordinator)], False)
 
 
-class IAlarmPanel(CoordinatorEntity, AlarmControlPanelEntity):
+class IAlarmPanel(
+    CoordinatorEntity[IAlarmDataUpdateCoordinator], AlarmControlPanelEntity
+):
     """Representation of an iAlarm device."""
 
+    _attr_name = "iAlarm"
     _attr_supported_features = (
         AlarmControlPanelEntityFeature.ARM_HOME
         | AlarmControlPanelEntityFeature.ARM_AWAY
     )
 
-    @property
-    def device_info(self) -> DeviceInfo:
-        """Return device info for this device."""
-        return DeviceInfo(
-            identifiers={(DOMAIN, self.unique_id)},
+    def __init__(self, coordinator: IAlarmDataUpdateCoordinator) -> None:
+        """Create the entity with a DataUpdateCoordinator."""
+        super().__init__(coordinator)
+        self._attr_device_info = DeviceInfo(
+            identifiers={(DOMAIN, coordinator.mac)},
             manufacturer="Antifurto365 - Meian",
-            name=self.name,
+            name="iAlarm",
         )
+        self._attr_unique_id = coordinator.mac
 
     @property
-    def unique_id(self):
-        """Return a unique id."""
-        return self.coordinator.mac
-
-    @property
-    def name(self):
-        """Return the name."""
-        return "iAlarm"
-
-    @property
-    def state(self):
+    def state(self) -> str | None:
         """Return the state of the device."""
         return self.coordinator.state