Move overkiz climate definitions (#125741)

homeassistant/components/overkiz/climate.py

@@ -1,62 +0,0 @@
-"""Support for Overkiz climate devices."""
-
-from __future__ import annotations
-
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import Platform
-from homeassistant.core import HomeAssistant
-from homeassistant.helpers.entity import Entity
-from homeassistant.helpers.entity_platform import AddEntitiesCallback
-
-from . import HomeAssistantOverkizData
-from .climate_entities import (
-    WIDGET_AND_CONTROLLABLE_TO_CLIMATE_ENTITY,
-    WIDGET_AND_PROTOCOL_TO_CLIMATE_ENTITY,
-    WIDGET_TO_CLIMATE_ENTITY,
-)
-from .const import DOMAIN
-
-
-async def async_setup_entry(
-    hass: HomeAssistant,
-    entry: ConfigEntry,
-    async_add_entities: AddEntitiesCallback,
-) -> None:
-    """Set up the Overkiz climate from a config entry."""
-    data: HomeAssistantOverkizData = hass.data[DOMAIN][entry.entry_id]
-
-    # Match devices based on the widget.
-    entities_based_on_widget: list[Entity] = [
-        WIDGET_TO_CLIMATE_ENTITY[device.widget](device.device_url, data.coordinator)
-        for device in data.platforms[Platform.CLIMATE]
-        if device.widget in WIDGET_TO_CLIMATE_ENTITY
-    ]
-
-    # Match devices based on the widget and controllableName.
-    # ie Atlantic APC
-    entities_based_on_widget_and_controllable: list[Entity] = [
-        WIDGET_AND_CONTROLLABLE_TO_CLIMATE_ENTITY[device.widget][
-            device.controllable_name
-        ](device.device_url, data.coordinator)
-        for device in data.platforms[Platform.CLIMATE]
-        if device.widget in WIDGET_AND_CONTROLLABLE_TO_CLIMATE_ENTITY
-        and device.controllable_name
-        in WIDGET_AND_CONTROLLABLE_TO_CLIMATE_ENTITY[device.widget]
-    ]
-
-    # Match devices based on the widget and protocol.
-    # #ie Hitachi Air To Air Heat Pumps
-    entities_based_on_widget_and_protocol: list[Entity] = [
-        WIDGET_AND_PROTOCOL_TO_CLIMATE_ENTITY[device.widget][device.protocol](
-            device.device_url, data.coordinator
-        )
-        for device in data.platforms[Platform.CLIMATE]
-        if device.widget in WIDGET_AND_PROTOCOL_TO_CLIMATE_ENTITY
-        and device.protocol in WIDGET_AND_PROTOCOL_TO_CLIMATE_ENTITY[device.widget]
-    ]
-
-    async_add_entities(
-        entities_based_on_widget
-        + entities_based_on_widget_and_controllable
-        + entities_based_on_widget_and_protocol
-    )

homeassistant/components/overkiz/climate/__init__.py

@@ -1,10 +1,20 @@
 """Climate entities for the Overkiz (by Somfy) integration."""
 
+from __future__ import annotations
+
 from enum import StrEnum, unique
 
 from pyoverkiz.enums import Protocol
 from pyoverkiz.enums.ui import UIWidget
 
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import Platform
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import Entity
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .. import HomeAssistantOverkizData
+from ..const import DOMAIN
 from .atlantic_electrical_heater import AtlanticElectricalHeater
 from .atlantic_electrical_heater_with_adjustable_temperature_setpoint import (
     AtlanticElectricalHeaterWithAdjustableTemperatureSetpoint,
@@ -65,3 +75,48 @@ WIDGET_AND_PROTOCOL_TO_CLIMATE_ENTITY = {
         Protocol.OVP: HitachiAirToAirHeatPumpOVP,
     },
 }
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    entry: ConfigEntry,
+    async_add_entities: AddEntitiesCallback,
+) -> None:
+    """Set up the Overkiz climate from a config entry."""
+    data: HomeAssistantOverkizData = hass.data[DOMAIN][entry.entry_id]
+
+    # Match devices based on the widget.
+    entities_based_on_widget: list[Entity] = [
+        WIDGET_TO_CLIMATE_ENTITY[device.widget](device.device_url, data.coordinator)
+        for device in data.platforms[Platform.CLIMATE]
+        if device.widget in WIDGET_TO_CLIMATE_ENTITY
+    ]
+
+    # Match devices based on the widget and controllableName.
+    # ie Atlantic APC
+    entities_based_on_widget_and_controllable: list[Entity] = [
+        WIDGET_AND_CONTROLLABLE_TO_CLIMATE_ENTITY[device.widget][
+            device.controllable_name
+        ](device.device_url, data.coordinator)
+        for device in data.platforms[Platform.CLIMATE]
+        if device.widget in WIDGET_AND_CONTROLLABLE_TO_CLIMATE_ENTITY
+        and device.controllable_name
+        in WIDGET_AND_CONTROLLABLE_TO_CLIMATE_ENTITY[device.widget]
+    ]
+
+    # Match devices based on the widget and protocol.
+    # #ie Hitachi Air To Air Heat Pumps
+    entities_based_on_widget_and_protocol: list[Entity] = [
+        WIDGET_AND_PROTOCOL_TO_CLIMATE_ENTITY[device.widget][device.protocol](
+            device.device_url, data.coordinator
+        )
+        for device in data.platforms[Platform.CLIMATE]
+        if device.widget in WIDGET_AND_PROTOCOL_TO_CLIMATE_ENTITY
+        and device.protocol in WIDGET_AND_PROTOCOL_TO_CLIMATE_ENTITY[device.widget]
+    ]
+
+    async_add_entities(
+        entities_based_on_widget
+        + entities_based_on_widget_and_controllable
+        + entities_based_on_widget_and_protocol
+    )