Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2025-07-06 19:50:08 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Use SslTransport only if encryption_type is AES, log the config and support SMART.KLAP.HTTPS

Browse Source
This commit is contained in:
Teemu Rytilahti 2025-01-03 18:59:24 +01:00
parent 3da27a5f22
commit 05ad95104c
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
kasa/device_factory.py
View File

@ -6,6 +6,7 @@ import logging
import time import time
from typing import Any from typing import Any
from . import EncryptType
from .device import Device from .device import Device
from .device_type import DeviceType from .device_type import DeviceType
from .deviceconfig import DeviceConfig, DeviceEncryptionType, DeviceFamily from .deviceconfig import DeviceConfig, DeviceEncryptionType, DeviceFamily
@ -189,6 +190,7 @@ def get_protocol(config: DeviceConfig, *, strict: bool = False) -> BaseProtocol
:param config: Device config to derive protocol :param config: Device config to derive protocol
:param strict: Require exact match on encrypt type :param strict: Require exact match on encrypt type
""" """
_LOGGER.debug("Finding protocol for %s", config.host)
ctype = config.connection_type ctype = config.connection_type
protocol_name = ctype.device_family.value.split(".")[0] protocol_name = ctype.device_family.value.split(".")[0]
_LOGGER.debug("Finding protocol for %s", ctype.device_family) _LOGGER.debug("Finding protocol for %s", ctype.device_family)
@ -203,9 +205,11 @@ def get_protocol(config: DeviceConfig, *, strict: bool = False) -> BaseProtocol
return None return None
return IotProtocol(transport=LinkieTransportV2(config=config)) return IotProtocol(transport=LinkieTransportV2(config=config))
if ctype.device_family is DeviceFamily.SmartTapoRobovac: # Older FW used a different transport
if strict and ctype.encryption_type is not DeviceEncryptionType.Aes: if strict and ctype.encryption_type is not DeviceEncryptionType.Aes:
return None return None
and ctype.encryption_type is EncryptType.Aes
):
return SmartProtocol(transport=SslTransport(config=config)) return SmartProtocol(transport=SslTransport(config=config))
protocol_transport_key = ( protocol_transport_key = (
@ -223,6 +227,7 @@ def get_protocol(config: DeviceConfig, *, strict: bool = False) -> BaseProtocol
"IOT.KLAP": (IotProtocol, KlapTransport), "IOT.KLAP": (IotProtocol, KlapTransport),
"SMART.AES": (SmartProtocol, AesTransport), "SMART.AES": (SmartProtocol, AesTransport),
"SMART.KLAP": (SmartProtocol, KlapTransportV2), "SMART.KLAP": (SmartProtocol, KlapTransportV2),
"SMART.KLAP.HTTPS": (SmartProtocol, KlapTransportV2),
# H200 is device family SMART.TAPOHUB and uses SmartCamProtocol so use # H200 is device family SMART.TAPOHUB and uses SmartCamProtocol so use
# https to distuingish from SmartProtocol devices # https to distuingish from SmartProtocol devices
"SMART.AES.HTTPS": (SmartCamProtocol, SslAesTransport), "SMART.AES.HTTPS": (SmartCamProtocol, SslAesTransport),