diff --git a/zhaquirks/sonoff/swv.py b/zhaquirks/sonoff/swv.py index 74e2a6a984..984558246a 100644 --- a/zhaquirks/sonoff/swv.py +++ b/zhaquirks/sonoff/swv.py @@ -31,6 +31,12 @@ class AttributeDefs(BaseAttributeDefs): type=ValveState, ) + auto_close_water_shortage = ZCLAttributeDef( + id=0x5011, + type=t.uint16_t, + is_manufacturer_specific=True, + ) + ( QuirkBuilder("SONOFF", "SWV") @@ -56,5 +62,13 @@ class AttributeDefs(BaseAttributeDefs): translation_key="water_supply", fallback_name="Water supply", ) + .switch( + CustomSonoffCluster.AttributeDefs.auto_close_water_shortage.name, + CustomSonoffCluster.cluster_id, + off_value=0, + on_value=30, + translation_key="water_shortage_auto_close", + fallback_name="Water shortage auto-close", + ) .add_to_registry() )