jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-08-01 09:38:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor coordinator setup in Iron OS (#149600)

Browse Source
This commit is contained in:
Manu 2025-07-29 09:37:32 +02:00 committed by GitHub
parent 62ee1fbc64
commit 45ec9c7dad
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
homeassistant/components/iron_os/__init__.py
View File

@ -9,9 +9,7 @@ from pynecil import IronOSUpdate, Pynecil
from homeassistant.const import Platform from homeassistant.const import Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.aiohttp_client import async_get_clientsession from homeassistant.helpers.aiohttp_client import async_get_clientsession
from homeassistant.helpers.typing import ConfigType
from homeassistant.util.hass_dict import HassKey from homeassistant.util.hass_dict import HassKey
from .const import DOMAIN from .const import DOMAIN
@ -33,8 +31,6 @@ PLATFORMS: list[Platform] = [
Platform.UPDATE, Platform.UPDATE,
] ]
CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
IRON_OS_KEY: HassKey[IronOSFirmwareUpdateCoordinator] = HassKey(DOMAIN) IRON_OS_KEY: HassKey[IronOSFirmwareUpdateCoordinator] = HassKey(DOMAIN)
@ -42,19 +38,15 @@ IRON_OS_KEY: HassKey[IronOSFirmwareUpdateCoordinator] = HassKey(DOMAIN)
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
"""Set up IronOS firmware update coordinator."""
session = async_get_clientsession(hass)
github = IronOSUpdate(session)
hass.data[IRON_OS_KEY] = IronOSFirmwareUpdateCoordinator(hass, github)
await hass.data[IRON_OS_KEY].async_request_refresh()
return True
async def async_setup_entry(hass: HomeAssistant, entry: IronOSConfigEntry) -> bool: async def async_setup_entry(hass: HomeAssistant, entry: IronOSConfigEntry) -> bool:
"""Set up IronOS from a config entry.""" """Set up IronOS from a config entry."""
if IRON_OS_KEY not in hass.data:
session = async_get_clientsession(hass)
github = IronOSUpdate(session)
hass.data[IRON_OS_KEY] = IronOSFirmwareUpdateCoordinator(hass, github)
await hass.data[IRON_OS_KEY].async_request_refresh()
if TYPE_CHECKING: if TYPE_CHECKING:
assert entry.unique_id assert entry.unique_id
@ -77,4 +69,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: IronOSConfigEntry) -> bo
async def async_unload_entry(hass: HomeAssistant, entry: IronOSConfigEntry) -> bool: async def async_unload_entry(hass: HomeAssistant, entry: IronOSConfigEntry) -> bool:
"""Unload a config entry.""" """Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS) unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
if not hass.config_entries.async_loaded_entries(DOMAIN):
await hass.data[IRON_OS_KEY].async_shutdown()
hass.data.pop(IRON_OS_KEY)
return unload_ok