fix vibration sensor not reset

Author: mikechouto

custom_components/aqara_gateway/binary_sensor.py

@@ -159,6 +159,10 @@ def update(self, data: dict = None):
                     self._state = not value
 
         self.schedule_update_ha_state()
+    
+    def reset_state(self, state: str = ''):
+        self._state = state
+        self.async_write_ha_state()
 
 
 class GatewayMotionSensor(GatewayBinarySensor):
@@ -598,10 +602,6 @@ def update(self, data: dict = None):
 
         self.schedule_update_ha_state()
 
-    def reset_state(self):
-        self._state = ''
-        self.async_write_ha_state()
-
 
 class GatewayAction(GatewayBinarySensor, BinarySensorEntity):
     """ Xiaomi/Aqara Action Cube """
@@ -714,9 +714,11 @@ def update(self, data: dict = None):
                 'entity_id': self.entity_id, 'click_type': self._state
             })
 
-            time.sleep(.1)
-
-            self._state = ''
+            # reset the state to empty after 0.1 second
+            if self.device['model'] in ('lumi.vibration.aq1'):
+                self.hass.loop.call_later(.1, self.reset_state('none'))
+            else:
+                self.hass.loop.call_later(.1, self.reset_state)
 
         self.schedule_update_ha_state()