chore(auto_configure): adjust device class / network for qt category

xZetsubou · Jan 17, 2025 · 3a63542 · 3a63542
1 parent 363491a
commit 3a63542
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/custom_components/localtuya/core/ha_entities/selects.py b/custom_components/localtuya/core/ha_entities/selects.py
@@ -487,6 +487,12 @@ def localtuya_selector(options):
                 {"enabled_energy": "Enable", "forbidden_energy": "Disable"}
             ),
         ),
+        LocalTuyaEntity(
+            id=DPCode.NETWORK_MODEL,
+            name="Network",
+            entity_category=EntityCategory.CONFIG,
+            custom_configs=localtuya_selector({"LAN": "LAN", "4G": "4G"}),
+        ),
     ),
     # Siren Alarm
     # https://developer.tuya.com/en/docs/iot/categorysgbj?id=Kaiuz37tlpbnu

diff --git a/custom_components/localtuya/core/ha_entities/sensors.py b/custom_components/localtuya/core/ha_entities/sensors.py
@@ -759,16 +759,16 @@ def localtuya_sensor(unit_of_measurement=None, scale_factor: float = 1) -> dict:
         LocalTuyaEntity(
             id=DPCode.ELECTRICITY_PHASE_A,
             name="Electricity Phase A",
-            device_class=SensorDeviceClass.VOLTAGE,
+            device_class=SensorDeviceClass.CURRENT,
             state_class=SensorStateClass.MEASUREMENT,
-            custom_configs=localtuya_sensor(UnitOfElectricPotential.VOLT, 0.01),
+            custom_configs=localtuya_sensor(UnitOfElectricCurrent.AMPERE, 0.01),
         ),
         LocalTuyaEntity(
             id=DPCode.ELECTRICITY_PHASE_B,
             name="Electricity Phase B",
-            device_class=SensorDeviceClass.VOLTAGE,
+            device_class=SensorDeviceClass.CURRENT,
             state_class=SensorStateClass.MEASUREMENT,
-            custom_configs=localtuya_sensor(UnitOfElectricPotential.VOLT, 0.01),
+            custom_configs=localtuya_sensor(UnitOfElectricCurrent.AMPERE, 0.01),
         ),
         LocalTuyaEntity(
             id=DPCode.ELECTRICITY_PHASE_C,
@@ -808,11 +808,6 @@ def localtuya_sensor(unit_of_measurement=None, scale_factor: float = 1) -> dict:
             name="Version",
             entity_category=EntityCategory.DIAGNOSTIC,
         ),
-        LocalTuyaEntity(
-            id=DPCode.NETWORK_MODEL,
-            name="Network",
-            entity_category=EntityCategory.DIAGNOSTIC,
-        ),
     ),
     # Gas Detector
     # https://developer.tuya.com/en/docs/iot/categoryrqbj?id=Kaiuz3d162ubw