Use runtime_data in ccm15 (#136378)

This commit is contained in:
epenet 2025-01-24 01:02:38 +01:00 committed by GitHub
parent c691f8cc1e
commit 3bbcd37ec8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 19 additions and 20 deletions

View File

@ -2,34 +2,30 @@
from __future__ import annotations
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_HOST, CONF_PORT, Platform
from homeassistant.core import HomeAssistant
from .const import DOMAIN
from .coordinator import CCM15Coordinator
from .coordinator import CCM15ConfigEntry, CCM15Coordinator
PLATFORMS: list[Platform] = [Platform.CLIMATE]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_setup_entry(hass: HomeAssistant, entry: CCM15ConfigEntry) -> bool:
"""Set up Midea ccm15 AC Controller from a config entry."""
coordinator = CCM15Coordinator(
hass,
entry,
entry.data[CONF_HOST],
entry.data[CONF_PORT],
)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator
entry.runtime_data = coordinator
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
return True
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_unload_entry(hass: HomeAssistant, entry: CCM15ConfigEntry) -> bool:
"""Unload a config entry."""
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
hass.data[DOMAIN].pop(entry.entry_id)
return unload_ok
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)

View File

@ -17,7 +17,6 @@ from homeassistant.components.climate import (
ClimateEntityFeature,
HVACMode,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, UnitOfTemperature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.device_registry import DeviceInfo
@ -25,18 +24,18 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.update_coordinator import CoordinatorEntity
from .const import CONST_CMD_FAN_MAP, CONST_CMD_STATE_MAP, DOMAIN
from .coordinator import CCM15Coordinator
from .coordinator import CCM15ConfigEntry, CCM15Coordinator
_LOGGER = logging.getLogger(__name__)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
config_entry: CCM15ConfigEntry,
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up all climate."""
coordinator: CCM15Coordinator = hass.data[DOMAIN][config_entry.entry_id]
coordinator = config_entry.runtime_data
ac_data: CCM15DeviceState = coordinator.data
entities = [

View File

@ -7,6 +7,7 @@ from ccm15 import CCM15Device, CCM15DeviceState, CCM15SlaveDevice
import httpx
from homeassistant.components.climate import HVACMode
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@ -19,15 +20,20 @@ from .const import (
_LOGGER = logging.getLogger(__name__)
type CCM15ConfigEntry = ConfigEntry[CCM15Coordinator]
class CCM15Coordinator(DataUpdateCoordinator[CCM15DeviceState]):
"""Class to coordinate multiple CCM15Climate devices."""
def __init__(self, hass: HomeAssistant, host: str, port: int) -> None:
def __init__(
self, hass: HomeAssistant, entry: CCM15ConfigEntry, host: str, port: int
) -> None:
"""Initialize the coordinator."""
super().__init__(
hass,
_LOGGER,
config_entry=entry,
name=host,
update_interval=datetime.timedelta(seconds=DEFAULT_INTERVAL),
)

View File

@ -4,18 +4,16 @@ from __future__ import annotations
from typing import Any
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from .const import DOMAIN
from .coordinator import CCM15Coordinator
from .coordinator import CCM15ConfigEntry
async def async_get_config_entry_diagnostics(
hass: HomeAssistant, config_entry: ConfigEntry
hass: HomeAssistant, config_entry: CCM15ConfigEntry
) -> dict[str, Any]:
"""Return diagnostics for a config entry."""
coordinator: CCM15Coordinator = hass.data[DOMAIN][config_entry.entry_id]
coordinator = config_entry.runtime_data
return {
str(device_id): {