fix: make strings (#5)

Bluetooth-Devices · Oct 25, 2022 · c4dda18 · c4dda18
1 parent b37a409
commit c4dda18
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/src/oralb_ble/parser.py b/src/oralb_ble/parser.py
@@ -164,23 +164,25 @@ def _process_mfr_data(
         else:
             tb_sector = "sector " + str(sector)
 
-        self.update_sensor(OralBSensor.COUNTER, None, counter, None, "Counter")
-        self.update_sensor(OralBSensor.SECTOR, None, tb_sector, None, "Sector")
+        self.update_sensor(str(OralBSensor.COUNTER), None, counter, None, "Counter")
+        self.update_sensor(str(OralBSensor.SECTOR), None, tb_sector, None, "Sector")
         self.update_sensor(
-            OralBSensor.NUMBER_OF_SECTORS,
+            str(OralBSensor.NUMBER_OF_SECTORS),
             None,
             no_of_sectors,
             None,
             "Number of sectors",
         )
         self.update_sensor(
-            OralBSensor.SECTOR_TIMER, None, sector_timer, None, "Sector Timer"
+            str(OralBSensor.SECTOR_TIMER), None, sector_timer, None, "Sector Timer"
         )
         self.update_sensor(
-            OralBSensor.TOOTHBRUSH_STATE, None, tb_state, None, "Toothbrush State"
+            str(OralBSensor.TOOTHBRUSH_STATE), None, tb_state, None, "Toothbrush State"
         )
-        self.update_sensor(OralBSensor.PRESSURE, None, tb_pressure, None, "Pressure")
-        self.update_sensor(OralBSensor.MODE, None, tb_mode, None, "Mode")
+        self.update_sensor(
+            str(OralBSensor.PRESSURE), None, tb_pressure, None, "Pressure"
+        )
+        self.update_sensor(str(OralBSensor.MODE), None, tb_mode, None, "Mode")
         self.update_binary_sensor(
-            OralBBinarySensor.BRUSHING, bool(state == 3), None, "Brushing"
+            str(OralBBinarySensor.BRUSHING), bool(state == 3), None, "Brushing"
         )