From 135a718a0e92718949b36241c94ab0c8741e9c3f Mon Sep 17 00:00:00 2001
From: Marc Mueller <30130371+cdce8p@users.noreply.github.com>
Date: Sat, 6 Jan 2024 10:33:22 +0100
Subject: [PATCH] Enable strict typing for energyzero (#107300)

---
 .strict-typing                                  |  1 +
 homeassistant/components/energyzero/services.py |  3 ++-
 mypy.ini                                        | 10 ++++++++++
 3 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/.strict-typing b/.strict-typing
index 0db793ace7a..6eeb39ceddb 100644
--- a/.strict-typing
+++ b/.strict-typing
@@ -153,6 +153,7 @@ homeassistant.components.elgato.*
 homeassistant.components.elkm1.*
 homeassistant.components.emulated_hue.*
 homeassistant.components.energy.*
+homeassistant.components.energyzero.*
 homeassistant.components.enigma2.*
 homeassistant.components.esphome.*
 homeassistant.components.event.*
diff --git a/homeassistant/components/energyzero/services.py b/homeassistant/components/energyzero/services.py
index d8e548c22f8..325c443375e 100644
--- a/homeassistant/components/energyzero/services.py
+++ b/homeassistant/components/energyzero/services.py
@@ -109,7 +109,8 @@ def __get_coordinator(
             },
         )
 
-    return hass.data[DOMAIN][entry_id]
+    coordinator: EnergyZeroDataUpdateCoordinator = hass.data[DOMAIN][entry_id]
+    return coordinator
 
 
 async def __get_prices(
diff --git a/mypy.ini b/mypy.ini
index 30b4ee9f048..fc020a7b209 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -1291,6 +1291,16 @@ disallow_untyped_defs = true
 warn_return_any = true
 warn_unreachable = true
 
+[mypy-homeassistant.components.energyzero.*]
+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.enigma2.*]
 check_untyped_defs = true
 disallow_incomplete_defs = true