From 27073abc62dff126e23188480566aa249b3520cf Mon Sep 17 00:00:00 2001
From: epenet <epenet@users.noreply.github.com>
Date: Mon, 10 Jan 2022 08:13:58 +0000
Subject: [PATCH] Add setup type hints to envisalink

---
 .../envisalink/alarm_control_panel.py           | 17 +++++++++++++----
 .../components/envisalink/binary_sensor.py      | 15 +++++++++++++--
 homeassistant/components/envisalink/sensor.py   | 15 +++++++++++++--
 3 files changed, 39 insertions(+), 8 deletions(-)

diff --git a/homeassistant/components/envisalink/alarm_control_panel.py b/homeassistant/components/envisalink/alarm_control_panel.py
index d93e980064d..e6b54e586f8 100644
--- a/homeassistant/components/envisalink/alarm_control_panel.py
+++ b/homeassistant/components/envisalink/alarm_control_panel.py
@@ -1,4 +1,6 @@
 """Support for Envisalink-based alarm control panels (Honeywell/DSC)."""
+from __future__ import annotations
+
 import logging
 
 import voluptuous as vol
@@ -24,9 +26,11 @@ from homeassistant.const import (
     STATE_ALARM_TRIGGERED,
     STATE_UNKNOWN,
 )
-from homeassistant.core import ServiceCall, callback
+from homeassistant.core import HomeAssistant, ServiceCall, callback
 import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
 from . import (
     CONF_PANIC,
@@ -51,8 +55,15 @@ ALARM_KEYPRESS_SCHEMA = vol.Schema(
 )
 
 
-async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
+async def async_setup_platform(
+    hass: HomeAssistant,
+    config: ConfigType,
+    async_add_entities: AddEntitiesCallback,
+    discovery_info: DiscoveryInfoType | None = None,
+) -> None:
     """Perform the setup for Envisalink alarm panels."""
+    if not discovery_info:
+        return
     configured_partitions = discovery_info["partitions"]
     code = discovery_info[CONF_CODE]
     panic_type = discovery_info[CONF_PANIC]
@@ -93,8 +104,6 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
         schema=ALARM_KEYPRESS_SCHEMA,
     )
 
-    return True
-
 
 class EnvisalinkAlarm(EnvisalinkDevice, AlarmControlPanelEntity):
     """Representation of an Envisalink-based alarm panel."""
diff --git a/homeassistant/components/envisalink/binary_sensor.py b/homeassistant/components/envisalink/binary_sensor.py
index 22089ee7907..0c2a5ed325c 100644
--- a/homeassistant/components/envisalink/binary_sensor.py
+++ b/homeassistant/components/envisalink/binary_sensor.py
@@ -1,11 +1,15 @@
 """Support for Envisalink zone states- represented as binary sensors."""
+from __future__ import annotations
+
 import datetime
 import logging
 
 from homeassistant.components.binary_sensor import BinarySensorEntity
 from homeassistant.const import ATTR_LAST_TRIP_TIME
-from homeassistant.core import callback
+from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 from homeassistant.util import dt as dt_util
 
 from . import (
@@ -20,8 +24,15 @@ from . import (
 _LOGGER = logging.getLogger(__name__)
 
 
-async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
+async def async_setup_platform(
+    hass: HomeAssistant,
+    config: ConfigType,
+    async_add_entities: AddEntitiesCallback,
+    discovery_info: DiscoveryInfoType | None = None,
+) -> None:
     """Set up the Envisalink binary sensor devices."""
+    if not discovery_info:
+        return
     configured_zones = discovery_info["zones"]
 
     devices = []
diff --git a/homeassistant/components/envisalink/sensor.py b/homeassistant/components/envisalink/sensor.py
index 88aa7fa988c..51526cd000c 100644
--- a/homeassistant/components/envisalink/sensor.py
+++ b/homeassistant/components/envisalink/sensor.py
@@ -1,9 +1,13 @@
 """Support for Envisalink sensors (shows panel info)."""
+from __future__ import annotations
+
 import logging
 
 from homeassistant.components.sensor import SensorEntity
-from homeassistant.core import callback
+from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
 from . import (
     CONF_PARTITIONNAME,
@@ -17,8 +21,15 @@ from . import (
 _LOGGER = logging.getLogger(__name__)
 
 
-async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
+async def async_setup_platform(
+    hass: HomeAssistant,
+    config: ConfigType,
+    async_add_entities: AddEntitiesCallback,
+    discovery_info: DiscoveryInfoType | None = None,
+) -> None:
     """Perform the setup for Envisalink sensor devices."""
+    if not discovery_info:
+        return
     configured_partitions = discovery_info["partitions"]
 
     devices = []