added battery sensor

custom_components/bluetti_bt/__init__.py

@@ -2,14 +2,18 @@
 
 from __future__ import annotations
 
+import re
 import logging
 
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_ADDRESS, Platform
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import DeviceInfo
+
+from .const import DOMAIN, MANUFACTURER
 
 _LOGGER = logging.getLogger(__name__)
-PLATFORMS: [Platform] = []
+PLATFORMS: [Platform] = [Platform.SENSOR]
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
@@ -20,5 +24,25 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     if address is None:
         return False
 
+    # Setup platforms
+    await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
+
     _LOGGER.error("Device address: %s", address)
     return True
+
+
+def device_info(entry: ConfigEntry) -> DeviceInfo:
+    """Device info."""
+    return DeviceInfo(
+        identifiers={(DOMAIN, entry.data.get(CONF_ADDRESS))},
+        name=entry.title,
+        manufacturer=MANUFACTURER,
+    )
+
+
+def get_unique_id(name: str, sensor_type: str | None = None):
+    """Generate an unique id."""
+    res = re.sub("[^A-Za-z0-9]+", "_", name).lower()
+    if sensor_type is not None:
+        return f"{sensor_type}.{res}"
+    return res