Merge pull request #11 from DiUS/feature/rssi-diagnostic

Feature/rssi diagnostic

Author: jmattsson

custom_components/powersensor/PowersensorMessageDispatcher.py

@@ -140,7 +140,6 @@ def _create_api(self, mac_address, ip, port, name):
         self._known_plugs.add(mac_address)
         self._known_plug_names[name] = mac_address
         known_evs = [
-            #'exception',
             'average_flow',
             'average_power',
             'average_power_components',
@@ -154,6 +153,7 @@ def _create_api(self, mac_address, ip, port, name):
         for ev in known_evs:
             api.subscribe(ev, self.handle_message)
         api.subscribe('now_relaying_for', self.handle_relaying_for)
+        api.subscribe('exception', self.handle_exception)
         api.connect()
 
     def cancel_any_pending_removal(self, mac, source):
@@ -162,6 +162,9 @@ def cancel_any_pending_removal(self, mac, source):
             task.cancel()
             _LOGGER.debug(f"Cancelled pending removal for {mac} by {source}.")
 
+    async def handle_exception(self, event: str, exc: BaseException):
+        _LOGGER.error(f"Plug connection reported exception: {exc}")
+
     async def handle_relaying_for(self, event: str, message: dict):
         """Handle a potentially new sensor being reported."""
         mac = message.get('mac', None)

custom_components/powersensor/PowersensorSensorEntity.py

@@ -1,5 +1,5 @@
 from homeassistant.components.sensor import SensorDeviceClass, SensorStateClass
-from homeassistant.const import EntityCategory, UnitOfPower, UnitOfEnergy, PERCENTAGE
+from homeassistant.const import EntityCategory, UnitOfPower, UnitOfEnergy, PERCENTAGE, SIGNAL_STRENGTH_DECIBELS
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.device_registry import DeviceInfo
 
@@ -47,6 +47,15 @@
         'event': 'role',
         'message_key': 'role',
     },
+    SensorMeasurements.RSSI: {
+        'name': 'Signal strength (Bluetooth)',
+        "device_class": SensorDeviceClass.SIGNAL_STRENGTH,
+        "unit": SIGNAL_STRENGTH_DECIBELS,
+        "precision": 1,
+        'category': EntityCategory.DIAGNOSTIC,
+        'event': 'radio_signal_quality',
+        'message_key': 'average_rssi',
+    },
 }
 
 class PowersensorSensorEntity(PowersensorEntity):

custom_components/powersensor/SensorMeasurements.py

@@ -6,3 +6,4 @@ class SensorMeasurements(Enum):
     WATTS = 2
     SUMMATION_ENERGY = 3
     ROLE = 4
+    RSSI = 5

custom_components/powersensor/sensor.py

@@ -110,6 +110,7 @@ async def handle_discovered_sensor(sensor_mac: str, sensor_role: str):
             PowersensorSensorEntity(hass, sensor_mac, sensor_role, SensorMeasurements.WATTS),
             PowersensorSensorEntity(hass, sensor_mac, sensor_role, SensorMeasurements.SUMMATION_ENERGY),
             PowersensorSensorEntity(hass, sensor_mac, sensor_role, SensorMeasurements.ROLE),
+            PowersensorSensorEntity(hass, sensor_mac, sensor_role, SensorMeasurements.RSSI),
         ]
         async_add_entities(new_sensors, True)
         async_dispatcher_send(hass, SENSOR_ADDED_TO_HA_SIGNAL, sensor_mac, sensor_role)