From 7068986c14ed383de57a17b4be50abbaebca23e6 Mon Sep 17 00:00:00 2001
From: Dan Raper <me@danr.uk>
Date: Tue, 1 Apr 2025 14:48:45 +0100
Subject: [PATCH] Bump Ohme to platinum (#141762)

* Bump version of ohmepy and fix types

* Add strict typing to ohme

* Inject websession for ohme

* CI/code formatting fixes for ohme

* Update mypy.ini for ohme

* Fix typing in services for ohme

* Bump ohme quality in manifest
---
 .strict-typing                                   |  1 +
 homeassistant/components/ohme/__init__.py        |  7 ++++++-
 homeassistant/components/ohme/manifest.json      |  2 +-
 homeassistant/components/ohme/quality_scale.yaml |  6 +++---
 homeassistant/components/ohme/services.py        |  5 +++--
 mypy.ini                                         | 10 ++++++++++
 6 files changed, 24 insertions(+), 7 deletions(-)

diff --git a/.strict-typing b/.strict-typing
index e0c4e569f4b..3e8ad0ddbaf 100644
--- a/.strict-typing
+++ b/.strict-typing
@@ -364,6 +364,7 @@ homeassistant.components.notify.*
 homeassistant.components.notion.*
 homeassistant.components.number.*
 homeassistant.components.nut.*
+homeassistant.components.ohme.*
 homeassistant.components.onboarding.*
 homeassistant.components.oncue.*
 homeassistant.components.onedrive.*
diff --git a/homeassistant/components/ohme/__init__.py b/homeassistant/components/ohme/__init__.py
index e3e252cbf8b..c304bfdf72d 100644
--- a/homeassistant/components/ohme/__init__.py
+++ b/homeassistant/components/ohme/__init__.py
@@ -6,6 +6,7 @@ from homeassistant.const import CONF_EMAIL, CONF_PASSWORD
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
 from homeassistant.helpers import config_validation as cv
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
 from homeassistant.helpers.typing import ConfigType
 
 from .const import DOMAIN, PLATFORMS
@@ -31,7 +32,11 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
 async def async_setup_entry(hass: HomeAssistant, entry: OhmeConfigEntry) -> bool:
     """Set up Ohme from a config entry."""
 
-    client = OhmeApiClient(entry.data[CONF_EMAIL], entry.data[CONF_PASSWORD])
+    client = OhmeApiClient(
+        email=entry.data[CONF_EMAIL],
+        password=entry.data[CONF_PASSWORD],
+        session=async_get_clientsession(hass),
+    )
 
     try:
         await client.async_login()
diff --git a/homeassistant/components/ohme/manifest.json b/homeassistant/components/ohme/manifest.json
index 30a55360ce2..786c615d68a 100644
--- a/homeassistant/components/ohme/manifest.json
+++ b/homeassistant/components/ohme/manifest.json
@@ -6,6 +6,6 @@
   "documentation": "https://www.home-assistant.io/integrations/ohme/",
   "integration_type": "device",
   "iot_class": "cloud_polling",
-  "quality_scale": "silver",
+  "quality_scale": "platinum",
   "requirements": ["ohme==1.5.1"]
 }
diff --git a/homeassistant/components/ohme/quality_scale.yaml b/homeassistant/components/ohme/quality_scale.yaml
index 12473a08edd..2f7aece5bb6 100644
--- a/homeassistant/components/ohme/quality_scale.yaml
+++ b/homeassistant/components/ohme/quality_scale.yaml
@@ -75,6 +75,6 @@ rules:
     comment: |
       Not supported by the API. Accounts and devices have a one-to-one relationship.
   # Platinum
-  async-dependency: todo
-  inject-websession: todo
-  strict-typing: todo
+  async-dependency: done
+  inject-websession: done
+  strict-typing: done
diff --git a/homeassistant/components/ohme/services.py b/homeassistant/components/ohme/services.py
index 249fb1abdab..8ed29aa373d 100644
--- a/homeassistant/components/ohme/services.py
+++ b/homeassistant/components/ohme/services.py
@@ -5,7 +5,7 @@ from typing import Final
 from ohme import OhmeApiClient
 import voluptuous as vol
 
-from homeassistant.config_entries import ConfigEntry, ConfigEntryState
+from homeassistant.config_entries import ConfigEntryState
 from homeassistant.core import (
     HomeAssistant,
     ServiceCall,
@@ -16,6 +16,7 @@ from homeassistant.exceptions import ServiceValidationError
 from homeassistant.helpers import selector
 
 from .const import DOMAIN
+from .coordinator import OhmeConfigEntry
 
 ATTR_CONFIG_ENTRY: Final = "config_entry"
 ATTR_PRICE_CAP: Final = "price_cap"
@@ -47,7 +48,7 @@ SERVICE_SET_PRICE_CAP_SCHEMA: Final = vol.Schema(
 def __get_client(call: ServiceCall) -> OhmeApiClient:
     """Get the client from the config entry."""
     entry_id: str = call.data[ATTR_CONFIG_ENTRY]
-    entry: ConfigEntry | None = call.hass.config_entries.async_get_entry(entry_id)
+    entry: OhmeConfigEntry | None = call.hass.config_entries.async_get_entry(entry_id)
 
     if not entry:
         raise ServiceValidationError(
diff --git a/mypy.ini b/mypy.ini
index 9831a183ec4..685412e6e98 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -3396,6 +3396,16 @@ disallow_untyped_defs = true
 warn_return_any = true
 warn_unreachable = true
 
+[mypy-homeassistant.components.ohme.*]
+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.onboarding.*]
 check_untyped_defs = true
 disallow_incomplete_defs = true