tplinkrouterc6u 5.10.3__tar.gz → 5.11.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (43) hide show
  1. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/PKG-INFO +2 -1
  2. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/README.md +1 -0
  3. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/setup.py +1 -1
  4. tplinkrouterc6u-5.11.0/test/test_client_re330.py +296 -0
  5. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/__init__.py +1 -0
  6. tplinkrouterc6u-5.11.0/tplinkrouterc6u/client/mr200.py +412 -0
  7. tplinkrouterc6u-5.11.0/tplinkrouterc6u/client/re330.py +423 -0
  8. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/provider.py +2 -1
  9. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u.egg-info/PKG-INFO +2 -1
  10. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u.egg-info/SOURCES.txt +2 -0
  11. tplinkrouterc6u-5.10.3/tplinkrouterc6u/client/mr200.py +0 -57
  12. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/LICENSE +0 -0
  13. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/setup.cfg +0 -0
  14. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/test/__init__.py +0 -0
  15. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/test/test_client_c1200.py +0 -0
  16. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/test/test_client_c6u.py +0 -0
  17. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/test/test_client_c80.py +0 -0
  18. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/test/test_client_deco.py +0 -0
  19. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/test/test_client_ex.py +0 -0
  20. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/test/test_client_mr.py +0 -0
  21. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/test/test_client_wdr.py +0 -0
  22. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/test/test_client_xdr.py +0 -0
  23. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/__init__.py +0 -0
  24. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/c1200.py +0 -0
  25. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/c5400x.py +0 -0
  26. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/c6u.py +0 -0
  27. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/c80.py +0 -0
  28. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/deco.py +0 -0
  29. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/ex.py +0 -0
  30. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/mr.py +0 -0
  31. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/vr.py +0 -0
  32. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/wdr.py +0 -0
  33. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client/xdr.py +0 -0
  34. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/client_abstract.py +0 -0
  35. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/common/__init__.py +0 -0
  36. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/common/dataclass.py +0 -0
  37. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/common/encryption.py +0 -0
  38. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/common/exception.py +0 -0
  39. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/common/helper.py +0 -0
  40. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u/common/package_enum.py +0 -0
  41. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u.egg-info/dependency_links.txt +0 -0
  42. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u.egg-info/requires.txt +0 -0
  43. {tplinkrouterc6u-5.10.3 → tplinkrouterc6u-5.11.0}/tplinkrouterc6u.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: tplinkrouterc6u
3
- Version: 5.10.3
3
+ Version: 5.11.0
4
4
  Summary: TP-Link Router API (supports also Mercusys Router)
5
5
  Home-page: https://github.com/AlexandrErohin/TP-Link-Archer-C6U
6
6
  Author: Alex Erohin
@@ -382,6 +382,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
382
382
  - VX420-G2h v1.1
383
383
  - VX800v v1
384
384
  - XC220-G3v v2.30
385
+ - RE330 v1
385
386
  ### <a id="mercusys">MERCUSYS routers</a>
386
387
  - MR47BE v1.0
387
388
  - MR50G 1.0
@@ -350,6 +350,7 @@ or you have TP-link C5400X or similar router you need to get web encrypted passw
350
350
  - VX420-G2h v1.1
351
351
  - VX800v v1
352
352
  - XC220-G3v v2.30
353
+ - RE330 v1
353
354
  ### <a id="mercusys">MERCUSYS routers</a>
354
355
  - MR47BE v1.0
355
356
  - MR50G 1.0
@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
5
5
 
6
6
  setuptools.setup(
7
7
  name="tplinkrouterc6u",
8
- version="5.10.3",
8
+ version="5.11.0",
9
9
  author="Alex Erohin",
10
10
  author_email="alexanderErohin@yandex.ru",
11
11
  description="TP-Link Router API (supports also Mercusys Router)",
@@ -0,0 +1,296 @@
1
+ from unittest import main, TestCase
2
+ from ipaddress import IPv4Address
3
+ from macaddress import EUI48
4
+ from tplinkrouterc6u.common.dataclass import Firmware, Status, Device
5
+ from tplinkrouterc6u.common.dataclass import IPv4Status, IPv4Reservation, IPv4DHCPLease
6
+ from tplinkrouterc6u import Connection, ClientException
7
+ from tplinkrouterc6u.client.re330 import TplinkRE330Router
8
+
9
+
10
+ IPV4_STATUS_RESPONSE = ('00000\r\nid 1|1,0,0\r\noldAuthKey \r\nsetWzd 1\r\nmode 3\r\nlogLevel 3\r\nfastpath 1\r\n'
11
+ 'mac 0 00-00-00-00-00-01\r\nmac 1 00-00-00-00-00-02\r\nauthKey keykeykey\r\nid 4|1,0,0\r\n'
12
+ 'ip 2.2.2.2\r\nmask 255.255.255.0\r\ngateway 3.3.3.3\r\ndns 0 1.1.1.1\r\n'
13
+ 'dns 1 8.8.8.8\r\nmode 0\r\nid 8|1,0,0\r\nmode 2\r\npoolStart 192.168.1.100\r\n'
14
+ 'poolEnd 192.168.1.199\r\n'
15
+ 'leaseTime 120\r\ndns 0 0.0.0.0\r\ndns 1 0.0.0.0\r\ngateway 0.0.0.0\r\nhostName \r\n'
16
+ 'id 22|1,0,0\r\nenable 1\r\nwirelessWanNoUsed 0\r\nlinkMode 0\r\nlinkType 0\r\nid 23|1,0,0\r\n'
17
+ 'ip 4.4.4.4\r\nmask 255.255.252.0\r\ngateway 5.5.5.5\r\ndns 0 1.1.1.1\r\ndns 1 8.8.8.8\r\n'
18
+ 'status 1\r\ncode 0\r\nupTime 0\r\ninPkts 0\r\ninOctets 0\r\noutPkts 0\r\noutOctets 0\r\n'
19
+ 'inRates 0\r\noutRates 0\r\ndualMode 0\r\ndualIp 0.0.0.0\r\ndualMask 0.0.0.0\r\n'
20
+ 'dualGateway 0.0.0.0\r\ndualDns 0 0.0.0.0\r\ndualDns 1 0.0.0.0\r\ndualCode 0\r\ndualStatus 0')
21
+
22
+ STATUS_RESPONSE_TEXT = ('00000\r\nid 1|1,0,0\r\noldAuthKey \r\nsetWzd 1\r\nmode 3\r\nlogLevel 3\r\nfastpath 1\r\n'
23
+ 'mac 0 00-00-00-00-00-01\r\nmac 1 00-00-00-00-00-02\r\nauthKey keykeykey\r\nid 4|1,0,0\r\n'
24
+ 'ip 2.2.2.2\r\nmask 255.255.255.0\r\ngateway 3.3.3.3\r\ndns 0 1.1.1.1\r\ndns 1 8.8.8.8\r\n'
25
+ 'mode 0\r\nid 23|1,0,0\r\nip 4.4.4.4\r\nmask 255.255.255.0\r\ngateway 5.5.5.5\r\n'
26
+ 'dns 0 1.1.1.1\r\ndns 1 8.8.8.8\r\nstatus 1\r\ncode 0\r\nupTime 0\r\ninPkts 0\r\ninOctets 0\r\n'
27
+ 'outPkts 0\r\noutOctets 0\r\ninRates 0\r\noutRates 0\r\ndualMode 0\r\ndualIp 0.0.0.0\r\n'
28
+ 'dualMask 0.0.0.0\r\ndualGateway 0.0.0.0\r\ndualDns 0 0.0.0.0\r\ndualDns 1 0.0.0.0\r\n'
29
+ 'dualCode 0\r\ndualStatus 0\r\nid 13|1,0,0\r\nip 0 10.10.10.10\r\nip 1 11.11.11.11\r\n'
30
+ 'ip 2 0.0.0.0\r\nip 3 0.0.0.0\r\nip 4 0.0.0.0\r\nip 5 0.0.0.0\r\nmac 0 00-00-00-00-00-03\r\n'
31
+ 'mac 1 00-00-00-00-00-04\r\nmac 2 00-00-00-00-00-00\r\nmac 3 00-00-00-00-00-00\r\n'
32
+ 'mac 4 00-00-00-00-00-00\r\nmac 5 00-00-00-00-00-00\r\n'
33
+ 'reserved 0 \r\nreserved 1 \r\nreserved 2 \r\nreserved 3 \r\nreserved 4 \r\nreserved 5 \r\n'
34
+ 'bindEntry 0 0\r\nbindEntry 1 0\r\nbindEntry 2 0\r\nbindEntry 3 0\r\nbindEntry 4 0\r\n'
35
+ 'bindEntry 5 0\r\nstaMgtEntry 0 0\r\nstaMgtEntry 1 0\r\nstaMgtEntry 2 0\r\nstaMgtEntry 3 0\r\n'
36
+ 'staMgtEntry 4 0\r\nstaMgtEntry 5 0\r\ntype 0 3\r\ntype 1 1\r\ntype 2 0\r\ntype 3 0\r\n'
37
+ 'type 4 0\r\ntype 5 0\r\nonline 0 1\r\nonline 1 1\r\nonline 2 0\r\nonline 3 0\r\nonline 4 0\r\n'
38
+ 'online 5 0\r\nname 0 ANONYMOUS\r\nname 1 BANANA-12\r\nname 2 \r\nname 3 \r\nname 4 \r\n'
39
+ 'name 5 \r\nDevType 0 OTHER\r\nDevType 1 OTHER\r\nDevType 2 \r\nDevType 3 \r\nDevType 4 \r\n'
40
+ 'DevType 5 \r\nid 33|1,1,0\r\nuUnit 0\r\ncSsidPrefix \r\nuRadiusIp 0.0.0.0\r\n'
41
+ 'uRadiusGKUpdateIntvl 0\r\nuPskGKUpdateIntvl 0\r\nuKeyLength 0 0\r\nuKeyLength 1 0\r\n'
42
+ 'uKeyLength 2 0\r\nuKeyLength 3 0\r\ncKeyVal 0 \r\ncKeyVal 1 \r\ncKeyVal 2 \r\ncKeyVal 3 \r\n'
43
+ 'uRadiusPort 1812\r\nuKeyType 1\r\nuDefaultKey 1\r\nbEnable 1\r\nbBcastSsid 1\r\n'
44
+ 'cSsid SuperWifi\r\nbSecurityEnable 1\r\nuAuthType 3\r\nuWEPSecOpt 3\r\nuRadiusSecOpt 3\r\n'
45
+ 'uPSKSecOpt 2\r\nuRadiusEncryptType 4\r\nuPSKEncryptType 3\r\ncRadiusSecret \r\n'
46
+ 'cPskSecret YouThoughtIdForgetMyKey?\r\nbSecCheck 0\r\nbEnabled 1\r\nbPinEnabled 0\r\n'
47
+ 'cUsrPIN 11100111\r\nbConfigured 0\r\nbIsLocked 0\r\nid 33|2,1,0\r\nuUnit 0\r\ncSsidPrefix \r\n'
48
+ 'uRadiusIp 0.0.0.0\r\nuRadiusGKUpdateIntvl 0\r\nuPskGKUpdateIntvl 0\r\nuKeyLength 0 0\r\n'
49
+ 'uKeyLength 1 0\r\nuKeyLength 2 0\r\nuKeyLength 3 0\r\ncKeyVal 0 \r\ncKeyVal 1 \r\n'
50
+ 'cKeyVal 2 \r\ncKeyVal 3 \r\nuRadiusPort 1812\r\nuKeyType 1\r\nuDefaultKey 1\r\nbEnable 1\r\n'
51
+ 'bBcastSsid 1\r\ncSsid SuperWifi\r\nbSecurityEnable 1\r\nuAuthType 3\r\nuWEPSecOpt 3\r\n'
52
+ 'uRadiusSecOpt 3\r\nuPSKSecOpt 2\r\nuRadiusEncryptType 4\r\nuPSKEncryptType 3\r\n'
53
+ 'cRadiusSecret \r\ncPskSecret YouThoughtIdForgetMyKey?\r\nbSecCheck 0\r\n'
54
+ 'bEnabled 1\r\nbPinEnabled 0\r\ncUsrPIN 11100111\r\nbConfigured 0\r\nbIsLocked 0')
55
+
56
+
57
+ class ResponseMock():
58
+ def __init__(self, text, status_code=0):
59
+ self.text = text
60
+ self.status_code = status_code
61
+
62
+
63
+ class TplinkRE330RouterTest(TplinkRE330Router):
64
+ response = ''
65
+
66
+ def _init_session(self) -> None:
67
+ pass
68
+
69
+ def request(self, code: int, asyn: int, use_token: bool = False, data: str = None) -> dict | None:
70
+
71
+ # Responses
72
+ if code == 2 and (asyn == 0 or asyn == 1):
73
+ if use_token is False:
74
+ if data == '50|1,0,0':
75
+ # Supports
76
+ return ResponseMock(self.response, 200)
77
+ else:
78
+ # Authorization
79
+ return ResponseMock('blabla\r\nblabla\r\nblabla\r\nauthinfo1\r\nauthinfo2')
80
+ elif use_token is True:
81
+ return ResponseMock(self.response)
82
+ if code == 7 and asyn == 1:
83
+ if use_token is False:
84
+ # Authorization
85
+ return ResponseMock('00007\r\n00004\r\n00002\r\n'
86
+ 'BC97577E65233B3E1137C61091D64176C334E52AD78FFBDDABC826B685435E'
87
+ '9D3DE83FE70C2AC62D6B13BD8EADA10B5623F9354DA0E99636A4F5519CA2DC2DC3\r\n'
88
+ '12345656\r\n00000')
89
+ elif use_token is True:
90
+ return ResponseMock('00000')
91
+ elif code == 16 and asyn == 0:
92
+ if use_token is False:
93
+ # Authorization
94
+ return ResponseMock('00000\r\n010001\r\nBC97577E65233B3E1137C61091D64176C334E52AD78FFBDDABC826B685435E'
95
+ '9D3DE83FE70C2AC62D6B13BD8EADA10B5623F9354DA0E99636A4F5519CA2DC2DC3\r\n12345656')
96
+ elif use_token is True:
97
+ # Authorization
98
+ return ResponseMock('00000')
99
+ elif code == 7 and asyn == 0:
100
+ return ResponseMock('00000')
101
+
102
+ raise ClientException()
103
+
104
+ def set_encrypted_response(self, response_text) -> None:
105
+ self.response = self._encrypt_body(response_text).split('data=')[1]
106
+
107
+
108
+ class TestTPLinkClient(TestCase):
109
+
110
+ def test_supports(self) -> None:
111
+ response = ('00000\r\nid 50|1,0,0\r\ncurrentLanguage\r\n'
112
+ 'languageList bg_BG,cs_CZ,de_DE,en_US,es_ES,es_LA,fr_FR,hu_HU,it_IT,ja_JP,ko_KR,nl_NL,pl_PL,pt_BR,'
113
+ 'pt_PT,ro_RO,ru_RU,sk_SK,tr_TR,uk_UA,vi_VN,zh_TW\r\nsetByUser 0')
114
+
115
+ client = TplinkRE330RouterTest('', '')
116
+ client.response = response
117
+ supports = client.supports()
118
+ self.assertTrue(supports)
119
+
120
+ def test_authorize(self) -> None:
121
+ client = TplinkRE330RouterTest('', '')
122
+ client.authorize()
123
+
124
+ encryption = client._encryption
125
+ self.assertEqual(encryption.ee_rsa, '010001')
126
+ self.assertEqual(encryption.nn_rsa, 'BC97577E65233B3E1137C61091D64176C334E52AD78FFBDDABC826B685435E9D3DE83FE70C'
127
+ '2AC62D6B13BD8EADA10B5623F9354DA0E99636A4F5519CA2DC2DC3')
128
+ self.assertEqual(encryption.seq, '12345656')
129
+
130
+ def test_get_firmware(self) -> None:
131
+ response = ('00000\r\nid 0|1,0,0\r\nfullName TP-Link%20Wireless%20Extender%20RE330\r\nfacturer TP-Link\r\n'
132
+ 'modelName RE330\r\nmodelVer 1\r\nsoftVer 1.0.23%20Build%20230418%20Rel.60395n\r\n'
133
+ 'hardVer %20RE330%201.0\r\nprodId 0x3300001\r\ncloudShouldActive 1\r\ncountryId 0x0\r\n'
134
+ 'specialId 0x5545\r\ncountryCode 0x4544\r\nmainVer 0x5a010017\r\nminorVer 0x1\r\noemId 0x1\r\n'
135
+ 'deviceId 8002A1F018FA0C879DB62FA981FB0D1D231D490F\r\n'
136
+ 'hardwareId 5E055ADC85F0800C6C3044E5A3180E2A\r\nfirmwareId FFFFFFFFFFFFFFFFFFFF033001004555\r\n'
137
+ 'oem_id B30DDAF6C31C08B9C50A48B0B9168003\r\nfacturerType 0')
138
+
139
+ client = TplinkRE330RouterTest('', '')
140
+ client.authorize()
141
+
142
+ client.set_encrypted_response(response)
143
+
144
+ firmware = client.get_firmware()
145
+
146
+ self.assertIsInstance(firmware, Firmware)
147
+ self.assertEqual(firmware.hardware_version, ' RE330 1.0')
148
+ self.assertEqual(firmware.model, 'RE330')
149
+ self.assertEqual(firmware.firmware_version, '1.0.23 Build 230418 Rel.60395n')
150
+
151
+ def test_get_ipv4_status(self) -> None:
152
+
153
+ client = TplinkRE330RouterTest('', '')
154
+ client.authorize()
155
+
156
+ client.set_encrypted_response(IPV4_STATUS_RESPONSE)
157
+
158
+ ipv4_status: IPv4Status = client.get_ipv4_status()
159
+
160
+ self.assertIsInstance(ipv4_status, IPv4Status)
161
+ self.assertEqual(ipv4_status.wan_macaddress, EUI48('00-00-00-00-00-02'))
162
+ self.assertEqual(ipv4_status._wan_ipv4_ipaddr, IPv4Address('4.4.4.4'))
163
+ self.assertEqual(ipv4_status._wan_ipv4_gateway, IPv4Address('5.5.5.5'))
164
+ self.assertEqual(ipv4_status._wan_ipv4_conntype, 'Dynamic IP')
165
+ self.assertEqual(ipv4_status._wan_ipv4_netmask, IPv4Address('255.255.252.0'))
166
+ self.assertEqual(ipv4_status._wan_ipv4_pridns, IPv4Address('1.1.1.1'))
167
+ self.assertEqual(ipv4_status._wan_ipv4_snddns, IPv4Address('8.8.8.8'))
168
+ self.assertEqual(ipv4_status._lan_macaddr, EUI48('00-00-00-00-00-01'))
169
+ self.assertEqual(ipv4_status._lan_ipv4_ipaddr, IPv4Address('2.2.2.2'))
170
+ self.assertEqual(ipv4_status.lan_ipv4_dhcp_enable, False)
171
+ self.assertEqual(ipv4_status._lan_ipv4_netmask, IPv4Address('255.255.255.0'))
172
+
173
+ def test_get_ipv4_reservations(self) -> None:
174
+ response = ('00000\r\nid 12|1,0,0\r\nip 0 192.168.0.112\r\nip 1 0.0.0.0\r\nmac 0 00-00-00-00-00-00\r\n'
175
+ 'mac 1 00-00-00-00-00-01\r\nreserved 0\r\nreserved 1\r\nbindEntry 0 0\r\nbindEntry 1 0\r\n'
176
+ 'staMgtEntry 0 0\r\nstaMgtEntry 1 1\r\nname 0 Galaxy-S21\r\nname 1 Camera\r\nreserved_name 0\r\n'
177
+ 'reserved_name 1\r\nblocked 0 0\r\nblocked 1 0\r\nupLimit 0 0\r\nupLimit 1 0\r\ndownLimit 0 0'
178
+ '\r\ndownLimit 1 0\r\nqosEntry 0 0\r\nqosEntry 1 0\r\npriTime 0 0\r\npriTime 1 0\r\n'
179
+ 'dhcpsEntry 0 1\r\ndhcpsEntry 1 0\r\ndhcpsEnable 0 1\r\ndhcpsEnable 1 0\r\nslEnable 0 0\r\n'
180
+ 'slEnable 1 0\r\nstart 0 0\r\nstart 1 0\r\nend 0 0\r\nend 1 0\r\nday 0 0\r\nday 1 0\r\n'
181
+ 'startMin 0 0\r\nstartMin 1 0\r\nendMin 0 0\r\nendMin 1 0\r\ndevType 0 0\r\ndevType 1 0\r\n'
182
+ 'reserved2 0 0\r\nreserved2 1 0\r\ndisable 1')
183
+
184
+ client = TplinkRE330RouterTest('', '')
185
+ client.authorize()
186
+
187
+ client.set_encrypted_response(response)
188
+
189
+ ipv4_reservations: list[IPv4Reservation] = client.get_ipv4_reservations()
190
+ ipv4_reservation: IPv4Reservation = ipv4_reservations[0]
191
+
192
+ self.assertIsInstance(ipv4_reservation, IPv4Reservation)
193
+ self.assertEqual(ipv4_reservation.macaddress, EUI48('00-00-00-00-00-00'))
194
+ self.assertEqual(ipv4_reservation.ipaddress, IPv4Address('192.168.0.112'))
195
+ self.assertEqual(ipv4_reservation.hostname, 'Galaxy-S21')
196
+ self.assertEqual(ipv4_reservation.enabled, True)
197
+
198
+ def test_get_dhcp_leases(self) -> None:
199
+ response = ('00000\r\nid 9|1,0,0\r\nhostName 0 Galaxy-S21\r\nhostName 1 iPhone\r\nhostName 2 PC\r\n'
200
+ 'hostName 3 Laptop\r\nmac 0 00-00-00-00-00-00\r\nmac 1 00-00-00-00-00-01\r\n'
201
+ 'mac 2 00-00-00-00-00-02\r\nmac 3 00-00-00-00-00-03\r\nreserved 0\r\nreserved 1'
202
+ '\r\nreserved 2\r\nreserved 3\r\nstate 0 5\r\nstate 1 5\r\nstate 2 5\r\nstate 3 5'
203
+ '\r\nip 0 192.168.0.112\r\nip 1 192.168.0.101\r\nip 2 192.168.0.245\r\nip 3 192.168.0.186'
204
+ '\r\nexpires 0 4294967295\r\nexpires 1 3669\r\nexpires 2 4025\r\nexpires 3 4202')
205
+
206
+ client = TplinkRE330RouterTest('', '')
207
+ client.authorize()
208
+
209
+ client.set_encrypted_response(response)
210
+
211
+ dhcp_leases: list[IPv4DHCPLease] = client.get_dhcp_leases()
212
+
213
+ self.assertIsInstance(dhcp_leases[0], IPv4DHCPLease)
214
+ self.assertEqual(dhcp_leases[0].macaddress, EUI48('00-00-00-00-00-00'))
215
+ self.assertEqual(dhcp_leases[0].ipaddress, IPv4Address('192.168.0.112'))
216
+ self.assertEqual(dhcp_leases[0].hostname, 'Galaxy-S21')
217
+ self.assertEqual(dhcp_leases[0].lease_time, 'expires 4294967295')
218
+
219
+ self.assertIsInstance(dhcp_leases[1], IPv4DHCPLease)
220
+ self.assertEqual(dhcp_leases[1].macaddress, EUI48('00-00-00-00-00-01'))
221
+ self.assertEqual(dhcp_leases[1].ipaddress, IPv4Address('192.168.0.101'))
222
+ self.assertEqual(dhcp_leases[1].hostname, 'iPhone')
223
+ self.assertEqual(dhcp_leases[1].lease_time, 'expires 3669')
224
+
225
+ def test_get_status(self) -> None:
226
+ client = TplinkRE330RouterTest('', '')
227
+ client.authorize()
228
+
229
+ client.set_encrypted_response(STATUS_RESPONSE_TEXT)
230
+ status = client.get_status()
231
+
232
+ self.assertIsInstance(status, Status)
233
+ self.assertEqual(status.wan_macaddr, '00-00-00-00-00-02')
234
+ self.assertIsInstance(status.wan_macaddress, EUI48)
235
+ self.assertEqual(status.lan_macaddr, '00-00-00-00-00-01')
236
+ self.assertIsInstance(status.lan_macaddress, EUI48)
237
+ self.assertEqual(status.wan_ipv4_addr, '4.4.4.4')
238
+ self.assertIsInstance(status.lan_ipv4_address, IPv4Address)
239
+ self.assertEqual(status.lan_ipv4_addr, '2.2.2.2')
240
+ self.assertEqual(status.wan_ipv4_gateway, '5.5.5.5')
241
+ self.assertIsInstance(status.wan_ipv4_address, IPv4Address)
242
+ self.assertEqual(status.wired_total, 0)
243
+ self.assertEqual(status.wifi_clients_total, 2)
244
+ self.assertEqual(status.guest_clients_total, 0)
245
+ self.assertEqual(status.clients_total, 2)
246
+ self.assertEqual(status.iot_clients_total, 0)
247
+ self.assertFalse(status.guest_2g_enable)
248
+ self.assertFalse(status.guest_5g_enable)
249
+ self.assertFalse(status.iot_2g_enable)
250
+ self.assertFalse(status.iot_5g_enable)
251
+ self.assertTrue(status.wifi_2g_enable)
252
+ self.assertTrue(status.wifi_5g_enable)
253
+ self.assertEqual(status.wan_ipv4_uptime, 0)
254
+ self.assertEqual(status.mem_usage, None)
255
+ self.assertEqual(status.cpu_usage, None)
256
+ self.assertEqual(len(status.devices), 2)
257
+
258
+ device = status.devices[0]
259
+ self.assertIsInstance(device, Device)
260
+ self.assertEqual(device.type, Connection.HOST_5G)
261
+ self.assertEqual(device.macaddr, '00-00-00-00-00-03')
262
+ self.assertIsInstance(device.macaddress, EUI48)
263
+ self.assertEqual(device.ipaddr, '10.10.10.10')
264
+ self.assertIsInstance(device.ipaddress, IPv4Address)
265
+ self.assertEqual(device.hostname, 'ANONYMOUS')
266
+ self.assertEqual(device.up_speed, 0)
267
+ self.assertEqual(device.down_speed, 0)
268
+ self.assertEqual(device.active, True)
269
+
270
+ device = status.devices[1]
271
+ self.assertIsInstance(device, Device)
272
+ self.assertEqual(device.type, Connection.HOST_2G)
273
+ self.assertEqual(device.macaddr, '00-00-00-00-00-04')
274
+ self.assertIsInstance(device.macaddress, EUI48)
275
+ self.assertEqual(device.ipaddr, '11.11.11.11')
276
+ self.assertIsInstance(device.ipaddress, IPv4Address)
277
+ self.assertEqual(device.hostname, 'BANANA-12')
278
+ self.assertEqual(device.up_speed, 0)
279
+ self.assertEqual(device.down_speed, 0)
280
+ self.assertEqual(device.active, True)
281
+
282
+ def test_get_led_status(self) -> None:
283
+ client = TplinkRE330RouterTest('', '')
284
+ client.authorize()
285
+
286
+ client.set_encrypted_response('00000\r\nid 112|1,0,0\r\nenable 1')
287
+ led_status = client.get_led_status()
288
+ self.assertEqual(led_status, True)
289
+
290
+ client.set_encrypted_response('00000\r\nid 112|1,0,0\r\nenable 0')
291
+ led_status = client.get_led_status()
292
+ self.assertEqual(led_status, False)
293
+
294
+
295
+ if __name__ == '__main__':
296
+ main()
@@ -10,6 +10,7 @@ from tplinkrouterc6u.client.c5400x import TplinkC5400XRouter
10
10
  from tplinkrouterc6u.client.c1200 import TplinkC1200Router
11
11
  from tplinkrouterc6u.client.xdr import TPLinkXDRClient
12
12
  from tplinkrouterc6u.client.wdr import TplinkWDRRouter
13
+ from tplinkrouterc6u.client.re330 import TplinkRE330Router
13
14
  from tplinkrouterc6u.provider import TplinkRouterProvider
14
15
  from tplinkrouterc6u.common.package_enum import Connection, VPN
15
16
  from tplinkrouterc6u.common.dataclass import (