switchbot-api 2.9.0__tar.gz → 2.10.0__tar.gz

{switchbot_api-2.9.0 → switchbot_api-2.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: switchbot_api
-Version: 2.9.0
+Version: 2.10.0
 Summary: An asynchronous library to use Switchbot API
 License: MIT
 License-File: LICENSE

{switchbot_api-2.9.0 → switchbot_api-2.10.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "switchbot_api"
-version = "2.9.0"
+version = "2.10.0"
 description = "An asynchronous library to use Switchbot API"
 authors = ["Ravaka Razafimanantsoa <contact@ravaka.dev>"]
 license = "MIT"

{switchbot_api-2.9.0 → switchbot_api-2.10.0}/switchbot_api/__init__.py

@@ -18,6 +18,7 @@ from aiohttp.hdrs import METH_GET, METH_POST
 from switchbot_api.commands import (
     AirConditionerCommands,
     AirPurifierCommands,
+    ArtFrameCommands,
     BatteryCirculatorFanCommands,
     BlindTiltCommands,
     BotCommands,
@@ -30,9 +31,11 @@ from switchbot_api.commands import (
     FanCommands,
     HumidifierCommands,
     HumidifierV2Commands,
+    KeyPadCommands,
     LightCommands,
     LockCommands,
     LockV2Commands,
+    LockV3Commands,
     OthersCommands,
     RGBWLightCommands,
     RGBWWLightCommands,
@@ -54,6 +57,7 @@ from switchbot_api.exceptions import (
 )
 from switchbot_api.models import (
     BatteryCirculatorFanMode,
+    BatteryLevel,
     PowerState,
     SmartRadiatorThermostatMode,
     VacuumCleanMode,
@@ -64,8 +68,10 @@ from switchbot_api.models import (
 __all__ = [
     "AirConditionerCommands",
     "AirPurifierCommands",
+    "ArtFrameCommands",
     "BatteryCirculatorFanCommands",
     "BatteryCirculatorFanMode",
+    "BatteryLevel",
     "BlindTiltCommands",
     "BotCommands",
     "CeilingLightCommands",
@@ -78,9 +84,11 @@ __all__ = [
     "FanCommands",
     "HumidifierCommands",
     "HumidifierV2Commands",
+    "KeyPadCommands",
     "LightCommands",
     "LockCommands",
     "LockV2Commands",
+    "LockV3Commands",
     "OthersCommands",
     "PowerState",
     "RGBWLightCommands",

{switchbot_api-2.9.0 → switchbot_api-2.10.0}/switchbot_api/commands.py

@@ -93,7 +93,21 @@ class LockV2Commands(Commands):
     @classmethod
     def get_supported_devices(cls) -> list[str]:
         """Get supported devices."""
-        return ["Smart Lock", "Smart Lock Pro", "Smart Lock Ultra"]
+        return ["Smart Lock", "Smart Lock Pro", "Smart Lock Ultra", "Smart Lock Vision"]
+
+
+class LockV3Commands(Commands):
+    """Lock commands."""
+
+    LOCK = "lock"
+    UNLOCK = "unlock"
+    DEADBOLT = "deadbolt"
+    NIGHT_LATCH_UNLOCK = "nightLatchUnlock"
+
+    @classmethod
+    def get_supported_devices(cls) -> list[str]:
+        """Get supported devices."""
+        return ["Smart Lock Vision Pro"]
 
 
 class HumidifierCommands(Commands):

{switchbot_api-2.9.0 → switchbot_api-2.10.0}/switchbot_api/exceptions.py

@@ -15,3 +15,7 @@ class SwitchBotAuthenticationError(SwitchBotError):
 
 class SwitchBotDeviceOfflineError(SwitchBotError):
     """Device currently offline."""
+
+
+class SwitchBotDeviceRequestError(SwitchBotError):
+    """Device request error."""

switchbot_api-2.10.0/switchbot_api/utils.py

@@ -0,0 +1,28 @@
+"""util for  SwitchBot API."""
+
+import aiohttp
+from aiohttp import ClientResponse
+
+from .exceptions import SwitchBotDeviceRequestError
+
+
+def check_response_status(response: ClientResponse) -> None:
+    """Check https response status."""
+    if response.status != 200:
+        msg = f"status code != 200 (actual: {response.status})"
+        raise SwitchBotDeviceRequestError(msg)
+
+
+async def get_file_stream_from_cloud(url: str, timeout: float = 5) -> bytes:
+    """Get file stream from cloud."""
+    # now only for download <AI Art Frame> Picture
+    try:
+        async with (
+            aiohttp.ClientSession() as session,
+            session.get(url, timeout=aiohttp.ClientTimeout(total=timeout)) as response,
+        ):
+            check_response_status(response)
+            return await response.read()
+    except Exception as e:
+        msg = f"{e}"
+        raise SwitchBotDeviceRequestError(msg) from e