Add Aqara Wall Outlet H2 EU

zhaquirks/xiaomi/__init__.py

@@ -369,15 +369,20 @@ def _parse_aqara_attributes(self, value):
                 {
                     100: TEMPERATURE_MEASUREMENT,
                     101: HUMIDITY_MEASUREMENT,
-                    102: TVOC_MEASUREMENT
-                    if self.endpoint.device.model == "lumi.airmonitor.acn01"
-                    else PRESSURE_MEASUREMENT_PRECISION
-                    if self.endpoint.device.model == "lumi.weather"
-                    else PRESSURE_MEASUREMENT,
+                    102: (
+                        TVOC_MEASUREMENT
+                        if self.endpoint.device.model == "lumi.airmonitor.acn01"
+                        else (
+                            PRESSURE_MEASUREMENT_PRECISION
+                            if self.endpoint.device.model == "lumi.weather"
+                            else PRESSURE_MEASUREMENT
+                        )
+                    ),
                 }
             )
         elif self.endpoint.device.model in [
             "lumi.plug",
+            "lumi.plug.aeu001",
             "lumi.plug.maus01",
             "lumi.plug.maeu01",
             "lumi.plug.mmeu01",

zhaquirks/xiaomi/aqara/plug_eu.py

@@ -1,8 +1,13 @@
 """Xiaomi Aqara EU plugs."""
 
+from enum import Enum
+import logging
+
 import zigpy
 from zigpy import types
 from zigpy.profiles import zgp, zha
+from zigpy.quirks.v2 import QuirkBuilder
+from zigpy.quirks.v2.homeassistant import UnitOfPower
 from zigpy.zcl.clusters.general import (
     Alarms,
     AnalogInput,
@@ -17,6 +22,7 @@
     Time,
 )
 from zigpy.zcl.clusters.homeautomation import ElectricalMeasurement
+from zigpy.zcl.clusters.measurement import TemperatureMeasurement
 from zigpy.zcl.clusters.smartenergy import Metering
 
 from zhaquirks.const import (
@@ -39,6 +45,8 @@
 
 OPPLE_MFG_CODE = 0x115F
 
+_LOGGER = logging.getLogger(__name__)
+
 
 async def remove_from_ep(dev: zigpy.device.Device) -> None:
     """Remove devices that are in group 0 by default, so IKEA devices don't control them.
@@ -346,3 +354,102 @@
     }
 
     replacement = PlugMAEU01.replacement
+
+
+class AqaraPowerOutageMemoryEnum(types.uint8_t, Enum):
+    """Power Outage Memory enum."""
+
+    On = 0x00
+    Previous = 0x01
+    Off = 0x02
+    Inverted = 0x03
+
+
+class PlugAEU001Cluster(XiaomiAqaraE1Cluster):
+    """Custom cluster for Aqara lumi plug AEU001."""
+
+    attributes = {
+        0x0200: ("button_lock", types.uint8_t, True),
+        0x0202: ("charging_protection", types.Bool, True),
+        0x0203: ("led_indicator", types.Bool, True),
+        0x0206: ("charging_limit", types.Single, True),
+        0x020B: ("overload_protection", types.Single, True),
+        0x0517: ("power_on_behavior", AqaraPowerOutageMemoryEnum, True),
+    }
+
+
+class PlugAEU001MeteringCluster(MeteringCluster):
+    """Custom cluster for Aqara lumi plug AEU001."""
+
+    def _update_attribute(self, attrid, value):
+        if attrid == self.CURRENT_SUMM_DELIVERED_ID:
+            current_value = self._attr_cache.get(attrid, 0)
+            if value < current_value:
+                _LOGGER.debug(
+                    "Ignoring attribute update for %s: new value %s is less than current value %s",
+                    attrid,
+                    value,
+                    current_value,
+                )
+                return
+        super()._update_attribute(attrid, value)
+
+
+(
+    QuirkBuilder("Aqara", "lumi.plug.aeu001")
+    .friendly_name(model="Wall Outlet H2 EU", manufacturer="Aqara")
+    .removes(TemperatureMeasurement.cluster_id)
+    .adds(DeviceTemperature)
+    .removes(OnOff.cluster_id, endpoint_id=2)
+    .replaces(BasicCluster)
+    .replaces(PlugAEU001MeteringCluster)
+    .replaces(ElectricalMeasurementCluster)
+    .replaces(PlugAEU001Cluster)
+    .replaces(AnalogInputCluster, endpoint_id=21)
+    .switch(
+        attribute_name="button_lock",
+        cluster_id=PlugAEU001Cluster.cluster_id,
+        force_inverted=True,
+        translation_key="button_lock",
+        fallback_name="Button Lock",
+    )
+    .enum(
+        attribute_name="power_on_behavior",
+        enum_class=AqaraPowerOutageMemoryEnum,
+        cluster_id=PlugAEU001Cluster.cluster_id,
+        translation_key="power_on_behavior",
+        fallback_name="Power On Behavior",
+    )
+    .number(
+        attribute_name="overload_protection",
+        cluster_id=PlugAEU001Cluster.cluster_id,
+        min_value=100,
+        max_value=3840,
+        unit=UnitOfPower.WATT,
+        translation_key="overload_protection",
+        fallback_name="Overload Protection",
+    )
+    .switch(
+        attribute_name="led_indicator",
+        cluster_id=PlugAEU001Cluster.cluster_id,
+        translation_key="led_indicator",
+        fallback_name="LED Indicator",
+    )
+    .switch(
+        attribute_name="charging_protection",
+        cluster_id=PlugAEU001Cluster.cluster_id,
+        translation_key="charging_protection",
+        fallback_name="Charging Protection",
+    )
+    .number(
+        attribute_name="charging_limit",
+        cluster_id=PlugAEU001Cluster.cluster_id,
+        min_value=0.1,
+        max_value=2,
+        step=0.1,
+        unit=UnitOfPower.WATT,
+        translation_key="charging_limit",
+        fallback_name="Charging Limit",
+    )
+    .add_to_registry()
+)