tplinkrouterc6u 5.13.0__py3-none-any.whl → 5.14.0__py3-none-any.whl

test/test_client_ex.py

@@ -284,6 +284,125 @@ class TestTPLinkEXClient(TestCase):
         self.assertEqual(status.devices[1].packets_sent, None)  # TODO
         self.assertEqual(status.devices[1].packets_received, None)  # TODO
 
+    def test_get_status_M8550(self) -> None:
+        DEV2_ADT_LAN = ('{"data":[{"MACAddress":"a0:28:84:de:dd:5c","IPAddress":"192.168.4.1","stack":"1,0,0,0,0,0"}],'
+                        '"operation":"gl","oid":"DEV2_ADT_LAN","success":true}')
+        DEV2_ADT_WAN = ('{"data":[{"enable":"1","MACAddr":"","connIPv4Address":"","connIPv4Gateway":""'
+                        ',"stack":"1,0,0,0,0,0"}],"operation":"gl","oid":"DEV2_ADT_WAN","success":true}')
+        DEV2_ADT_WIFI_COMMON = ('{"data": [{"band": "2.4GHz","standard": "bgnax","bandwidth": "Auto",'
+                                '"currentBandwidth": "20MHz","autoChannel": "1","channel": "1","transmitPower": "100",'
+                                '"beaconInterval": "100","RTSThreshold": "2346","DTIMPeriod": "1",'
+                                '"fragmentThreshold": "2346","regulatoryDomain": "DE ","guardInterval": "400nsec",'
+                                '"primaryEnable": "1","primaryBSSID": "b8:7a:96:53:db:e2","primaryStatus": "Enabled",'
+                                '"primarySSID": "JustAnotherWiFi","primarySSIDAdvertise": "1",'
+                                '"primaryModeEnabled": "WPA2-Personal","primaryWPAWPA2EncryptionMode": "AES",'
+                                '"primaryPSK": "","primaryGroupKeyUpdateInterval": "0","primaryRadiusServerIP": "",'
+                                '"primaryRadiusServerPort": "1812","primaryRadiusServerSecret": "",'
+                                '"primaryIsolationEnable": "0","primaryWMMEnable": "1","primaryTWTEnable": "0",'
+                                '"PSCEnable": "0","WPSEnable": "0","WPSDevicePassword": "","WPSAPPinLock": "0",'
+                                '"WPSPeerPassword": "0","WPSState": "Idle","WPSStartWPS": "0","WPSSetupLock": "1",'
+                                '"ACLEnable": "0","ACLRule": "","X_TP_LteCoex": "0","DMSaveOnly": "0",'
+                                '"stack": "1,0,0,0,0,0"}, {"band": "5GHz","standard": "anacax","bandwidth": "Auto",'
+                                '"currentBandwidth": "80MHz","autoChannel": "1","channel": "44","transmitPower": "100",'
+                                '"beaconInterval": "100","RTSThreshold": "2346","DTIMPeriod": "1",'
+                                '"fragmentThreshold": "2346","regulatoryDomain": "DE ","guardInterval": "400nsec",'
+                                '"primaryEnable": "1","primaryBSSID": "b8:7a:96:53:db:e2","primaryStatus": "Enabled",'
+                                '"primarySSID": "JustAnotherWiFi","primarySSIDAdvertise": "1",'
+                                '"primaryModeEnabled": "WPA2-Personal","primaryWPAWPA2EncryptionMode": "AES",'
+                                '"primaryPSK": "","primaryGroupKeyUpdateInterval": "0","primaryRadiusServerIP": "",'
+                                '"primaryRadiusServerPort": "1812","primaryRadiusServerSecret": "",'
+                                '"primaryIsolationEnable": "0","primaryWMMEnable": "1","primaryTWTEnable": "0",'
+                                '"PSCEnable": "0","WPSEnable": "0","WPSDevicePassword": "","WPSAPPinLock": "0",'
+                                '"WPSPeerPassword": "0","WPSState": "Idle","WPSStartWPS": "0","WPSSetupLock": "1",'
+                                '"ACLEnable": "0","ACLRule": "","X_TP_LteCoex": "0","DMSaveOnly": "0",'
+                                '"stack": "2,0,0,0,0,0"}, {"band": "6GHz","standard": "ax","bandwidth": "Auto",'
+                                '"currentBandwidth": "","autoChannel": "1","channel": "37","transmitPower": "100",'
+                                '"beaconInterval": "100","RTSThreshold": "2346","DTIMPeriod": "1",'
+                                '"fragmentThreshold": "2346","regulatoryDomain": "DE ","guardInterval": "400nsec",'
+                                '"primaryEnable": "0","primaryBSSID": "b8:7a:96:53:db:e2","primaryStatus": "Disabled",'
+                                '"primarySSID": "JustAnotherWiFi","primarySSIDAdvertise": "1",'
+                                '"primaryModeEnabled": "WPA3-Personal","primaryWPAWPA2EncryptionMode": "AES",'
+                                '"primaryPSK": "","primaryGroupKeyUpdateInterval": "0","primaryRadiusServerIP": "",'
+                                '"primaryRadiusServerPort": "1812","primaryRadiusServerSecret": "",'
+                                '"primaryIsolationEnable": "0","primaryWMMEnable": "1","primaryTWTEnable": "0",'
+                                '"PSCEnable": "1","WPSEnable": "0","WPSDevicePassword": "0","WPSAPPinLock": "0",'
+                                '"WPSPeerPassword": "0","WPSState": "","WPSStartWPS": "0","WPSSetupLock": "0",'
+                                '"ACLEnable": "0","ACLRule": "","X_TP_LteCoex": "0","DMSaveOnly": "0",'
+                                '"stack": "3,0,0,0,0,0"}],"operation": "gl","oid": "DEV2_ADT_WIFI_COMMON",'
+                                '"success": true}')
+        DEV2_HOST_ENTRY = ('{"data":[{"active":"1","X_TP_LanConnType":"1","physAddress":"66-E2-02-BD-B5-1B",'
+                           '"IPAddress":"192.168.30.10","hostName":"host1","stack":"1,0,0,0,0,0"},'
+                           '{"active":"1","X_TP_LanConnType":"1","physAddress":"F4-A3-86-2D-41-B5",'
+                           '"IPAddress":"192.168.30.11","hostName":"host2","stack":"2,0,0,0,0,0"}],"operation":"gl",'
+                           '"oid":"DEV2_HOST_ENTRY","success":true}')
+        DEV2_MEM_STATUS = ('{"data":{"total":"192780","free":"78400","stack":"0,0,0,0,0,0"},"operation":"go",'
+                           '"oid":"DEV2_MEM_STATUS","success":true}')
+        DEV2_PROC_STATUS = ('{"data":{"CPUUsage":"69","stack":"0,0,0,0,0,0"},"operation":"go",'
+                            '"oid":"DEV2_PROC_STATUS","success":true}')
+
+        class TPLinkEXClientTest(TPLinkEXClient):
+            self._token = True
+
+            def _request(self, url, method='POST', data_str=None, encrypt=False):
+                if 'DEV2_ADT_LAN' in data_str:
+                    return 200, DEV2_ADT_LAN
+                elif 'DEV2_ADT_WAN' in data_str:
+                    return 200, DEV2_ADT_WAN
+                elif 'DEV2_ADT_WIFI_COMMON' in data_str:
+                    return 200, DEV2_ADT_WIFI_COMMON
+                elif 'DEV2_HOST_ENTRY' in data_str:
+                    return 200, DEV2_HOST_ENTRY
+                elif 'DEV2_MEM_STATUS' in data_str:
+                    return 200, DEV2_MEM_STATUS
+                elif 'DEV2_PROC_STATUS' in data_str:
+                    return 200, DEV2_PROC_STATUS
+                raise ClientException()
+
+        client = TPLinkEXClientTest('', '')
+        status = client.get_status()
+
+        self.assertIsInstance(status, Status)
+        self.assertEqual(status.wan_macaddr, None)
+        self.assertEqual(status.lan_macaddr, 'A0-28-84-DE-DD-5C')
+        self.assertIsInstance(status.lan_macaddress, EUI48)
+        self.assertEqual(status.wan_ipv4_addr, None)
+        self.assertEqual(status.lan_ipv4_addr, '192.168.4.1')
+        self.assertEqual(status.wan_ipv4_gateway, None)
+        self.assertEqual(status.wired_total, 0)
+        self.assertEqual(status.wifi_clients_total, 2)
+        self.assertEqual(status.guest_clients_total, 0)
+        self.assertEqual(status.clients_total, 2)
+        self.assertEqual(status.guest_2g_enable, None)
+        self.assertEqual(status.guest_5g_enable, None)
+        self.assertEqual(status.iot_2g_enable, None)
+        self.assertEqual(status.iot_5g_enable, None)
+        self.assertEqual(status.wifi_2g_enable, True)
+        self.assertEqual(status.wifi_5g_enable, True)
+        self.assertEqual(status.wan_ipv4_uptime, None)
+        self.assertGreaterEqual(status.mem_usage, 0)
+        self.assertLessEqual(status.mem_usage, 1)
+        self.assertGreaterEqual(status.cpu_usage, 0)
+        self.assertLessEqual(status.cpu_usage, 1)
+        self.assertEqual(len(status.devices), 2)
+        self.assertIsInstance(status.devices[0], Device)
+        self.assertEqual(status.devices[0].type, Connection.HOST_2G)
+        self.assertEqual(status.devices[0].macaddr, '66-E2-02-BD-B5-1B')
+        self.assertIsInstance(status.devices[0].macaddress, EUI48)
+        self.assertEqual(status.devices[0].ipaddr, '192.168.30.10')
+        self.assertIsInstance(status.devices[0].ipaddress, IPv4Address)
+        self.assertEqual(status.devices[0].hostname, 'host1')
+        self.assertEqual(status.devices[0].packets_sent, None)
+        self.assertEqual(status.devices[0].packets_received, None)
+        self.assertIsInstance(status.devices[1], Device)
+        self.assertEqual(status.devices[1].type, Connection.HOST_2G)
+        self.assertEqual(status.devices[1].macaddr, 'F4-A3-86-2D-41-B5')
+        self.assertIsInstance(status.devices[1].macaddress, EUI48)
+        self.assertEqual(status.devices[1].ipaddr, '192.168.30.11')
+        self.assertIsInstance(status.devices[1].ipaddress, IPv4Address)
+        self.assertEqual(status.devices[1].hostname, 'host2')
+        self.assertEqual(status.devices[1].packets_sent, None)  # TODO
+        self.assertEqual(status.devices[1].packets_received, None)  # TODO
+
     def test_get_ipv4_reservations(self) -> None:
 
         response = ('{"data":[{"enable":"1","chaddr":"bf:75:44:4c:dc:9e","yiaddr":"192.168.8.21",'

test/test_client_mr.py

@@ -93,7 +93,7 @@ softwareVersion=1.1
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -158,7 +158,7 @@ X_TP_TotalPacketsReceived=467
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -236,7 +236,7 @@ active=1
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -299,7 +299,7 @@ X_TP_TotalPacketsReceived=467
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -356,7 +356,7 @@ name=wlan1
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -407,7 +407,7 @@ name=wlan1
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -450,7 +450,7 @@ yiaddr=192.168.8.21
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -470,7 +470,7 @@ yiaddr=192.168.8.21
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -485,7 +485,7 @@ yiaddr=192.168.8.21
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -505,7 +505,7 @@ leaseTimeRemaining=85841
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -535,7 +535,7 @@ leaseTimeRemaining=86372
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -580,7 +580,7 @@ DNSServers=7.7.7.7,2.2.2.2
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -611,7 +611,7 @@ DNSServers=7.7.7.7,2.2.2.2
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -645,7 +645,7 @@ DNSServers=0.0.0.0,0.0.0.0
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -675,7 +675,7 @@ DNSServers=0.0.0.0,0.0.0.0
         check_data = ''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 nonlocal check_url, check_data
                 check_url = url
                 check_data = data_str
@@ -697,7 +697,7 @@ DNSServers=0.0.0.0,0.0.0.0
         check_data = ''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 nonlocal check_url, check_data
                 check_url = url
                 check_data = data_str
@@ -733,7 +733,7 @@ ussdStatus=1
         check_data = []
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 check_url.append(url)
                 check_data.append(data_str)
                 return 200, responses.pop(0)
@@ -774,7 +774,7 @@ ussdStatus=2
         check_data = []
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 check_url.append(url)
                 check_data.append(data_str)
                 return 200, responses.pop(0)
@@ -812,7 +812,7 @@ unread=0
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -838,7 +838,7 @@ unread=0
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -856,7 +856,7 @@ unread=0
         check_data = ''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 nonlocal check_url, check_data
                 check_url = url
                 check_data = data_str
@@ -878,7 +878,7 @@ unread=0
         check_data = ''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 nonlocal check_url, check_data
                 check_url = url
                 check_data = data_str
@@ -922,7 +922,7 @@ ispName=Name
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -975,7 +975,7 @@ ispName=Name
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -1033,7 +1033,7 @@ connAct=0
 '''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 return 200, response
 
         client = TPLinkMRClientTest('', '')
@@ -1055,7 +1055,7 @@ connAct=0
         check_data = ''
 
         class TPLinkMRClientTest(TPLinkMRClient):
-            def _request(self, url, method='POST', data_str=None, encrypt=False):
+            def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
                 nonlocal check_url, check_data
                 check_url = url
                 check_data = data_str

tplinkrouterc6u/__init__.py

@@ -3,11 +3,13 @@ from tplinkrouterc6u.client.deco import TPLinkDecoClient
 from tplinkrouterc6u.client_abstract import AbstractRouter
 from tplinkrouterc6u.client.mr import TPLinkMRClient, TPLinkMRClientGCM
 from tplinkrouterc6u.client.mr200 import TPLinkMR200Client
+from tplinkrouterc6u.client.mr6400v7 import TPLinkMR6400v7Client
 from tplinkrouterc6u.client.ex import TPLinkEXClient, TPLinkEXClientGCM
 from tplinkrouterc6u.client.vr import TPLinkVRClient
 from tplinkrouterc6u.client.vr400v2 import TPLinkVR400v2Client
 from tplinkrouterc6u.client.c80 import TplinkC80Router
 from tplinkrouterc6u.client.c5400x import TplinkC5400XRouter
+from tplinkrouterc6u.client.c3200 import TplinkC3200Router
 from tplinkrouterc6u.client.c1200 import TplinkC1200Router
 from tplinkrouterc6u.client.xdr import TPLinkXDRClient
 from tplinkrouterc6u.client.wdr import TplinkWDRRouter

tplinkrouterc6u/client/c3200.py

@@ -0,0 +1,198 @@
+import base64
+import re
+from time import sleep
+from urllib.parse import urlparse
+
+import requests
+from requests import Response
+
+from tplinkrouterc6u.client.mr200 import TPLinkMR200Client
+from tplinkrouterc6u.common.package_enum import VPN
+from tplinkrouterc6u.common.dataclass import (
+    LTEStatus,
+)
+from tplinkrouterc6u.common.exception import ClientException, ClientError
+from tplinkrouterc6u.common.package_enum import Connection
+
+
+class TplinkC3200Router(TPLinkMR200Client):
+    # This is a session variable that will contain everything needed as soon as the router is connected.
+    #  - The "Referer" header which allows to be accepted by the CGI module.
+    # - the Authentification cookie
+    SESSION: requests.Session
+
+    # Possible retries limit
+    REQUEST_RETRIES = 1
+
+    # Router name to be included in logs for example,
+    # or to be redefined by subclasses.
+    ROUTER_NAME = "TP Link Router C3200"
+
+    # Connection method
+    def supports(self) -> bool:
+        if len(self.password) > 125:
+            return False
+
+        try:
+            # This method checks if we can recognize the router type.
+            welcome_page = requests.get(self.host, timeout=5)
+            if welcome_page and welcome_page.status_code == 200 and re.search("Archer", welcome_page.text):
+                return True
+        except ClientException:
+            pass
+
+        return False
+
+    def authorize(self) -> None:
+
+        # ———————————————————————————————————————————
+        # Create the SESSION object and the authorization cookie
+        # ———————————————————————————————————————————
+        self.SESSION = requests.Session()
+
+        if self._logger:
+            self._logger.debug("!")
+        # We need to extract the domain form the host to fill the cookie.
+        router_host = urlparse(self.host).hostname
+        if not router_host:
+            raise ValueError(self.host & " must contain a valid host, ex. http://192.168.168.1")
+
+        """Return the string "Basic <base64(username:password)>"."""
+        token_bytes = f"{self.username}:{self.password}".encode()
+        encoded = base64.b64encode(token_bytes).decode()
+        auth_cookie_value = f"Basic {encoded}"
+
+        self.SESSION.cookies.set(
+            name="Authorization",
+            value=auth_cookie_value,
+            domain=router_host,
+            path="/",
+        )
+
+        self.SESSION.headers = {"Referer": f"{self.host}/", "Origin": self.host}
+
+        login_url = '{}/'.format(self.host)
+        response = Response()
+
+        try:
+            response = self.SESSION.post(login_url, timeout=10)
+        except Exception as e:
+            error = self.ROUTER_NAME + " - Cannot authorize! Error - {}; Response - {}".format(e, response.text)
+            if self._logger:
+                self._logger.debug(error)
+            raise ClientException(error)
+
+    def logout(self) -> None:
+        self.SESSION.cookies.clear(domain=urlparse(self.host).hostname, path="/")
+
+    def get_lte_status(self) -> LTEStatus:
+        pass
+
+    def _get_params(self, retry=False) -> None:
+        pass
+
+    def set_vpn(self, vpn: VPN, enable: bool) -> None:
+        # Unable to test it on my C3200
+        pass
+
+    def req_act(self, acts: list):
+        act_types, act_data = self._fill_acts(acts)
+
+        url = f"{self.host}/cgi?" + '&'.join(act_types)
+        data_str = ''.join(act_data)
+        (code, response) = self._request(url, data_str=data_str)
+
+        if code != 200:
+            error = self.ROUTER_NAME + ' -  Response with error; Request {} - Response {}'.format(data_str, response)
+            if self._logger:
+                self._logger.debug(error)
+            raise ClientError(error)
+
+        result = self._merge_response(response)
+
+        return response, result.get('0') if len(result) == 1 and result.get('0') else result
+
+    def _request(self, url, method='POST', data_str=None, encrypt=False, is_login=False):
+        r = Response()
+
+        retry = 0
+        while retry < self.REQUEST_RETRIES:
+            # send the request
+            if method == 'POST':
+                r = self.SESSION.post(url, data=data_str)
+            elif method == 'GET':
+                r = self.SESSION.get(url, data=data_str)
+            else:
+                raise Exception('Unsupported method ' + str(method))
+
+            # sometimes we get 500 here, not sure why... just retry the request
+            if (r.status_code not in [500, 406]
+                    and '<title>500 Internal Server Error</title>' not in r.text
+                    and '<title>406 Not Acceptable</title>' not in r.text):
+                break
+
+            sleep(0.1)
+            retry += 1
+
+        return r.status_code, r.text
+
+    # Overriding the method since we have two 5G bands in the rooter.
+    # We manage both as one.
+    def set_wifi(self, wifi: Connection, enable: bool) -> None:
+        acts = []
+
+        match wifi:
+            case Connection.HOST_2G:
+                acts = [
+                    self.ActItem(
+                        self.ActItem.SET,
+                        'LAN_WLAN',
+                        '1,1,0,0,0,0',
+                        attrs=['enable={}'.format(int(enable))]),
+                ]
+            case Connection.HOST_5G:
+                # We activate both 5G bands
+                acts = [
+                    self.ActItem(
+                        self.ActItem.SET,
+                        'LAN_WLAN',
+                        '1,2,0,0,0,0',
+                        attrs=['enable={}'.format(int(enable))]),
+                    self.ActItem(
+                        self.ActItem.SET,
+                        'LAN_WLAN',
+                        '1,3,0,0,0,0',
+                        attrs=['enable={}'.format(int(enable))]),
+                ]
+            case Connection.GUEST_2G:
+                acts = [
+                    self.ActItem(
+                        self.ActItem.SET,
+                        'LAN_WLAN_MSSIDENTRY',
+                        '1,1,1,0,0,0',
+                        attrs=['enable={}'.format(int(enable))]),
+                ]
+            case Connection.GUEST_5G:
+                acts = [
+                    self.ActItem(
+                        self.ActItem.SET,
+                        'LAN_WLAN_MSSIDENTRY',
+                        '1,2,1,0,0,0',
+                        attrs=['enable={}'.format(int(enable))]),
+                    self.ActItem(
+                        self.ActItem.SET,
+                        'LAN_WLAN_MSSIDENTRY',
+                        '1,3,1,0,0,0',
+                        attrs=['enable={}'.format(int(enable))]),
+                ]
+        self.req_act(acts)
+
+    def reboot(self) -> None:
+        # CGI 7 et [ACT_REBOOT#0,0,0,0,0,0#0,0,0,0,0,0]0,0
+
+        acts = [
+            self.ActItem(self.ActItem.OP, 'ACT_REBOOT'),
+        ]
+        _, values = self.req_act(acts)
+
+        # print(values.keys())

tplinkrouterc6u/client/ex.py

@@ -92,7 +92,7 @@ class TPLinkEXClient(TPLinkMRClientBase):
             self.ActItem(self.ActItem.GL, 'DEV2_ADT_LAN', attrs=['MACAddress', 'IPAddress']),
             self.ActItem(self.ActItem.GL, 'DEV2_ADT_WAN',
                          attrs=['enable', 'MACAddr', 'connIPv4Address', 'connIPv4Gateway']),
-            self.ActItem(self.ActItem.GL, 'DEV2_ADT_WIFI_COMMON', attrs=['primaryEnable', 'guestEnable']),
+            self.ActItem(self.ActItem.GL, 'DEV2_ADT_WIFI_COMMON'),
             self.ActItem(self.ActItem.GL, 'DEV2_HOST_ENTRY',
                          attrs=['active', 'X_TP_LanConnType', 'physAddress', 'IPAddress', 'hostName']),
             self.ActItem(self.ActItem.GO, 'DEV2_MEM_STATUS', attrs=['total', 'free']),
@@ -117,12 +117,14 @@ class TPLinkEXClient(TPLinkMRClientBase):
         if values[2]:
             if values[2].__class__ != list:
                 status.wifi_2g_enable = bool(int(values[2]['primaryEnable']))
-                status.guest_2g_enable = bool(int(values[2]['guestEnable']))
+                status.guest_2g_enable = bool(int(values[2]['guestEnable'])) if values[2].get('guestEnable') else None
             else:
                 status.wifi_2g_enable = bool(int(values[2][0]['primaryEnable']))
                 status.wifi_5g_enable = bool(int(values[2][1]['primaryEnable']))
-                status.guest_2g_enable = bool(int(values[2][0]['guestEnable']))
-                status.guest_5g_enable = bool(int(values[2][1]['guestEnable']))
+                status.guest_2g_enable = bool(int(values[2][0]['guestEnable'])) \
+                    if values[2][0].get('guestEnable') else None
+                status.guest_5g_enable = bool(int(values[2][1]['guestEnable'])) \
+                    if values[2][1].get('guestEnable') else None
 
         devices = {}
         for val in self._to_list(values[3]):