forked from HomeAssistant/fn_nas
114 lines
4.2 KiB
Python
114 lines
4.2 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)
|
||
|
||
setup_task = hass.async_create_task(
|
||
async_delayed_setup(hass, entry, coordinator),
|
||
f"fn_nas_setup_{entry.entry_id}"
|
||
)
|
||
|
||
|
||
return True
|
||
|
||
async def async_delayed_setup(hass: HomeAssistant, entry: ConfigEntry, coordinator: FlynasCoordinator):
|
||
try:
|
||
# 首先进行轻量级系统状态检测
|
||
is_online = await coordinator.ping_system()
|
||
|
||
if not is_online:
|
||
_LOGGER.warning("系统离线，等待系统上线...")
|
||
# 等待系统上线
|
||
while not await coordinator.ping_system():
|
||
await asyncio.sleep(30)
|
||
|
||
_LOGGER.info("系统已上线，继续初始化飞牛NAS集成")
|
||
|
||
# 系统在线，继续正常初始化
|
||
await coordinator.async_config_entry_first_refresh()
|
||
|
||
# 检查是否启用Docker，并初始化Docker管理器（如果有）
|
||
enable_docker = coordinator.config.get(CONF_ENABLE_DOCKER, False)
|
||
if enable_docker:
|
||
# 导入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.setdefault(DOMAIN, {})
|
||
hass.data[DOMAIN][entry.entry_id] = {
|
||
DATA_UPDATE_COORDINATOR: coordinator,
|
||
"ups_coordinator": ups_coordinator,
|
||
CONF_ENABLE_DOCKER: enable_docker # 存储启用状态
|
||
}
|
||
|
||
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 |