mirror of
https://github.com/xiaochao99/fn_nas
synced 2025-12-26 14:57:12 +00:00
89 lines
3.6 KiB
Python
89 lines
3.6 KiB
Python
import logging
|
||
import asyncio
|
||
import asyncssh
|
||
from homeassistant.config_entries import ConfigEntry
|
||
from homeassistant.core import HomeAssistant
|
||
from homeassistant.helpers import config_validation as cv
|
||
|
||
from .const import (
|
||
DOMAIN, DATA_UPDATE_COORDINATOR, PLATFORMS, CONF_ENABLE_DOCKER,
|
||
CONF_HOST, DEFAULT_PORT
|
||
)
|
||
from .coordinator import FlynasCoordinator, UPSDataUpdateCoordinator
|
||
|
||
_LOGGER = logging.getLogger(__name__)
|
||
|
||
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
|
||
config = {**entry.data, **entry.options}
|
||
coordinator = FlynasCoordinator(hass, config, entry)
|
||
# 直接初始化，不阻塞等待NAS上线
|
||
hass.data.setdefault(DOMAIN, {})
|
||
hass.data[DOMAIN][entry.entry_id] = {
|
||
DATA_UPDATE_COORDINATOR: coordinator,
|
||
"ups_coordinator": None,
|
||
CONF_ENABLE_DOCKER: coordinator.config.get(CONF_ENABLE_DOCKER, False)
|
||
}
|
||
# 异步后台初始化
|
||
hass.async_create_task(async_delayed_setup(hass, entry, coordinator))
|
||
return True
|
||
|
||
async def async_delayed_setup(hass: HomeAssistant, entry: ConfigEntry, coordinator: FlynasCoordinator):
|
||
try:
|
||
# 不阻塞等待NAS上线，直接尝试刷新数据
|
||
await coordinator.async_config_entry_first_refresh()
|
||
enable_docker = coordinator.config.get(CONF_ENABLE_DOCKER, False)
|
||
if enable_docker:
|
||
from .docker_manager import DockerManager
|
||
coordinator.docker_manager = DockerManager(coordinator)
|
||
_LOGGER.debug("已启用Docker容器监控")
|
||
else:
|
||
coordinator.docker_manager = None
|
||
_LOGGER.debug("未启用Docker容器监控")
|
||
ups_coordinator = UPSDataUpdateCoordinator(hass, coordinator.config, coordinator)
|
||
await ups_coordinator.async_config_entry_first_refresh()
|
||
hass.data[DOMAIN][entry.entry_id]["ups_coordinator"] = ups_coordinator
|
||
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
||
entry.async_on_unload(entry.add_update_listener(async_update_entry))
|
||
_LOGGER.info("飞牛NAS集成初始化完成")
|
||
except Exception as e:
|
||
_LOGGER.error("飞牛NAS集成初始化失败: %s", str(e))
|
||
await coordinator.async_disconnect()
|
||
if hasattr(coordinator, '_ping_task') and coordinator._ping_task:
|
||
coordinator._ping_task.cancel()
|
||
|
||
async def async_update_entry(hass: HomeAssistant, entry: ConfigEntry):
|
||
"""更新配置项"""
|
||
# 卸载现有集成
|
||
await async_unload_entry(hass, entry)
|
||
# 重新加载集成
|
||
await async_setup_entry(hass, entry)
|
||
|
||
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry):
|
||
"""卸载集成"""
|
||
# 获取集成数据
|
||
domain_data = hass.data.get(DOMAIN, {}).get(entry.entry_id, {})
|
||
unload_ok = True
|
||
|
||
if DATA_UPDATE_COORDINATOR in domain_data:
|
||
coordinator = domain_data[DATA_UPDATE_COORDINATOR]
|
||
ups_coordinator = domain_data.get("ups_coordinator")
|
||
|
||
# 卸载平台
|
||
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
|
||
|
||
if unload_ok:
|
||
# 关闭主协调器的SSH连接
|
||
await coordinator.async_disconnect()
|
||
|
||
# 关闭UPS协调器（如果存在）
|
||
if ups_coordinator:
|
||
await ups_coordinator.async_shutdown()
|
||
|
||
# 取消监控任务（如果存在）
|
||
if hasattr(coordinator, '_ping_task') and coordinator._ping_task and not coordinator._ping_task.done():
|
||
coordinator._ping_task.cancel()
|
||
|
||
# 从DOMAIN中移除该entry的数据
|
||
hass.data[DOMAIN].pop(entry.entry_id, None)
|
||
|
||
return unload_ok |