updated sensor data handling

custom_components/bluetti_bt/binary_sensor.py

@@ -11,7 +11,6 @@ from homeassistant.core import HomeAssistant, callback
 from homeassistant.const import (
     CONF_ADDRESS,
     CONF_NAME,
-    CONF_UNIT_OF_MEASUREMENT,
     EntityCategory,
 )
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -21,14 +20,10 @@ from homeassistant.helpers.update_coordinator import (
 )
 
 from bluetti_mqtt.bluetooth import build_device
-from bluetti_mqtt.mqtt_client import NORMAL_DEVICE_FIELDS
+from bluetti_mqtt.mqtt_client import NORMAL_DEVICE_FIELDS, MqttFieldType
 
 from . import device_info as dev_info, get_unique_id
-from .const import (
-    DOMAIN,
-    API_RESPONSE_BATTERY_RANGE_START,
-    API_RESPONSE_BATTERY_RANGE_END,
-)
+from .const import DOMAIN
 from .coordinator import PollingCoordinator
 
 _LOGGER = logging.getLogger(__name__)
@@ -54,15 +49,9 @@ async def async_setup_entry(
     sensors_to_add = []
     for field_key, field_config in NORMAL_DEVICE_FIELDS.items():
         if bluetti_device.has_field(field_key):
-            if (
-                field_config.home_assistant_extra.get(CONF_UNIT_OF_MEASUREMENT) is None
-                and not field_config.setter
-            ):
+            if field_config.type == MqttFieldType.BOOL:
                 category = None
-                if field_key in (
-                    API_RESPONSE_BATTERY_RANGE_START,
-                    API_RESPONSE_BATTERY_RANGE_END,
-                ):
+                if field_config.setter is True:
                     category = EntityCategory.DIAGNOSTIC
 
                 sensors_to_add.append(
@@ -72,7 +61,7 @@ async def async_setup_entry(
                         address,
                         field_key,
                         field_config.home_assistant_extra.get(CONF_NAME, ""),
-                        category,
+                        category=category,
                     )
                 )
 
@@ -114,7 +103,22 @@ class BluettiBinarySensor(CoordinatorEntity, BinarySensorEntity):
         """Handle updated data from the coordinator."""
         _LOGGER.debug("Updating state of %s", self._attr_unique_id)
         if not isinstance(self.coordinator.data, dict):
-            _LOGGER.error("Invalid data from coordinator")
+            _LOGGER.error(
+                "Invalid data from coordinator (sensor.%s)", self._attr_unique_id
+            )
             return
+
+        response_data = self.coordinator.data.get(self._response_key)
+        if response_data is None:
+            return
+
+        if not isinstance(response_data, bool):
+            _LOGGER.warning(
+                "Invalid response data type from coordinator (sensor.%s): %s",
+                self._attr_unique_id,
+                response_data,
+            )
+            return
+
         self._attr_is_on = self.coordinator.data[self._response_key] is True
         self.async_write_ha_state()

custom_components/bluetti_bt/const.py

@@ -3,5 +3,4 @@
 DOMAIN = "bluetti_bt"
 MANUFACTURER = "Bluetti"
 
-API_RESPONSE_BATTERY_RANGE_START = "battery_range_start"
-API_RESPONSE_BATTERY_RANGE_END = "battery_range_end"
+CONF_OPTIONS = "options"

custom_components/bluetti_bt/coordinator.py

@@ -54,11 +54,11 @@ class PollingCoordinator(DataUpdateCoordinator):
 
         device = bluetooth.async_ble_device_from_address(self.hass, self._address)
         if device is None:
-            self.logger.error("Device not available")
+            self.logger.error("Device %s not available", self._address)
             return None
 
         if self.bluetti_device is None:
-            self.logger.error("Device type not found")
+            self.logger.error("Device type for %s not found", self._address)
             return None
 
         # Polling
@@ -99,11 +99,13 @@ class PollingCoordinator(DataUpdateCoordinator):
                             command.starting_address, body
                         )
 
-                        self.logger.warning("Parsed data: %s", parsed)
+                        self.logger.debug("Parsed data: %s", parsed)
                         parsed_data.update(parsed)
 
                     except TimeoutError:
-                        self.logger.error("Polling timed out (address: %s)", self._address)
+                        self.logger.error(
+                            "Polling timed out (address: %s)", self._address
+                        )
                     except ParseError:
                         self.logger.debug("Got a parse exception...")
                     except ModbusError as err:
@@ -115,7 +117,7 @@ class PollingCoordinator(DataUpdateCoordinator):
                     except (BadConnectionError, BleakError) as err:
                         self.logger.debug("Needed to disconnect due to error: %s", err)
         except TimeoutError:
-            self.logger.error("Polling timed out")
+            self.logger.error("Polling timed out for device %s", self._address)
             return None
         except BleakError as err:
             self.logger.error("Bleak error: %s", err)

custom_components/bluetti_bt/sensor.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import logging
+from decimal import *
 
 from homeassistant.components import bluetooth
 from homeassistant.components.sensor import SensorEntity, CONF_STATE_CLASS
@@ -22,14 +23,10 @@ from homeassistant.helpers.update_coordinator import (
 )
 
 from bluetti_mqtt.bluetooth import build_device
-from bluetti_mqtt.mqtt_client import NORMAL_DEVICE_FIELDS
+from bluetti_mqtt.mqtt_client import NORMAL_DEVICE_FIELDS, MqttFieldType
 
 from . import device_info as dev_info, get_unique_id
-from .const import (
-    DOMAIN,
-    API_RESPONSE_BATTERY_RANGE_START,
-    API_RESPONSE_BATTERY_RANGE_END,
-)
+from .const import DOMAIN, CONF_OPTIONS
 from .coordinator import PollingCoordinator
 
 _LOGGER = logging.getLogger(__name__)
@@ -55,18 +52,10 @@ async def async_setup_entry(
     sensors_to_add = []
     for field_key, field_config in NORMAL_DEVICE_FIELDS.items():
         if bluetti_device.has_field(field_key):
-            if (
-                field_config.home_assistant_extra.get(CONF_UNIT_OF_MEASUREMENT)
-                is not None
-                and not field_config.setter
-            ):
-                category = None
-                if field_key in (
-                    API_RESPONSE_BATTERY_RANGE_START,
-                    API_RESPONSE_BATTERY_RANGE_END,
-                ):
-                    category = EntityCategory.DIAGNOSTIC
-
+            category = None
+            if field_config.setter is True:
+                category = EntityCategory.DIAGNOSTIC
+            if field_config.type == MqttFieldType.NUMERIC:
                 sensors_to_add.append(
                     BluettiSensor(
                         hass.data[DOMAIN][entry.entry_id],
@@ -79,7 +68,19 @@ async def async_setup_entry(
                         ),
                         field_config.home_assistant_extra.get(CONF_DEVICE_CLASS, ""),
                         field_config.home_assistant_extra.get(CONF_STATE_CLASS, ""),
-                        category,
+                        category=category,
+                    )
+                )
+            elif field_config.type == MqttFieldType.ENUM:
+                sensors_to_add.append(
+                    BluettiSensor(
+                        hass.data[DOMAIN][entry.entry_id],
+                        device_info,
+                        address,
+                        field_key,
+                        field_config.home_assistant_extra.get(CONF_NAME, ""),
+                        options=field_config.home_assistant_extra.get(CONF_OPTIONS),
+                        category=category,
                     )
                 )
 
@@ -96,10 +97,11 @@ class BluettiSensor(CoordinatorEntity, SensorEntity):
         address,
         response_key: str,
         name: str,
-        unit_of_measurement: str,
-        device_class: str,
-        state_class: str,
+        unit_of_measurement: str | None = None,
+        device_class: str | None = None,
+        state_class: str | None = None,
         category: EntityCategory | None = None,
+        options: list[str] | None = None,
     ):
         """Init battery entity."""
         super().__init__(coordinator)
@@ -115,6 +117,7 @@ class BluettiSensor(CoordinatorEntity, SensorEntity):
         self._attr_device_class = device_class
         self._attr_state_class = state_class
         self._attr_entity_category = category
+        self._options = options
 
     @property
     def available(self) -> bool:
@@ -127,7 +130,38 @@ class BluettiSensor(CoordinatorEntity, SensorEntity):
         """Handle updated data from the coordinator."""
         _LOGGER.debug("Updating state of %s", self._attr_unique_id)
         if not isinstance(self.coordinator.data, dict):
-            _LOGGER.error("Invalid data from coordinator")
+            _LOGGER.error(
+                "Invalid data from coordinator (sensor.%s)", self._attr_unique_id
+            )
             return
-        self._attr_native_value = self.coordinator.data[self._response_key]
+
+        response_data = self.coordinator.data.get(self._response_key)
+        if response_data is None:
+            return
+
+        if (
+            not isinstance(response_data, int)
+            and not isinstance(response_data, float)
+            and not isinstance(response_data, complex)
+            and not isinstance(response_data, Decimal)
+        ):
+            _LOGGER.warning(
+                "Invalid response data type from coordinator (sensor.%s): %s has type %s",
+                self._attr_unique_id,
+                response_data,
+                type(response_data),
+            )
+            return
+
+        # Different for enum and numeric
+        if (
+            self._options is not None
+            and isinstance(response_data, int)
+            and response_data < len(self._options)
+        ):
+            # Enum
+            self._attr_native_value = self._options[response_data]
+        else:
+            # Numeric
+            self._attr_native_value = response_data
         self.async_write_ha_state()