WiFi Repeater: Wifi roaming and signal strange indicator added (#285)

miio/discovery.py

@@ -5,7 +5,7 @@
 import codecs
 from . import (Device, Vacuum, ChuangmiPlug, PowerStrip, AirPurifier, Ceil,
                PhilipsBulb, PhilipsEyecare, ChuangmiIr, AirHumidifier,
-               WaterPurifier, WifiSpeaker, Yeelight)
+               WaterPurifier, WifiSpeaker, WifiRepeater, Yeelight)
 from .chuangmi_plug import (MODEL_CHUANGMI_PLUG_V1, MODEL_CHUANGMI_PLUG_V3,
                             MODEL_CHUANGMI_PLUG_M1)
 
@@ -48,6 +48,8 @@
     "philips-light-zyceiling": Ceil,
     "philips-light-sread1": PhilipsEyecare,  # name needs to be checked
     "xiaomi-wifispeaker-v1": WifiSpeaker,  # name needs to be checked
+    "xiaomi-repeater-v1": WifiRepeater,  # name needs to be checked
+    "xiaomi-repeater-v3": WifiRepeater,  # name needs to be checked
     "yeelink-light-": Yeelight,
     "lumi-gateway-": lambda x: other_package_info(
         x, "https://github.com/Danielhiversen/PyXiaomiGateway")

miio/tests/test_wifirepeater.py

@@ -100,6 +100,12 @@ def configuration(self):
     def info(self):
         return self.device.info()
 
+    def wifi_roaming(self):
+        return self.device.wifi_roaming()
+
+    def rssi_accesspoint(self):
+        return self.device.rssi_accesspoint()
+
     def test_status(self):
         self.device._reset_state()
 
@@ -110,10 +116,10 @@ def test_status(self):
 
     def test_set_wifi_roaming(self):
         self.device.set_wifi_roaming(True)
-        assert self.info().raw['desc']['wifi_explorer'] == 1
+        assert self.wifi_roaming() is True
 
         self.device.set_wifi_roaming(False)
-        assert self.info().raw['desc']['wifi_explorer'] == 0
+        assert self.wifi_roaming() is False
 
     def test_configuration(self):
         self.device._reset_state()
@@ -132,14 +138,17 @@ def configuration():
             'ssid': 'SSID2',
             'password': 'PASSWORD2',
             'hidden': True,
-            'wifi_explorer': False
         }
 
         self.device.set_configuration(
             dummy_configuration['ssid'],
             dummy_configuration['password'],
-            dummy_configuration['hidden'],
-            dummy_configuration['wifi_explorer'])
+            dummy_configuration['hidden'])
         assert configuration().ssid == dummy_configuration['ssid']
         assert configuration().password == dummy_configuration['password']
         assert configuration().ssid_hidden is dummy_configuration['hidden']
+
+    def test_rssi_accesspoint(self):
+        self.device._reset_state()
+
+        assert self.rssi_accesspoint() is self.device.start_device_info['ap']['rssi']

miio/wifirepeater.py

@@ -1,9 +1,13 @@
 import logging
-from .device import Device
+from .device import Device, DeviceException
 
 _LOGGER = logging.getLogger(__name__)
 
 
+class WifiRepeaterException(DeviceException):
+    pass
+
+
 class WifiRepeaterStatus:
     def __init__(self, data):
         """
@@ -86,12 +90,19 @@ def set_wifi_roaming(self, wifi_roaming: bool):
             'wifi_explorer': int(wifi_roaming)
         }])
 
-    def set_configuration(self, ssid: str, password: str, hidden: bool = False,
-                          wifi_roaming: bool = False):
+    def set_configuration(self, ssid: str, password: str, hidden: bool = False):
         """Update the configuration of the accesspoint."""
         return self.send("miIO.switch_wifi_ssid", [{
             'ssid': ssid,
             'pwd': password,
             'hidden': int(hidden),
-            'wifi_explorer': int(wifi_roaming)
+            'wifi_explorer': 0
         }])
+
+    def wifi_roaming(self) -> bool:
+        """Return the roaming setting."""
+        return self.info().raw['desc']['wifi_explorer'] == 1
+
+    def rssi_accesspoint(self) -> int:
+        """Received signal strength indicator of the accesspoint."""
+        return self.info().accesspoint['rssi']