Merge pull request #1165 from Necroneco/wd1

Add more details to Linptech Sliding Window Driver WD1
AlexxIT · Oct 15, 2023 · 55627fc · 55627fc
2 parents 000042e + bf2defe
commit 55627fc
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/custom_components/xiaomi_gateway3/core/converters/devices.py b/custom_components/xiaomi_gateway3/core/converters/devices.py
@@ -2672,6 +2672,9 @@
         Converter("target_position", mi="2.p.3"),
         CurtainPosConv("position", mi="2.p.2", parent="motor"),
         Converter("battery", "sensor", mi="3.p.1"),
+        Converter("security_mode", "switch", mi="4.p.6"),
+        Converter("power_replenishment", "sensor", mi="7.p.1", enabled=None),
+        Converter("realtime_current_in", "sensor", mi="7.p.2", enabled=None),
     ],
 }, {
     # https://home.miot-spec.com/spec/yeelink.curtain.crc1

diff --git a/custom_components/xiaomi_gateway3/core/entity.py b/custom_components/xiaomi_gateway3/core/entity.py
@@ -39,6 +39,7 @@
     "reverse": BinarySensorDeviceClass.LOCK,
     "square": BinarySensorDeviceClass.LOCK,
     "water_leak": BinarySensorDeviceClass.MOISTURE,
+    "realtime_current_in": SensorDeviceClass.CURRENT,
 }
 
 ICONS = {
@@ -131,6 +132,10 @@
     "min_brightness": EntityCategory.CONFIG,
     # Linptech Mesh Triple Wall Switch (no N)
     "compatible_mode": EntityCategory.CONFIG,
+    # Linptech Sliding Window Driver WD1
+    "security_mode": EntityCategory.CONFIG,
+    "power_replenishment": EntityCategory.DIAGNOSTIC,
+    "realtime_current_in": EntityCategory.DIAGNOSTIC,
 }
 
 STATE_TIMEOUT = timedelta(minutes=10)

diff --git a/custom_components/xiaomi_gateway3/sensor.py b/custom_components/xiaomi_gateway3/sensor.py
@@ -78,6 +78,9 @@ def new_entity(gateway: XGateway, device: XDevice, conv: Converter) -> XEntity:
     # "msg_received": "msg",
     # "msg_missed": "msg",
     # "unresponsive": "times"
+    "power_replenishment": "mAh",
+    # Deprecated: please use UnitOfElectricCurrent.MILLIAMPERE.
+    "realtime_current_in": ELECTRIC_CURRENT_MILLIAMPERE,
 }
 
 # https://developers.home-assistant.io/docs/core/entity/sensor/#long-term-statistics