From 1357ed3942e80dfa9ee0bbe91eea9239dc1c3ac1 Mon Sep 17 00:00:00 2001
From: epenet <epenet@users.noreply.github.com>
Date: Wed, 12 Jan 2022 17:25:13 +0000
Subject: [PATCH] Add setup type hints to binary_sensor

---
 .../components/raspihats/binary_sensor.py     | 21 +++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/raspihats/binary_sensor.py b/homeassistant/components/raspihats/binary_sensor.py
index ea3130da4ba..f8fbc0d010f 100644
--- a/homeassistant/components/raspihats/binary_sensor.py
+++ b/homeassistant/components/raspihats/binary_sensor.py
@@ -1,4 +1,6 @@
 """Support for raspihats board binary sensors."""
+from __future__ import annotations
+
 import logging
 
 import voluptuous as vol
@@ -10,7 +12,10 @@ from homeassistant.const import (
     CONF_NAME,
     DEVICE_DEFAULT_NAME,
 )
+from homeassistant.core import HomeAssistant
 import homeassistant.helpers.config_validation as cv
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
 
 from . import (
     CONF_BOARD,
@@ -18,9 +23,11 @@ from . import (
     CONF_I2C_HATS,
     CONF_INDEX,
     CONF_INVERT_LOGIC,
+    DOMAIN,
     I2C_HAT_NAMES,
     I2C_HATS_MANAGER,
     I2CHatsException,
+    I2CHatsManager,
 )
 
 _LOGGER = logging.getLogger(__name__)
@@ -54,15 +61,21 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
 )
 
 
-def setup_platform(hass, config, add_entities, discovery_info=None):
+def setup_platform(
+    hass: HomeAssistant,
+    config: ConfigType,
+    add_entities: AddEntitiesCallback,
+    discovery_info: DiscoveryInfoType | None = None,
+) -> None:
     """Set up the raspihats binary_sensor devices."""
-    I2CHatBinarySensor.I2C_HATS_MANAGER = hass.data[I2C_HATS_MANAGER]
+    I2CHatBinarySensor.I2C_HATS_MANAGER = hass.data[DOMAIN][I2C_HATS_MANAGER]
     binary_sensors = []
-    i2c_hat_configs = config.get(CONF_I2C_HATS)
+    i2c_hat_configs = config.get(CONF_I2C_HATS, [])
     for i2c_hat_config in i2c_hat_configs:
         address = i2c_hat_config[CONF_ADDRESS]
         board = i2c_hat_config[CONF_BOARD]
         try:
+            assert I2CHatBinarySensor.I2C_HATS_MANAGER
             I2CHatBinarySensor.I2C_HATS_MANAGER.register_board(board, address)
             for channel_config in i2c_hat_config[CONF_CHANNELS]:
                 binary_sensors.append(
@@ -84,7 +97,7 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
 class I2CHatBinarySensor(BinarySensorEntity):
     """Representation of a binary sensor that uses a I2C-HAT digital input."""
 
-    I2C_HATS_MANAGER = None
+    I2C_HATS_MANAGER: I2CHatsManager | None = None
 
     def __init__(self, address, channel, name, invert_logic, device_class):
         """Initialize the raspihats sensor."""