tplinkrouterc6u 5.8.0__py3-none-any.whl → 5.9.0__py3-none-any.whl

test/test_client_c1200.py

@@ -5,7 +5,6 @@ from tplinkrouterc6u import (
     Connection,
     ClientException,
 )
-from tplinkrouterc6u.common.package_enum import VPN
 
 
 class TestTPLinkC1200Client(TestCase):
@@ -201,13 +200,13 @@ class TestTPLinkC1200Client(TestCase):
                 ignore_response: bool = False,
                 ignore_errors: bool = False,
             ) -> dict | None:
-                if path == "/admin/openvpn?form=config&operation=read":
+                if path == "admin/openvpn?form=config&operation=read":
                     return loads(response_openvpn_read)
-                if path == "/admin/pptpd?form=config&operation=read":
+                if path == "admin/pptpd?form=config&operation=read":
                     return loads(response_pptp_read)
-                if path == "/admin/vpnconn?form=config&operation=list&vpntype=openvpn":
+                if path == "admin/vpnconn?form=config&operation=list&vpntype=openvpn":
                     return loads(respone_vpnconn_openvpn)
-                if path == "/admin/vpnconn?form=config&operation=list&vpntype=pptp":
+                if path == "admin/vpnconn?form=config&operation=list&vpntype=pptp":
                     return loads(respone_vpnconn_pptpvpn)
                 raise ClientException()
 
@@ -219,41 +218,6 @@ class TestTPLinkC1200Client(TestCase):
         self.assertEqual(vpn_status.openvpn_clients_total, 2)
         self.assertEqual(vpn_status.pptpvpn_clients_total, 3)
 
-    def test_set_vpn(self) -> None:
-        response_openvpn_read = """
-        {
-            "enabled": "on",
-            "proto": "udp",
-            "access": "home",
-            "cert_exist": true,
-            "mask": "255.255.255.0",
-            "port": "1194",
-            "serverip": "10.8.0.0"
-        }
-        """
-
-        class TPLinkRouterTest(TplinkC1200Router):
-            def request(
-                self,
-                path: str,
-                data: str,
-                ignore_response: bool = False,
-                ignore_errors: bool = False,
-            ) -> dict | None:
-                if path == "/admin/openvpn?form=config&operation=read":
-                    return loads(response_openvpn_read)
-                self.captured_path = path
-
-        client = TPLinkRouterTest("", "")
-        client.set_vpn(VPN.OPEN_VPN, True)
-
-        expected_path = (
-            "/admin/openvpn?form=config&operation=write&enabled=on"
-            "&proto=udp&access=home&cert_exist=True"
-            "&mask=255.255.255.0&port=1194&serverip=10.8.0.0"
-        )
-        self.assertEqual(client.captured_path, expected_path)
-
 
 if __name__ == "__main__":
     main()

test/test_client_c6u.py

@@ -9,6 +9,7 @@ from tplinkrouterc6u import (
     IPv4Status,
     Device,
     ClientException,
+    VPN,
 )
 
 
@@ -768,6 +769,107 @@ class TestTPLinkClient(TestCase):
         self.assertEqual(result.wan_ipv4_gateway, '0.0.0.0')
         self.assertEqual(result.lan_macaddr, '06-E6-97-9E-23-F5')
 
+    def test_vpn_status(self) -> None:
+        response_openvpn_read = """
+        {
+            "enabled": "on",
+            "proto": "udp",
+            "access": "home",
+            "cert_exist": true,
+            "mask": "255.255.255.0",
+            "port": "1194",
+            "serverip": "10.8.0.0"
+        }
+        """
+
+        response_pptp_read = """
+        {
+            "enabled": "off",
+            "unencrypted_access": "on",
+            "samba_access": "on",
+            "netbios_pass": "on",
+            "remoteip": "10.0.0.11-20"
+        }
+        """
+
+        respone_vpnconn_openvpn = """[
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "openvpn", "key": "7450"},
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "openvpn", "key": "7450"}
+        ]"""
+
+        respone_vpnconn_pptpvpn = """[
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "pptp", "key": "7450"},
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "pptp", "key": "7450"},
+            {"username": "admin", "remote_ip": "192.168.0.200", "ipaddr": "10.0.0.11",
+             "extra": "7450", "vpntype": "pptp", "key": "7450"}
+        ]"""
+
+        class TPLinkRouterTest(TplinkRouter):
+            def request(
+                self,
+                path: str,
+                data: str,
+                ignore_response: bool = False,
+                ignore_errors: bool = False,
+            ) -> dict | None:
+                if path == "admin/openvpn?form=config":
+                    return loads(response_openvpn_read)
+                if path == "admin/pptpd?form=config":
+                    return loads(response_pptp_read)
+                if path == "admin/vpnconn?form=config" and data == "operation=list&vpntype=openvpn":
+                    return loads(respone_vpnconn_openvpn)
+                if path == "admin/vpnconn?form=config" and data == "operation=list&vpntype=pptp":
+                    return loads(respone_vpnconn_pptpvpn)
+                raise ClientException()
+
+        client = TPLinkRouterTest("", "")
+
+        vpn_status = client.get_vpn_status()
+        self.assertTrue(vpn_status.openvpn_enable)
+        self.assertFalse(vpn_status.pptpvpn_enable)
+        self.assertEqual(vpn_status.openvpn_clients_total, 2)
+        self.assertEqual(vpn_status.pptpvpn_clients_total, 3)
+
+    def test_set_vpn(self) -> None:
+        response_openvpn_read = """
+        {
+            "enabled": "on",
+            "proto": "udp",
+            "access": "home",
+            "cert_exist": true,
+            "mask": "255.255.255.0",
+            "port": "1194",
+            "serverip": "10.8.0.0"
+        }
+        """
+
+        class TPLinkRouterTest(TplinkRouter):
+            def request(
+                self,
+                path: str,
+                data: str,
+                ignore_response: bool = False,
+                ignore_errors: bool = False,
+            ) -> dict | None:
+                if path == "admin/openvpn?form=config" and data == "operation=read":
+                    return loads(response_openvpn_read)
+                self.captured_path = path
+                self.captured_data = data
+
+        client = TPLinkRouterTest("", "")
+        client.set_vpn(VPN.OPEN_VPN, True)
+
+        expected_data = (
+            "operation=write&enabled=on"
+            "&proto=udp&access=home&cert_exist=True&mask=255.255.255.0&port=1194&serverip=10.8.0.0"
+        )
+        self.assertEqual(client.captured_path, 'admin/openvpn?form=config')
+        self.assertEqual(client.captured_data, expected_data)
+
 
 if __name__ == '__main__':
     main()

tplinkrouterc6u/client/c1200.py

@@ -1,11 +1,8 @@
 from re import search
 from requests import post, Response
-from urllib.parse import urlencode
 from tplinkrouterc6u.common.encryption import EncryptionWrapper
 from tplinkrouterc6u.common.exception import ClientException, AuthorizeError
 from tplinkrouterc6u.client.c5400x import TplinkC5400XRouter
-from tplinkrouterc6u.common.dataclass import VPNStatus
-from tplinkrouterc6u.common.package_enum import VPN
 
 
 class TplinkC1200Router(TplinkC5400XRouter):
@@ -103,36 +100,3 @@ class TplinkC1200Router(TplinkC5400XRouter):
     @staticmethod
     def _get_login_data(crypted_pwd: str) -> str:
         return 'operation=login&password={}'.format(crypted_pwd)
-
-    def get_vpn_status(self) -> VPNStatus:
-        status = VPNStatus()
-
-        values = [
-            self.request("/admin/openvpn?form=config&operation=read", "operation=read"),
-            self.request("/admin/pptpd?form=config&operation=read", "operation=read"),
-            self.request("/admin/vpnconn?form=config&operation=list&vpntype=openvpn",
-                         "operation=list&operation=list&vpntype=openvpn"),
-            self.request("/admin/vpnconn?form=config&operation=list&vpntype=pptp",
-                         "operation=list&operation=list&vpntype=pptp"),
-        ]
-
-        status.openvpn_enable = values[0]['enabled'] == 'on'
-        status.pptpvpn_enable = values[1]['enabled'] == 'on'
-
-        if isinstance(values[2], list):
-            status.openvpn_clients_total = len(values[2])
-            status.pptpvpn_clients_total = len(values[3])
-        else:
-            status.openvpn_clients_total = 0
-            status.pptpvpn_clients_total = 0
-
-        return status
-
-    def set_vpn(self, vpn: VPN, enable: bool) -> None:
-        path = "/admin/{}?form=config&operation=read".format(vpn.lowercase)
-        current_config = self.request(path, "operation=read")
-        current_config['enabled'] = "on" if enable else "off"
-        data = urlencode(current_config)
-        data = "&operation=write&{}".format(data)
-        path = "/admin/{}?form=config{}".format(vpn.lowercase, data)
-        self.request(path, data)

tplinkrouterc6u/client/c6u.py

@@ -1,14 +1,23 @@
 from hashlib import md5
 from re import search
 from json import loads
+from urllib.parse import urlencode
 from requests import post, Response
 from macaddress import EUI48
 from ipaddress import IPv4Address
 from logging import Logger
 from tplinkrouterc6u.common.helper import get_ip, get_mac
 from tplinkrouterc6u.common.encryption import EncryptionWrapper
-from tplinkrouterc6u.common.package_enum import Connection
-from tplinkrouterc6u.common.dataclass import Firmware, Status, Device, IPv4Reservation, IPv4DHCPLease, IPv4Status
+from tplinkrouterc6u.common.package_enum import Connection, VPN
+from tplinkrouterc6u.common.dataclass import (
+    Firmware,
+    Status,
+    Device,
+    IPv4Reservation,
+    IPv4DHCPLease,
+    IPv4Status,
+    VPNStatus,
+)
 from tplinkrouterc6u.common.exception import ClientException, ClientError
 from tplinkrouterc6u.client_abstract import AbstractRouter
 from abc import abstractmethod
@@ -231,6 +240,10 @@ class TplinkBaseRouter(AbstractRouter, TplinkRequest):
         self._url_firmware = 'admin/firmware?form=upgrade&operation=read'
         self._url_ipv4_reservations = 'admin/dhcps?form=reservation&operation=load'
         self._url_ipv4_dhcp_leases = 'admin/dhcps?form=client&operation=load'
+        self._url_openvpn = 'admin/openvpn?form=config&operation=read'
+        self._url_pptpd = 'admin/pptpd?form=config&operation=read'
+        self._url_vpnconn_openvpn = 'admin/vpnconn?form=config&operation=list&vpntype=openvpn'
+        self._url_vpnconn_pptpd = 'admin/vpnconn?form=config&operation=list&vpntype=pptp'
         referer = '{}/webpages/index.html'.format(self.host)
         self._headers_request = {'Referer': referer}
         self._headers_login = {'Referer': referer, 'Content-Type': 'application/x-www-form-urlencoded'}
@@ -401,6 +414,36 @@ class TplinkBaseRouter(AbstractRouter, TplinkRequest):
 
         return dhcp_leases
 
+    def get_vpn_status(self) -> VPNStatus:
+        status = VPNStatus()
+
+        values = [
+            self.request(self._url_openvpn, "operation=read"),
+            self.request(self._url_pptpd, "operation=read"),
+            self.request(self._url_vpnconn_openvpn, "operation=list&vpntype=openvpn"),
+            self.request(self._url_vpnconn_pptpd, "operation=list&vpntype=pptp"),
+        ]
+
+        status.openvpn_enable = values[0]['enabled'] == 'on'
+        status.pptpvpn_enable = values[1]['enabled'] == 'on'
+
+        if isinstance(values[2], list):
+            status.openvpn_clients_total = len(values[2])
+            status.pptpvpn_clients_total = len(values[3])
+        else:
+            status.openvpn_clients_total = 0
+            status.pptpvpn_clients_total = 0
+
+        return status
+
+    def set_vpn(self, vpn: VPN, enable: bool) -> None:
+        path = self._url_openvpn if VPN.OPEN_VPN == vpn else self._url_pptpd
+        current_config = self.request(path, "operation=read")
+        current_config['enabled'] = "on" if enable else "off"
+        data = urlencode(current_config)
+        data = "operation=write&{}".format(data)
+        self.request(path, data)
+
     @staticmethod
     def _str2bool(v) -> bool | None:
         return str(v).lower() in ("yes", "true", "on") if v is not None else None
@@ -435,3 +478,7 @@ class TplinkRouter(TplinkEncryption, TplinkBaseRouter):
         self._url_firmware = 'admin/firmware?form=upgrade'
         self._url_ipv4_reservations = 'admin/dhcps?form=reservation'
         self._url_ipv4_dhcp_leases = 'admin/dhcps?form=client'
+        self._url_openvpn = 'admin/openvpn?form=config'
+        self._url_pptpd = 'admin/pptpd?form=config'
+        self._url_vpnconn_openvpn = 'admin/vpnconn?form=config'
+        self._url_vpnconn_pptpd = 'admin/vpnconn?form=config'

{tplinkrouterc6u-5.8.0.dist-info → tplinkrouterc6u-5.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tplinkrouterc6u
-Version: 5.8.0
+Version: 5.9.0
 Summary: TP-Link Router API (supports also Mercusys Router)
 Home-page: https://github.com/AlexandrErohin/TP-Link-Archer-C6U
 Author: Alex Erohin
@@ -59,6 +59,7 @@ from tplinkrouterc6u import (
     TPLinkXDRClient,
     TPLinkDecoClient,
     TplinkC80Router,
+    TplinkWDRRouter,
     Connection
 )
 from logging import Logger
@@ -312,6 +313,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - Archer AX3000 V1
 - Archer AX6000 V1
 - Archer AX11000 V1
+- Archer BE400 v1.0
 - Archer BE550 v1.0
 - Archer BE800 v1.0
 - Archer BE805 v1.0
@@ -334,6 +336,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - Archer VR1200v v1
 - Archer VR2100v v1
 - Archer VX1800v v1.0
+- BE11000 2.0
 - Deco M4 2.0
 - Deco M4R 2.0
 - Deco M5 v3
@@ -359,6 +362,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
 - TL-WA3001 v1.0
 - TL-XDR3010 V2
 - TL-WDR3600 V1
+- VX420-G2h v1.1
 - XC220-G3v v2.30
 ### <a id="mercusys">MERCUSYS routers</a>
 - MR47BE v1.0

{tplinkrouterc6u-5.8.0.dist-info → tplinkrouterc6u-5.9.0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 test/__init__.py,sha256=McQmUjeN3AwmwdS6QNfwGXXE77OKoPK852I2BM9XsrU,210
-test/test_client_c1200.py,sha256=4RdwMEwTGWLOxOVTuPikifXDcUJFQ13DuyWGm2Aevrk,8811
-test/test_client_c6u.py,sha256=ulKoOJiwQVaBslY7kxCa7HmLaqTPzKJaNjm-xIBjB6A,33755
+test/test_client_c1200.py,sha256=Sl-85JGqINNg-ckBZCIVqY0CC-V1UOc-yiIUljtePRM,7582
+test/test_client_c6u.py,sha256=VpxF5etPtGYuXX9TfUl78tOj0pjozvPkrVx6FKbZDjQ,37585
 test/test_client_c80.py,sha256=RY_1SgRVcQQdN9h0_IXA0YW4_0flEB_uel05QvDDfws,42359
 test/test_client_deco.py,sha256=YPLKRD8GoyDYHfRgdXvCk8iVNw8zdMJW-AHVnNbpdTM,31719
 test/test_client_ex.py,sha256=0dVvOJqxpR2xNca99vadfdpxmzW-4sk9X_4tAK8x9-c,21710
@@ -11,9 +11,9 @@ tplinkrouterc6u/__init__.py,sha256=DDy6XVFH8Ne5C9JToRWNzRqevUfNutLUTlDhvIMsebo,1
 tplinkrouterc6u/client_abstract.py,sha256=3UYzmll774S_Gb5E0FTVO_rI3-XFM7PSklg1-V-2jls,1419
 tplinkrouterc6u/provider.py,sha256=bzH0WW2peC66f8NqfiI3t_niX6MFO8SAdKP9WZL-IV4,2152
 tplinkrouterc6u/client/__init__.py,sha256=KBy3fmtA9wgyFrb0Urh2x4CkKtWVnESdp-vxmuOvq0k,27
-tplinkrouterc6u/client/c1200.py,sha256=_nY_pJ-wPWODAaes9kHPdVcM6YM54f1E54CfdoFHqbE,4771
+tplinkrouterc6u/client/c1200.py,sha256=4XEYidEGmVIJk0YQLvmTnd0Gqa7glH2gUWvjreHpWrk,3178
 tplinkrouterc6u/client/c5400x.py,sha256=9E0omBSbWY_ljrs5MTCMu5brmrLtzsDB5O62Db8lP8Q,4329
-tplinkrouterc6u/client/c6u.py,sha256=zcnMFsP-rn219PLfav-7KtHhOFkfrdtYnCyHWdcdKOw,17661
+tplinkrouterc6u/client/c6u.py,sha256=3Jon6_WZkaCyrbTAp9TckB-SIXLP_JyB5gtRI65IuVI,19485
 tplinkrouterc6u/client/c80.py,sha256=ArVhza_fnXcEO-_fsQOd1l2QvmSfsswtohKxrZxEnoU,18568
 tplinkrouterc6u/client/deco.py,sha256=cpKRggKD2RvSmMZuD6tzsZmehAUCU9oLiTTHcZBW81Y,8898
 tplinkrouterc6u/client/ex.py,sha256=ZK08mlOts3fbBJh9NUrzBADP0N9LHgzOXCKY2tF9gG0,13111
@@ -27,8 +27,8 @@ tplinkrouterc6u/common/encryption.py,sha256=4HelTxzN6esMfDZRBt3m8bwB9Nj_biKijnCn
 tplinkrouterc6u/common/exception.py,sha256=_0G8ZvW5__CsGifHrsZeULdl8c6EUD071sDCQsQgrHY,140
 tplinkrouterc6u/common/helper.py,sha256=23b04fk9HuVinrZXMCS5R1rmF8uZ7eM-Cdnp7Br9NR0,572
 tplinkrouterc6u/common/package_enum.py,sha256=4ykL_2Pw0nDEIH_qR9UJlFF6stTgSfhPz32r8KT-sh8,1624
-tplinkrouterc6u-5.8.0.dist-info/licenses/LICENSE,sha256=YF6QR6Vjxcg5b_sYIyqkME7FZYau5TfEUGTG-0JeRK0,35129
-tplinkrouterc6u-5.8.0.dist-info/METADATA,sha256=XceOtjPvarwp7C7OvSVjvmFgLTU-nYDwu5K_DSMjbnc,15496
-tplinkrouterc6u-5.8.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-tplinkrouterc6u-5.8.0.dist-info/top_level.txt,sha256=1iSCCIueqgEkrTxtQ-jiHe99jAB10zqrVdBcwvNfe_M,21
-tplinkrouterc6u-5.8.0.dist-info/RECORD,,
+tplinkrouterc6u-5.9.0.dist-info/licenses/LICENSE,sha256=YF6QR6Vjxcg5b_sYIyqkME7FZYau5TfEUGTG-0JeRK0,35129
+tplinkrouterc6u-5.9.0.dist-info/METADATA,sha256=PCYcU6cFG5lUp73dLZ-0UscRtpXxtWPan6B-J1tudlk,15568
+tplinkrouterc6u-5.9.0.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
+tplinkrouterc6u-5.9.0.dist-info/top_level.txt,sha256=1iSCCIueqgEkrTxtQ-jiHe99jAB10zqrVdBcwvNfe_M,21
+tplinkrouterc6u-5.9.0.dist-info/RECORD,,

{tplinkrouterc6u-5.8.0.dist-info → tplinkrouterc6u-5.9.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.7.1)
 Root-Is-Purelib: true
 Tag: py3-none-any