Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2025-12-23 05:00:25 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Switch from TPLinkSmartHomeProtocol to IotProtocol/XorTransport

Browse Source
This commit is contained in:
sdb9696
2024-01-25 18:55:13 +00:00
parent c318303255
commit 0f3e4fc675
14 changed files with 148 additions and 348 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
kasa/discover.py
View File

@@ -25,8 +25,8 @@ from kasa.deviceconfig import ConnectionType, DeviceConfig, EncryptType
from kasa.exceptions import TimeoutException, UnsupportedDeviceException
from kasa.json import dumps as json_dumps
from kasa.json import loads as json_loads
from kasa.protocol import TPLinkSmartHomeProtocol
from kasa.smartdevice import SmartDevice, SmartDeviceException
from kasa.xortransport import XorEncryption
_LOGGER = logging.getLogger(__name__)
@@ -103,7 +103,7 @@ class _DiscoverProtocol(asyncio.DatagramProtocol):
"""Send number of discovery datagrams."""
req = json_dumps(Discover.DISCOVERY_QUERY)
_LOGGER.debug("[DISCOVERY] %s >> %s", self.target, Discover.DISCOVERY_QUERY)
encrypted_req = TPLinkSmartHomeProtocol.encrypt(req)
encrypted_req = XorEncryption.encrypt(req)
sleep_between_packets = self.discovery_timeout / self.discovery_packets
for i in range(self.discovery_packets):
if self.target in self.seen_hosts: # Stop sending for discover_single
@@ -400,7 +400,7 @@ class Discover:
def _get_device_instance_legacy(data: bytes, config: DeviceConfig) -> SmartDevice:
"""Get SmartDevice from legacy 9999 response."""
try:
info = json_loads(TPLinkSmartHomeProtocol.decrypt(data))
info = json_loads(XorEncryption.decrypt(data))
except Exception as ex:
raise SmartDeviceException(
f"Unable to read response from device: {config.host}: {ex}"