diff --git a/haco/device.py b/haco/device.py index 294bc6c..08f18dc 100644 --- a/haco/device.py +++ b/haco/device.py @@ -4,7 +4,7 @@ from typing import List from haco import tools, constants from haco.control import Control -from haco.device_data import DeviceInfo +from haco.device_data import DeviceData from haco.tools import log_publish, split_into_chunks, log_subscribe @@ -105,7 +105,7 @@ class Device: for topic in data.keys(): await log_subscribe(client, str(topic)) - async def bind(self, client, data_announce: DeviceInfo): + async def bind(self, client, data_announce: DeviceData): msg = f'Found device with matching config {self.config_module.__name__}: Hostname: {data_announce.hostname} MAC:{data_announce.mac}' tools.logger.info(msg) diff --git a/haco/device_data.py b/haco/device_data.py index 6da1c75..58d0c31 100644 --- a/haco/device_data.py +++ b/haco/device_data.py @@ -4,7 +4,7 @@ from typing import Dict @dataclass -class DeviceInfo: +class DeviceData: wifi: Dict[str, any] device_name: str fulltopic: str diff --git a/haco/mqtt.py b/haco/mqtt.py index 3aa406e..46e1ca5 100644 --- a/haco/mqtt.py +++ b/haco/mqtt.py @@ -6,7 +6,7 @@ from pathlib import Path from haco import constants, tools from haco.constants import MQTT_PASSWORD, MQTT_HOST, MQTT_PORT, MQTT_USERNAME from haco.device import Device -from haco.device_data import DeviceInfo +from haco.device_data import DeviceData from haco.load_configs import load_devices from haco.tools import log_received, logger @@ -27,7 +27,7 @@ async def listen(client, devices): continue data_announce = json.loads(message.payload) - data_announce = DeviceInfo(**data_announce) + data_announce = DeviceData(**data_announce) for identifier in data_announce.identifiers: device: Device = devices.get(identifier)