feat: refactor to use sensor-state-data 2 (#13)

Bluetooth-Devices · Jul 21, 2022 · 53afe59 · 53afe59
1 parent 8f86754
commit 53afe59
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 8 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -30,7 +30,7 @@ Sphinx = {version = "^5.0", optional = true}
 sphinx-rtd-theme = {version = "^1.0", optional = true}
 myst-parser = {version = "^0.18", optional = true}
 home-assistant-bluetooth = ">=1.3.0"
-sensor-state-data = ">=1.11.1"
+sensor-state-data = ">=2.0"
 
 [tool.poetry.extras]
 docs = [

diff --git a/src/bluetooth_sensor_state_data/__init__.py b/src/bluetooth_sensor_state_data/__init__.py
@@ -5,8 +5,7 @@
 from abc import abstractmethod
 
 from home_assistant_bluetooth import BluetoothServiceInfo
-from sensor_state_data import DeviceClass, SensorData, SensorUpdate
-from sensor_state_data.units import SIGNAL_STRENGTH_DECIBELS_MILLIWATT
+from sensor_state_data import DeviceClass, SensorData, SensorUpdate, Units
 
 SIGNAL_STRENGTH_KEY = DeviceClass.SIGNAL_STRENGTH.value
 
@@ -36,7 +35,7 @@ def update_signal_strength(self, native_value: int | float) -> None:
         for device_id in self._device_id_to_type:
             self.update_sensor(
                 key=DeviceClass.SIGNAL_STRENGTH.value,
-                native_unit_of_measurement=SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
+                native_unit_of_measurement=Units.SIGNAL_STRENGTH_DECIBELS_MILLIWATT,
                 native_value=native_value,
                 device_class=DeviceClass.SIGNAL_STRENGTH,
                 device_id=device_id,