From e9109447a7818bedfb1b34fd574e76511c0adf54 Mon Sep 17 00:00:00 2001
From: "Steven B." <51370195+sdb9696@users.noreply.github.com>
Date: Mon, 16 Dec 2024 12:20:26 +0000
Subject: [PATCH] Add smartcam modules to package inits (#1376)

---
 kasa/__init__.py                   | 3 ++-
 kasa/protocols/__init__.py         | 2 ++
 kasa/protocols/smartcamprotocol.py | 2 +-
 kasa/transports/__init__.py        | 2 ++
 4 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/kasa/__init__.py b/kasa/__init__.py
index ee52eb3a..b8871f99 100755
--- a/kasa/__init__.py
+++ b/kasa/__init__.py
@@ -38,7 +38,7 @@ from kasa.feature import Feature
 from kasa.interfaces.light import HSV, ColorTempRange, Light, LightState
 from kasa.interfaces.thermostat import Thermostat, ThermostatState
 from kasa.module import Module
-from kasa.protocols import BaseProtocol, IotProtocol, SmartProtocol
+from kasa.protocols import BaseProtocol, IotProtocol, SmartCamProtocol, SmartProtocol
 from kasa.protocols.iotprotocol import _deprecated_TPLinkSmartHomeProtocol  # noqa: F401
 from kasa.smartcam.modules.camera import StreamResolution
 from kasa.transports import BaseTransport
@@ -52,6 +52,7 @@ __all__ = [
     "BaseTransport",
     "IotProtocol",
     "SmartProtocol",
+    "SmartCamProtocol",
     "LightState",
     "TurnOnBehaviors",
     "TurnOnBehavior",
diff --git a/kasa/protocols/__init__.py b/kasa/protocols/__init__.py
index 44130d7f..b994d732 100644
--- a/kasa/protocols/__init__.py
+++ b/kasa/protocols/__init__.py
@@ -2,6 +2,7 @@
 
 from .iotprotocol import IotProtocol
 from .protocol import BaseProtocol
+from .smartcamprotocol import SmartCamProtocol
 from .smartprotocol import SmartErrorCode, SmartProtocol
 
 __all__ = [
@@ -9,4 +10,5 @@ __all__ = [
     "IotProtocol",
     "SmartErrorCode",
     "SmartProtocol",
+    "SmartCamProtocol",
 ]
diff --git a/kasa/protocols/smartcamprotocol.py b/kasa/protocols/smartcamprotocol.py
index 12caa207..324f8056 100644
--- a/kasa/protocols/smartcamprotocol.py
+++ b/kasa/protocols/smartcamprotocol.py
@@ -19,7 +19,7 @@ from ..transports.sslaestransport import (
     SMART_RETRYABLE_ERRORS,
     SmartErrorCode,
 )
-from . import SmartProtocol
+from .smartprotocol import SmartProtocol
 
 _LOGGER = logging.getLogger(__name__)
 
diff --git a/kasa/transports/__init__.py b/kasa/transports/__init__.py
index 602d0cca..192b4156 100644
--- a/kasa/transports/__init__.py
+++ b/kasa/transports/__init__.py
@@ -4,6 +4,7 @@ from .aestransport import AesEncyptionSession, AesTransport
 from .basetransport import BaseTransport
 from .klaptransport import KlapTransport, KlapTransportV2
 from .linkietransport import LinkieTransportV2
+from .sslaestransport import SslAesTransport
 from .ssltransport import SslTransport
 from .xortransport import XorEncryption, XorTransport
 
@@ -11,6 +12,7 @@ __all__ = [
     "AesTransport",
     "AesEncyptionSession",
     "SslTransport",
+    "SslAesTransport",
     "BaseTransport",
     "KlapTransport",
     "KlapTransportV2",