311cfbbe7e
修复内容: 1. 所有平台文件添加 coordinator.data None 检查 - sensor.py, binary_sensor.py, switch.py, light.py, device_tracker.py 2. 修复 async_get_options_flow 签名 - 改为实例方法，移除 config_entry 参数 - 移除 @staticmethod 装饰器 3. OptionsFlow 添加正确的 __init__ 方法 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
"""SigMesh Gateway 设备追踪平台."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from homeassistant import config_entries
|
|
from homeassistant.components.device_tracker import SourceType, TrackerEntity
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.device_registry import DeviceInfo
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from .const import DOMAIN
|
|
from .coordinator import SigMeshGatewayCoordinator
|
|
from .protocol_parser import DeviceState
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
entry: config_entries.ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""设置设备追踪平台."""
|
|
coordinator: SigMeshGatewayCoordinator = hass.data[DOMAIN][entry.entry_id][
|
|
"coordinator"
|
|
]
|
|
|
|
entities: list[SigMeshDeviceTracker] = []
|
|
|
|
if coordinator.data is None:
|
|
_LOGGER.debug("协调器数据为空，等待设备数据")
|
|
return
|
|
|
|
for device in coordinator.data.values():
|
|
entities.append(SigMeshDeviceTracker(coordinator, device))
|
|
|
|
async_add_entities(entities)
|
|
|
|
|
|
class SigMeshDeviceTracker(CoordinatorEntity, TrackerEntity):
|
|
"""SigMesh 设备追踪实体."""
|
|
|
|
_attr_has_entity_name = True
|
|
_attr_source_type = SourceType.BLUETOOTH_LE
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: SigMeshGatewayCoordinator,
|
|
device: DeviceState,
|
|
) -> None:
|
|
"""初始化设备追踪器."""
|
|
super().__init__(coordinator)
|
|
self._device = device
|
|
self._mac = device.mac_address
|
|
|
|
self._attr_unique_id = f"{DOMAIN}_tracker_{self._mac}"
|
|
self._attr_name = f"Tracker {self._mac}"
|
|
|
|
@property
|
|
def mac_address(self) -> str:
|
|
"""返回 MAC 地址."""
|
|
return self._device.mac_address
|
|
|
|
@property
|
|
def source_type(self) -> SourceType:
|
|
"""返回源类型."""
|
|
return SourceType.BLUETOOTH_LE
|
|
|
|
@property
|
|
def extra_state_attributes(self) -> dict[str, Any]:
|
|
"""返回额外状态属性."""
|
|
return {
|
|
"mac_address": self._device.mac_address,
|
|
"model_id": self._device.model_id,
|
|
"last_update": self._device.last_update,
|
|
}
|
|
|
|
@property
|
|
def device_info(self) -> DeviceInfo:
|
|
"""返回设备信息."""
|
|
return DeviceInfo(
|
|
identifiers={(DOMAIN, self._device.mac_address)},
|
|
name=f"SigMesh Device {self._device.mac_address}",
|
|
manufacturer="SigMesh",
|
|
)
|