Add support for Inverter Pool Heat Pump (InverGo)

homeassistant/components/tuya/const.py

@@ -658,6 +658,7 @@ class DPCode(StrEnum):
     COLOUR_DATA = "colour_data"  # Colored light mode
     COLOUR_DATA_HSV = "colour_data_hsv"  # Colored light mode
     COLOUR_DATA_V2 = "colour_data_v2"  # Colored light mode
+    COMPRESSOR_STRENGTH = "compressor_strength"
     CONCENTRATION_SET = "concentration_set"  # Concentration setting
     CONTROL = "control"
     CONTROL_2 = "control_2"
@@ -911,8 +912,10 @@ class DPCode(StrEnum):
     TARGET_DIS_CLOSEST = "target_dis_closest"  # Closest target distance
     TDS_IN = "tds_in"  # Total dissolved solids
     TEMP = "temp"  # Temperature setting
+    TEMP_AROUND = "temp_around"  # Current around (outside) temperature
     TEMP_BOILING_C = "temp_boiling_c"
     TEMP_BOILING_F = "temp_boiling_f"
+    TEMP_COILER = "temp_coiler"  # Current coiler temperature
     TEMP_CONTROLLER = "temp_controller"
     TEMP_CORRECTION = "temp_correction"
     TEMP_CURRENT = "temp_current"  # Current temperature in °C
@@ -933,9 +936,11 @@ class DPCode(StrEnum):
         "temp_current_external_f"  # Current external temperature in Fahrenheit
     )
     TEMP_CURRENT_F = "temp_current_f"  # Current temperature in °F
+    TEMP_EFFLUENT = "temp_effluent"  # Current flow temperature
     TEMP_INDOOR = "temp_indoor"  # Indoor temperature in °C
     TEMP_SET = "temp_set"  # Set the temperature in °C
     TEMP_SET_F = "temp_set_f"  # Set the temperature in °F
+    TEMP_VENTING = "temp_venting"  # Current heat plate temperature
     TEMP_UNIT_CONVERT = "temp_unit_convert"  # Temperature unit switching
     TEMP_VALUE = "temp_value"  # Color temperature
     TEMP_VALUE_V2 = "temp_value_v2"

homeassistant/components/tuya/sensor.py

@@ -1,6 +1,6 @@
 """Support for Tuya sensors."""
 
-from dataclasses import dataclass
+from dataclasses import dataclass, replace
 
 from tuya_device_handlers.definition.sensor import (
     SensorDefinition,
@@ -38,6 +38,7 @@
     UnitOfElectricPotential,
     UnitOfEnergy,
     UnitOfPower,
+    UnitOfTemperature,
     UnitOfTime,
 )
 from homeassistant.core import HomeAssistant, callback
@@ -48,6 +49,7 @@
 from .const import (
     DEVICE_CLASS_UNITS,
     DOMAIN,
+    FAHRENHEIT_ALIASES,
     LOGGER,
     TUYA_DISCOVERY_NEW,
     DeviceCategory,
@@ -1613,12 +1615,42 @@ class TuyaSensorEntityDescription(SensorEntityDescription):
         ),
     ),
     DeviceCategory.ZNRB: (
+        TuyaSensorEntityDescription(
+            key=DPCode.COMPRESSOR_STRENGTH,
+            translation_key="compressor_strength",
+            state_class=SensorStateClass.MEASUREMENT,
+            native_unit_of_measurement=PERCENTAGE,
+        ),
+        TuyaSensorEntityDescription(
+            key=DPCode.TEMP_AROUND,
+            translation_key="outside_temperature",
+            device_class=SensorDeviceClass.TEMPERATURE,
+            state_class=SensorStateClass.MEASUREMENT,
+        ),
+        TuyaSensorEntityDescription(
+            key=DPCode.TEMP_COILER,
+            translation_key="coiler_temperature",
+            device_class=SensorDeviceClass.TEMPERATURE,
+            state_class=SensorStateClass.MEASUREMENT,
+        ),
         TuyaSensorEntityDescription(
             key=DPCode.TEMP_CURRENT,
             translation_key="temperature",
             device_class=SensorDeviceClass.TEMPERATURE,
             state_class=SensorStateClass.MEASUREMENT,
         ),
+        TuyaSensorEntityDescription(
+            key=DPCode.TEMP_EFFLUENT,
+            translation_key="flow_temperature",
+            device_class=SensorDeviceClass.TEMPERATURE,
+            state_class=SensorStateClass.MEASUREMENT,
+        ),
+        TuyaSensorEntityDescription(
+            key=DPCode.TEMP_VENTING,
+            translation_key="heat_exchanger_temperature",
+            device_class=SensorDeviceClass.TEMPERATURE,
+            state_class=SensorStateClass.MEASUREMENT,
+        ),
     ),
     DeviceCategory.ZWJCY: (
         TuyaSensorEntityDescription(
@@ -1647,6 +1679,19 @@ class TuyaSensorEntityDescription(SensorEntityDescription):
 SENSORS[DeviceCategory.PC] = SENSORS[DeviceCategory.KG]
 
 
+def _get_temp_unit_from_device(
+    device: CustomerDevice,
+) -> UnitOfTemperature | None:
+    """Return the temperature unit reported by the device via TEMP_UNIT_CONVERT, or None."""
+    unit_value = device.status.get(DPCode.TEMP_UNIT_CONVERT)
+    if unit_value is None:
+        return None
+    unit_str = str(unit_value).lower()
+    if unit_str in FAHRENHEIT_ALIASES:
+        return UnitOfTemperature.FAHRENHEIT
+    return UnitOfTemperature.CELSIUS
+
+
 async def async_setup_entry(
     hass: HomeAssistant,
     entry: TuyaConfigEntry,
@@ -1662,17 +1707,27 @@ def async_discover_device(device_ids: list[str]) -> None:
         for device_id in device_ids:
             device = manager.device_map[device_id]
             if descriptions := SENSORS.get(device.category):
-                entities.extend(
-                    TuyaSensorEntity(device, manager, description, definition)
-                    for description in descriptions
-                    if (
+                temp_unit = _get_temp_unit_from_device(device)
+                for description in descriptions:
+                    if not (
                         definition := get_default_definition(
                             device,
                             description.dpcode or description.key,
                             description.wrapper_class,
                         )
+                    ):
+                        continue
+                    if (
+                        temp_unit is not None
+                        and description.device_class == SensorDeviceClass.TEMPERATURE
+                        and description.native_unit_of_measurement is None
+                    ):
+                        description = replace(
+                            description, native_unit_of_measurement=temp_unit
+                        )
+                    entities.append(
+                        TuyaSensorEntity(device, manager, description, definition)
                     )
-                )
 
         async_add_entities(entities)
 

homeassistant/components/tuya/strings.json

@@ -655,6 +655,12 @@
       "cleaning_time": {
         "name": "Cleaning time"
       },
+      "coiler_temperature": {
+        "name": "Coiler temperature"
+      },
+      "compressor_strength": {
+        "name": "Compressor strength"
+      },
       "concentration_carbon_dioxide": {
         "name": "Concentration of carbon dioxide"
       },
@@ -691,12 +697,18 @@
       "filter_utilization": {
         "name": "Filter utilization"
       },
+      "flow_temperature": {
+        "name": "Flow temperature"
+      },
       "formaldehyde": {
         "name": "[%key:component::tuya::entity::binary_sensor::formaldehyde::name%]"
       },
       "gas": {
         "name": "Gas"
       },
+      "heat_exchanger_temperature": {
+        "name": "Heat exchanger temperature"
+      },
       "heat_index_temperature": {
         "name": "Heat index"
       },
@@ -779,6 +791,9 @@
           "work": "Working"
         }
       },
+      "outside_temperature": {
+        "name": "Outside temperature"
+      },
       "oxydo_reduction_potential": {
         "name": "Oxydo reduction potential"
       },

homeassistant/components/tuya/switch.py

@@ -901,6 +901,11 @@
         ),
     ),
     DeviceCategory.ZNRB: (
+        SwitchEntityDescription(
+            key=DPCode.CHILD_LOCK,
+            translation_key="child_lock",
+            icon="mdi:account-lock",
+        ),
         SwitchEntityDescription(
             key=DPCode.SWITCH,
             translation_key="switch",