From c744919032ca7ae81311ebae91b1d920db7752c6 Mon Sep 17 00:00:00 2001
From: KNOOP <knuop@icloud.com>
Date: Tue, 17 Dec 2024 18:21:03 +0800
Subject: [PATCH] fix: handle UnitOfConductivity import #54
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Move unit imports inside function 和 add fallback for older versions. Resolves
---
 .../xiaomi_home/miot/miot_device.py           | 32 +++++++++----------
 1 file changed, 15 insertions(+), 17 deletions(-)

diff --git a/custom_components/xiaomi_home/miot/miot_device.py b/custom_components/xiaomi_home/miot/miot_device.py
index 3531ebf..8de4c1c 100644
--- a/custom_components/xiaomi_home/miot/miot_device.py
+++ b/custom_components/xiaomi_home/miot/miot_device.py
@@ -54,11 +54,24 @@ from homeassistant.helpers.entity import Entity
 from homeassistant.const import (
     CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
     CONCENTRATION_MILLIGRAMS_PER_CUBIC_METER,
-    CONCENTRATION_PARTS_PER_MILLION,
     CONCENTRATION_PARTS_PER_BILLION,
+    CONCENTRATION_PARTS_PER_MILLION,
     LIGHT_LUX,
     PERCENTAGE,
     SIGNAL_STRENGTH_DECIBELS,
+    UnitOfEnergy,
+    UnitOfElectricCurrent,
+    UnitOfElectricPotential,
+    UnitOfInformation,
+    UnitOfLength,
+    UnitOfMass,
+    UnitOfSpeed,
+    UnitOfTime,
+    UnitOfTemperature,
+    UnitOfPressure,
+    UnitOfPower,
+    UnitOfVolume,
+    UnitOfVolumeFlowRate,
 )
 from homeassistant.helpers.entity import DeviceInfo
 from homeassistant.components.switch import SwitchDeviceClass
@@ -572,22 +585,6 @@ class MIoTDevice:
 
     def unit_convert(self, spec_unit: str) -> Optional[str]:
         """Convert MIoT unit to Home Assistant unit."""
-        from homeassistant.const import (
-            UnitOfEnergy,
-            UnitOfElectricCurrent,
-            UnitOfElectricPotential,
-            UnitOfInformation,
-            UnitOfLength,
-            UnitOfMass,
-            UnitOfSpeed,
-            UnitOfTime,
-            UnitOfTemperature,
-            UnitOfPressure,
-            UnitOfPower,
-            UnitOfVolume,
-            UnitOfVolumeFlowRate,
-        )
-
         unit_map = {
             'percentage': PERCENTAGE,
             'weeks': UnitOfTime.WEEKS,
@@ -625,6 +622,7 @@ class MIoTDevice:
             'kB': UnitOfInformation.KILOBYTES,
         }
 
+        # Handle UnitOfConductivity separately since it might not be available in all HA versions
         try:
             from homeassistant.const import UnitOfConductivity
             unit_map['μS/cm'] = UnitOfConductivity.MICROSIEMENS_PER_CM