From a831c375110019f3bd1f1140b05483ae38411dcd Mon Sep 17 00:00:00 2001
From: David Knowles <dknowles2@gmail.com>
Date: Thu, 28 Nov 2024 02:29:15 -0500
Subject: [PATCH] Enable strict typing for Schlage (#131734)

---
 .strict-typing                                  |  1 +
 homeassistant/components/schlage/coordinator.py |  4 +---
 mypy.ini                                        | 10 ++++++++++
 3 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/.strict-typing b/.strict-typing
index cb0cab984ee..f1383fa3528 100644
--- a/.strict-typing
+++ b/.strict-typing
@@ -405,6 +405,7 @@ homeassistant.components.ruuvitag_ble.*
 homeassistant.components.samsungtv.*
 homeassistant.components.scene.*
 homeassistant.components.schedule.*
+homeassistant.components.schlage.*
 homeassistant.components.scrape.*
 homeassistant.components.script.*
 homeassistant.components.search.*
diff --git a/homeassistant/components/schlage/coordinator.py b/homeassistant/components/schlage/coordinator.py
index 53bb43751a9..5d525e3c842 100644
--- a/homeassistant/components/schlage/coordinator.py
+++ b/homeassistant/components/schlage/coordinator.py
@@ -44,6 +44,7 @@ class SchlageDataUpdateCoordinator(DataUpdateCoordinator[SchlageData]):
         super().__init__(
             hass, LOGGER, name=f"{DOMAIN} ({username})", update_interval=UPDATE_INTERVAL
         )
+        self.data = SchlageData(locks={})
         self.api = api
         self.new_locks_callbacks: list[Callable[[dict[str, LockData]], None]] = []
         self.async_add_listener(self._add_remove_locks)
@@ -83,9 +84,6 @@ class SchlageDataUpdateCoordinator(DataUpdateCoordinator[SchlageData]):
     @callback
     def _add_remove_locks(self) -> None:
         """Add newly discovered locks and remove nonexistent locks."""
-        if self.data is None:
-            return
-
         device_registry = dr.async_get(self.hass)
         devices = dr.async_entries_for_config_entry(
             device_registry, self.config_entry.entry_id
diff --git a/mypy.ini b/mypy.ini
index a71f980dac9..aa9a2b81095 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -3806,6 +3806,16 @@ disallow_untyped_defs = true
 warn_return_any = true
 warn_unreachable = true
 
+[mypy-homeassistant.components.schlage.*]
+check_untyped_defs = true
+disallow_incomplete_defs = true
+disallow_subclassing_any = true
+disallow_untyped_calls = true
+disallow_untyped_decorators = true
+disallow_untyped_defs = true
+warn_return_any = true
+warn_unreachable = true
+
 [mypy-homeassistant.components.scrape.*]
 check_untyped_defs = true
 disallow_incomplete_defs = true