diff --git a/custom_components/bluetti_bt/__init__.py b/custom_components/bluetti_bt/__init__.py index ec16b3c..4ac4aac 100644 --- a/custom_components/bluetti_bt/__init__.py +++ b/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 diff --git a/custom_components/bluetti_bt/config_flow.py b/custom_components/bluetti_bt/config_flow.py index 8f46d3b..982e322 100644 --- a/custom_components/bluetti_bt/config_flow.py +++ b/custom_components/bluetti_bt/config_flow.py @@ -34,12 +34,12 @@ class BluettiConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): await self.async_set_unique_id(discovery_info.address) self._abort_if_unique_id_configured() self._discovery_info = discovery_info - self.context["title_placeholders"] = { - "name": discovery_info.name - } + self.context["title_placeholders"] = {"name": discovery_info.name} return await self.async_step_user() - async def async_step_user(self, user_input: dict[str, Any] | None = None) -> FlowResult: + async def async_step_user( + self, user_input: dict[str, Any] | None = None + ) -> FlowResult: """Handle user input.""" if user_input is not None: @@ -59,6 +59,7 @@ class BluettiConfigFlow(config_entries.ConfigFlow, domain=DOMAIN): else: current_addresses = self._async_current_ids() for discovery in async_discovered_service_info(self.hass): + address = discovery.address if address in current_addresses or address in self._discovered_devices: continue self._discovered_devices[discovery.address] = discovery diff --git a/custom_components/bluetti_bt/const.py b/custom_components/bluetti_bt/const.py index 562fba7..a19d9a7 100644 --- a/custom_components/bluetti_bt/const.py +++ b/custom_components/bluetti_bt/const.py @@ -1,3 +1,4 @@ """Constants for the Bluetti BT integration.""" DOMAIN = "bluetti_bt" +MANUFACTURER = "Bluetti" diff --git a/custom_components/bluetti_bt/sensor.py b/custom_components/bluetti_bt/sensor.py new file mode 100644 index 0000000..ccd3052 --- /dev/null +++ b/custom_components/bluetti_bt/sensor.py @@ -0,0 +1,80 @@ +"""Bluetti BT sensors.""" + +from __future__ import annotations + +from datetime import timedelta +import logging + +import async_timeout + +from homeassistant.components.sensor import SensorEntity +from homeassistant.config_entries import ConfigEntry +from homeassistant.core import HomeAssistant, callback +from homeassistant.const import CONF_ADDRESS +from homeassistant.helpers.entity_platform import AddEntitiesCallback +from homeassistant.helpers.entity import DeviceInfo +from homeassistant.helpers.update_coordinator import ( + CoordinatorEntity, + DataUpdateCoordinator, + UpdateFailed, +) + +from . import device_info as dev_info, get_unique_id + +_LOGGER = logging.getLogger(__name__) + + +async def async_setup_entry( + hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback +) -> None: + """Setup sensor entities.""" + + # Create coordinator for polling + coordinator = PollingCoordinator(hass) + await coordinator.async_config_entry_first_refresh() + + # Generate device info + _LOGGER.error("Creating sensors for device %s", entry.data.get(CONF_ADDRESS)) + device_info = dev_info(entry) + async_add_entities(Battery(coordinator, device_info)) + + +class PollingCoordinator(DataUpdateCoordinator): + """Polling coordinator.""" + + def __init__(self, hass): + """Initialize coordinator.""" + super().__init__( + hass, + _LOGGER, + name="Bluetti polling coordinator", + update_interval=timedelta(seconds=10), + ) + + async def _async_update_data(self): + """Fetch data from API endpoint. + + This is the place to pre-process the data to lookup tables + so entities can quickly look up their data. + """ + self.logger.error("Polling data") + + +class Battery(CoordinatorEntity, SensorEntity): + """Bluetti battery.""" + + def __init__(self, coordinator: PollingCoordinator, device_info: DeviceInfo): + """Init battery entity.""" + super().__init__(coordinator) + + self._attr_device_info = device_info + self._attr_name = f"{device_info.get('name')} Battery level" + self._attr_unique_id = get_unique_id(f"{device_info.get('name')} Battery level") + self._attr_unit_of_measurement = "%" + self._attr_device_class = "battery" + + @callback + def _handle_coordinator_update(self) -> None: + """Handle updated data from the coordinator.""" + _LOGGER.error("Updating state of %s", self._attr_unique_id) + self.async_write_ha_state() diff --git a/deploy.bat b/deploy.bat new file mode 100644 index 0000000..bf48c37 --- /dev/null +++ b/deploy.bat @@ -0,0 +1 @@ +robocopy .\custom_components\bluetti_bt T:\custom_components\bluetti_bt /E