PySwitchbot 0.51.0__tar.gz → 0.54.0__tar.gz

pyswitchbot-0.54.0/PKG-INFO

@@ -0,0 +1,59 @@
+Metadata-Version: 2.1
+Name: PySwitchbot
+Version: 0.54.0
+Summary: A library to communicate with Switchbot
+Home-page: https://github.com/sblibs/pySwitchbot/
+Author: Daniel Hjelseth Hoyer
+License: MIT
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Other Environment
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Topic :: Home Automation
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: aiohttp>=3.9.5
+Requires-Dist: bleak>=0.19.0
+Requires-Dist: bleak-retry-connector>=3.4.0
+Requires-Dist: cryptography>=39.0.0
+Requires-Dist: pyOpenSSL>=23.0.0
+
+# pySwitchbot [![Build Status](https://travis-ci.org/sblibs/pySwitchbot.svg?branch=master)](https://travis-ci.org/sblibs/pySwitchbot)
+
+Library to control Switchbot IoT devices https://www.switch-bot.com/bot
+
+## Obtaining locks encryption key
+
+Using the script `scripts/get_encryption_key.py` you can manually obtain locks encryption key.
+
+Usage:
+
+```shell
+$ python3 get_encryption_key.py MAC USERNAME
+Key ID: xx
+Encryption key: xxxxxxxxxxxxxxxx
+```
+
+Where `MAC` is MAC address of the lock and `USERNAME` is your SwitchBot account username, after that script will ask for your password.
+If authentication succeeds then script should output your key id and encryption key.
+
+Examples:
+
+- WoLock
+
+```python
+import asyncio
+from switchbot.discovery import GetSwitchbotDevices
+from switchbot.devices import lock
+
+
+async def main():
+    wolock = await GetSwitchbotDevices().get_locks()
+    await lock.SwitchbotLock(wolock['32C0F607-18B8-xxxx-xxxx-xxxxxxxxxx'].device, "key-id", "encryption-key").get_lock_status()
+
+
+asyncio.run(main())
+
+```

pyswitchbot-0.54.0/PySwitchbot.egg-info/PKG-INFO

@@ -0,0 +1,59 @@
+Metadata-Version: 2.1
+Name: PySwitchbot
+Version: 0.54.0
+Summary: A library to communicate with Switchbot
+Home-page: https://github.com/sblibs/pySwitchbot/
+Author: Daniel Hjelseth Hoyer
+License: MIT
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Other Environment
+Classifier: Intended Audience :: Developers
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python
+Classifier: Topic :: Home Automation
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: aiohttp>=3.9.5
+Requires-Dist: bleak>=0.19.0
+Requires-Dist: bleak-retry-connector>=3.4.0
+Requires-Dist: cryptography>=39.0.0
+Requires-Dist: pyOpenSSL>=23.0.0
+
+# pySwitchbot [![Build Status](https://travis-ci.org/sblibs/pySwitchbot.svg?branch=master)](https://travis-ci.org/sblibs/pySwitchbot)
+
+Library to control Switchbot IoT devices https://www.switch-bot.com/bot
+
+## Obtaining locks encryption key
+
+Using the script `scripts/get_encryption_key.py` you can manually obtain locks encryption key.
+
+Usage:
+
+```shell
+$ python3 get_encryption_key.py MAC USERNAME
+Key ID: xx
+Encryption key: xxxxxxxxxxxxxxxx
+```
+
+Where `MAC` is MAC address of the lock and `USERNAME` is your SwitchBot account username, after that script will ask for your password.
+If authentication succeeds then script should output your key id and encryption key.
+
+Examples:
+
+- WoLock
+
+```python
+import asyncio
+from switchbot.discovery import GetSwitchbotDevices
+from switchbot.devices import lock
+
+
+async def main():
+    wolock = await GetSwitchbotDevices().get_locks()
+    await lock.SwitchbotLock(wolock['32C0F607-18B8-xxxx-xxxx-xxxxxxxxxx'].device, "key-id", "encryption-key").get_lock_status()
+
+
+asyncio.run(main())
+
+```

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/PySwitchbot.egg-info/SOURCES.txt

@@ -23,11 +23,13 @@ switchbot/adv_parsers/contact.py
 switchbot/adv_parsers/curtain.py
 switchbot/adv_parsers/hub2.py
 switchbot/adv_parsers/humidifier.py
+switchbot/adv_parsers/keypad.py
 switchbot/adv_parsers/light_strip.py
 switchbot/adv_parsers/lock.py
 switchbot/adv_parsers/meter.py
 switchbot/adv_parsers/motion.py
 switchbot/adv_parsers/plug.py
+switchbot/adv_parsers/relay_switch.py
 switchbot/devices/__init__.py
 switchbot/devices/base_cover.py
 switchbot/devices/base_light.py
@@ -39,12 +41,15 @@ switchbot/devices/contact.py
 switchbot/devices/curtain.py
 switchbot/devices/device.py
 switchbot/devices/humidifier.py
+switchbot/devices/keypad.py
 switchbot/devices/light_strip.py
 switchbot/devices/lock.py
 switchbot/devices/meter.py
 switchbot/devices/motion.py
 switchbot/devices/plug.py
+switchbot/devices/relay_switch.py
 tests/test_adv_parser.py
 tests/test_base_cover.py
 tests/test_blind_tilt.py
-tests/test_curtain.py
+tests/test_curtain.py
+tests/test_relay_switch.py

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/README.md

@@ -1,4 +1,4 @@
-# pySwitchbot [![Build Status](https://travis-ci.org/Danielhiversen/pySwitchbot.svg?branch=master)](https://travis-ci.org/Danielhiversen/pySwitchbot)
+# pySwitchbot [![Build Status](https://travis-ci.org/sblibs/pySwitchbot.svg?branch=master)](https://travis-ci.org/sblibs/pySwitchbot)
 
 Library to control Switchbot IoT devices https://www.switch-bot.com/bot
 

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/setup.py

@@ -1,5 +1,10 @@
+from pathlib import Path
+
 from setuptools import setup
 
+this_directory = Path(__file__).parent
+long_description = (this_directory / "README.md").read_text()
+
 setup(
     name="PySwitchbot",
     packages=["switchbot", "switchbot.devices", "switchbot.adv_parsers"],
@@ -10,10 +15,12 @@ setup(
         "cryptography>=39.0.0",
         "pyOpenSSL>=23.0.0",
     ],
-    version="0.51.0",
+    version="0.54.0",
     description="A library to communicate with Switchbot",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
     author="Daniel Hjelseth Hoyer",
-    url="https://github.com/Danielhiversen/pySwitchbot/",
+    url="https://github.com/sblibs/pySwitchbot/",
     license="MIT",
     classifiers=[
         "Development Status :: 3 - Alpha",

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/__init__.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 from bleak_retry_connector import (
@@ -26,6 +27,7 @@ from .devices.humidifier import SwitchbotHumidifier
 from .devices.light_strip import SwitchbotLightStrip
 from .devices.lock import SwitchbotLock
 from .devices.plug import SwitchbotPlugMini
+from .devices.relay_switch import SwitchbotRelaySwitch
 from .discovery import GetSwitchbotDevices
 from .models import SwitchBotAdvertisement
 
@@ -54,4 +56,5 @@ __all__ = [
     "SwitchbotModel",
     "SwitchbotLock",
     "SwitchbotBlindTilt",
+    "SwitchbotRelaySwitch",
 ]

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parser.py

@@ -18,11 +18,16 @@ from .adv_parsers.contact import process_wocontact
 from .adv_parsers.curtain import process_wocurtain
 from .adv_parsers.hub2 import process_wohub2
 from .adv_parsers.humidifier import process_wohumidifier
+from .adv_parsers.keypad import process_wokeypad
 from .adv_parsers.light_strip import process_wostrip
 from .adv_parsers.lock import process_wolock, process_wolock_pro
 from .adv_parsers.meter import process_wosensorth, process_wosensorth_c
 from .adv_parsers.motion import process_wopresence
 from .adv_parsers.plug import process_woplugmini
+from .adv_parsers.relay_switch import (
+    process_worelay_switch_1,
+    process_worelay_switch_1pm,
+)
 from .const import SwitchbotModel
 from .models import SwitchBotAdvertisement
 
@@ -69,7 +74,6 @@ SUPPORTED_TYPES: dict[str, SwitchbotSupportedType] = {
         "modelFriendlyName": "Light Strip",
         "func": process_wostrip,
         "manufacturer_id": 2409,
-        "manufacturer_data_length": 16,
     },
     "{": {
         "modelName": SwitchbotModel.CURTAIN,
@@ -174,6 +178,24 @@ SUPPORTED_TYPES: dict[str, SwitchbotSupportedType] = {
         "func": process_woblindtilt,
         "manufacturer_id": 2409,
     },
+    "y": {
+        "modelName": SwitchbotModel.KEYPAD,
+        "modelFriendlyName": "Keypad",
+        "func": process_wokeypad,
+        "manufacturer_id": 2409,
+    },
+    "<": {
+        "modelName": SwitchbotModel.RELAY_SWITCH_1PM,
+        "modelFriendlyName": "Relay Switch 1PM",
+        "func": process_worelay_switch_1pm,
+        "manufacturer_id": 2409,
+    },
+    ";": {
+        "modelName": SwitchbotModel.RELAY_SWITCH_1,
+        "modelFriendlyName": "Relay Switch 1",
+        "func": process_worelay_switch_1,
+        "manufacturer_id": 2409,
+    },
 }
 
 _SWITCHBOT_MODEL_TO_CHAR = {

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/blind_tilt.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/bot.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/bulb.py

@@ -1,4 +1,5 @@
 """Bulb parser."""
+
 from __future__ import annotations
 
 

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/ceiling_light.py

@@ -1,4 +1,5 @@
 """Ceiling Light adv parser."""
+
 from __future__ import annotations
 
 import logging

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/contact.py

@@ -1,4 +1,5 @@
 """Contact sensor parser."""
+
 from __future__ import annotations
 
 

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/curtain.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/humidifier.py

@@ -1,4 +1,5 @@
 """Humidifier adv parser."""
+
 from __future__ import annotations
 
 import logging

pyswitchbot-0.54.0/switchbot/adv_parsers/keypad.py

@@ -0,0 +1,22 @@
+"""Keypad parser."""
+
+from __future__ import annotations
+
+import logging
+
+_LOGGER = logging.getLogger(__name__)
+
+
+def process_wokeypad(
+    data: bytes | None,
+    mfr_data: bytes | None,
+) -> dict[str, bool | int | None]:
+    """Process woKeypad services data."""
+    if data is None or mfr_data is None:
+        return {"battery": None, "attempt_state": None}
+
+    _LOGGER.debug("mfr_data: %s", mfr_data.hex())
+    if data:
+        _LOGGER.debug("data: %s", data.hex())
+
+    return {"battery": data[2] & 0b01111111, "attempt_state": mfr_data[6]}

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/light_strip.py

@@ -1,4 +1,5 @@
 """Light strip adv parser."""
+
 from __future__ import annotations
 
 

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/lock.py

@@ -1,4 +1,5 @@
 """Lock parser."""
+
 from __future__ import annotations
 
 import logging

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/meter.py

@@ -1,4 +1,5 @@
 """Meter parser."""
+
 from __future__ import annotations
 
 import struct

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/motion.py

@@ -1,4 +1,5 @@
 """Motion sensor parser."""
+
 from __future__ import annotations
 
 

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/adv_parsers/plug.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 

pyswitchbot-0.54.0/switchbot/adv_parsers/relay_switch.py

@@ -0,0 +1,32 @@
+"""Relay Switch adv parser."""
+
+from __future__ import annotations
+
+
+def process_worelay_switch_1pm(
+    data: bytes | None, mfr_data: bytes | None
+) -> dict[str, bool | int]:
+    """Process WoStrip services data."""
+    if mfr_data is None:
+        return {}
+    return {
+        "switchMode": True,  # for compatibility, useless
+        "sequence_number": mfr_data[6],
+        "isOn": bool(mfr_data[7] & 0b10000000),
+        "power": ((mfr_data[10] << 8) + mfr_data[11]) / 10,
+        "voltage": 0,
+        "current": 0,
+    }
+
+
+def process_worelay_switch_1(
+    data: bytes | None, mfr_data: bytes | None
+) -> dict[str, bool | int]:
+    """Process WoStrip services data."""
+    if mfr_data is None:
+        return {}
+    return {
+        "switchMode": True,  # for compatibility, useless
+        "sequence_number": mfr_data[6],
+        "isOn": bool(mfr_data[7] & 0b10000000),
+    }

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/const.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 from enum import Enum
@@ -52,6 +53,9 @@ class SwitchbotModel(StrEnum):
     LOCK_PRO = "WoLockPro"
     BLIND_TILT = "WoBlindTilt"
     HUB2 = "WoHub2"
+    KEYPAD = "WoKeypad"
+    RELAY_SWITCH_1PM = "Relay Switch 1PM"
+    RELAY_SWITCH_1 = "Relay Switch 1"
 
 
 class LockStatus(Enum):

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/devices/base_cover.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 import logging

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/devices/blind_tilt.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 import logging

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/devices/bot.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 import logging

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/devices/curtain.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 import logging

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/devices/device.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 import asyncio

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/devices/humidifier.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 import time

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/devices/lock.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot Lock."""
+
 from __future__ import annotations
 
 import asyncio

pyswitchbot-0.54.0/switchbot/devices/motion.py

@@ -0,0 +1 @@
+from __future__ import annotations

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/devices/plug.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 import time

pyswitchbot-0.54.0/switchbot/devices/relay_switch.py

@@ -0,0 +1,185 @@
+import asyncio
+import logging
+import time
+from typing import Any
+
+from bleak.backends.device import BLEDevice
+from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
+
+from ..const import SwitchbotModel
+from ..models import SwitchBotAdvertisement
+from .device import SwitchbotDevice
+
+_LOGGER = logging.getLogger(__name__)
+
+COMMAND_HEADER = "57"
+COMMAND_GET_CK_IV = f"{COMMAND_HEADER}0f2103"
+COMMAND_TURN_OFF = f"{COMMAND_HEADER}0f70010000"
+COMMAND_TURN_ON = f"{COMMAND_HEADER}0f70010100"
+COMMAND_TOGGLE = f"{COMMAND_HEADER}0f70010200"
+COMMAND_GET_VOLTAGE_AND_CURRENT = f"{COMMAND_HEADER}0f7106000000"
+PASSIVE_POLL_INTERVAL = 10 * 60
+
+
+class SwitchbotRelaySwitch(SwitchbotDevice):
+    """Representation of a Switchbot relay switch 1pm."""
+
+    def __init__(
+        self,
+        device: BLEDevice,
+        key_id: str,
+        encryption_key: str,
+        interface: int = 0,
+        model: SwitchbotModel = SwitchbotModel.RELAY_SWITCH_1PM,
+        **kwargs: Any,
+    ) -> None:
+        if len(key_id) == 0:
+            raise ValueError("key_id is missing")
+        elif len(key_id) != 2:
+            raise ValueError("key_id is invalid")
+        if len(encryption_key) == 0:
+            raise ValueError("encryption_key is missing")
+        elif len(encryption_key) != 32:
+            raise ValueError("encryption_key is invalid")
+        self._iv = None
+        self._cipher = None
+        self._key_id = key_id
+        self._encryption_key = bytearray.fromhex(encryption_key)
+        self._model: SwitchbotModel = model
+        self._force_next_update = False
+        super().__init__(device, None, interface, **kwargs)
+
+    def update_from_advertisement(self, advertisement: SwitchBotAdvertisement) -> None:
+        """Update device data from advertisement."""
+        # Obtain voltage and current through command.
+        adv_data = advertisement.data["data"]
+        if previous_voltage := self._get_adv_value("voltage"):
+            adv_data["voltage"] = previous_voltage
+        if previous_current := self._get_adv_value("current"):
+            adv_data["current"] = previous_current
+        current_state = self._get_adv_value("sequence_number")
+        super().update_from_advertisement(advertisement)
+        new_state = self._get_adv_value("sequence_number")
+        _LOGGER.debug(
+            "%s: update advertisement: %s (seq before: %s) (seq after: %s)",
+            self.name,
+            advertisement,
+            current_state,
+            new_state,
+        )
+        if current_state != new_state:
+            self._force_next_update = True
+
+    async def update(self, interface: int | None = None) -> None:
+        """Update state of device."""
+        if info := await self.get_voltage_and_current():
+            self._last_full_update = time.monotonic()
+            self._update_parsed_data(info)
+            self._fire_callbacks()
+
+    async def get_voltage_and_current(self) -> dict[str, Any] | None:
+        """Get voltage and current because advtisement don't have these"""
+        result = await self._send_command(COMMAND_GET_VOLTAGE_AND_CURRENT)
+        ok = self._check_command_result(result, 0, {1})
+        if ok:
+            return {
+                "voltage": ((result[9] << 8) + result[10]) / 10,
+                "current": (result[11] << 8) + result[12],
+            }
+        return None
+
+    def poll_needed(self, seconds_since_last_poll: float | None) -> bool:
+        """Return if device needs polling."""
+        if self._force_next_update:
+            self._force_next_update = False
+            return True
+        if (
+            seconds_since_last_poll is not None
+            and seconds_since_last_poll < PASSIVE_POLL_INTERVAL
+        ):
+            return False
+        time_since_last_full_update = time.monotonic() - self._last_full_update
+        if time_since_last_full_update < PASSIVE_POLL_INTERVAL:
+            return False
+        return True
+
+    async def turn_on(self) -> bool:
+        """Turn device on."""
+        result = await self._send_command(COMMAND_TURN_ON)
+        ok = self._check_command_result(result, 0, {1})
+        if ok:
+            self._override_state({"isOn": True})
+            self._fire_callbacks()
+        return ok
+
+    async def turn_off(self) -> bool:
+        """Turn device off."""
+        result = await self._send_command(COMMAND_TURN_OFF)
+        ok = self._check_command_result(result, 0, {1})
+        if ok:
+            self._override_state({"isOn": False})
+            self._fire_callbacks()
+        return ok
+
+    async def async_toggle(self, **kwargs) -> bool:
+        """Toggle device."""
+        result = await self._send_command(COMMAND_TOGGLE)
+        status = self._check_command_result(result, 0, {1})
+        return status
+
+    def is_on(self) -> bool | None:
+        """Return switch state from cache."""
+        return self._get_adv_value("isOn")
+
+    async def _send_command(
+        self, key: str, retry: int | None = None, encrypt: bool = True
+    ) -> bytes | None:
+        if not encrypt:
+            return await super()._send_command(key[:2] + "000000" + key[2:], retry)
+
+        result = await self._ensure_encryption_initialized()
+        if not result:
+            return None
+
+        encrypted = (
+            key[:2] + self._key_id + self._iv[0:2].hex() + self._encrypt(key[2:])
+        )
+        result = await super()._send_command(encrypted, retry)
+        return result[:1] + self._decrypt(result[4:])
+
+    async def _ensure_encryption_initialized(self) -> bool:
+        if self._iv is not None:
+            return True
+
+        result = await self._send_command(
+            COMMAND_GET_CK_IV + self._key_id, encrypt=False
+        )
+        ok = self._check_command_result(result, 0, {1})
+        if ok:
+            self._iv = result[4:]
+
+        return ok
+
+    async def _execute_disconnect(self) -> None:
+        await super()._execute_disconnect()
+        self._iv = None
+        self._cipher = None
+
+    def _get_cipher(self) -> Cipher:
+        if self._cipher is None:
+            self._cipher = Cipher(
+                algorithms.AES128(self._encryption_key), modes.CTR(self._iv)
+            )
+        return self._cipher
+
+    def _encrypt(self, data: str) -> str:
+        if len(data) == 0:
+            return ""
+        encryptor = self._get_cipher().encryptor()
+        return (encryptor.update(bytearray.fromhex(data)) + encryptor.finalize()).hex()
+
+    def _decrypt(self, data: bytearray) -> bytes:
+        if len(data) == 0:
+            return b""
+        decryptor = self._get_cipher().decryptor()
+        return decryptor.update(data) + decryptor.finalize()

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/discovery.py

@@ -124,6 +124,10 @@ class GetSwitchbotDevices:
         lock_pros = await self._get_devices_by_model("$")
         return {**locks, **lock_pros}
 
+    async def get_keypads(self) -> dict[str, SwitchBotAdvertisement]:
+        """Return all WoKeypad/Keypad devices with services data."""
+        return await self._get_devices_by_model("y")
+
     async def get_device_data(
         self, address: str
     ) -> dict[str, SwitchBotAdvertisement] | None:

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/enum.py

@@ -1,4 +1,5 @@
 """Enum backports from standard lib."""
+
 from __future__ import annotations
 
 from enum import Enum

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/switchbot/models.py

@@ -1,4 +1,5 @@
 """Library to handle connection with Switchbot."""
+
 from __future__ import annotations
 
 from dataclasses import dataclass

{pyswitchbot-0.51.0 → pyswitchbot-0.54.0}/tests/test_adv_parser.py

@@ -807,43 +807,6 @@ def test_bulb_active():
     )
 
 
-def test_lightstrip_passive():
-    """Test parsing lightstrip as passive."""
-    ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
-    adv_data = generate_advertisement_data(
-        manufacturer_data={
-            2409: b"`U\xf9(\xe5\x96\x00d\x02\xb0\x00\x00\x00\x00\x00\x00"
-        },
-        service_data={},
-        tx_power=-127,
-        rssi=-50,
-    )
-    result = parse_advertisement_data(ble_device, adv_data)
-    assert result == SwitchBotAdvertisement(
-        address="aa:bb:cc:dd:ee:ff",
-        data={
-            "data": {
-                "brightness": 100,
-                "color_mode": 2,
-                "delay": False,
-                "isOn": False,
-                "loop_index": 0,
-                "preset": False,
-                "sequence_number": 0,
-                "speed": 48,
-            },
-            "isEncrypted": False,
-            "model": "r",
-            "modelFriendlyName": "Light Strip",
-            "modelName": SwitchbotModel.LIGHT_STRIP,
-            "rawAdvData": None,
-        },
-        device=ble_device,
-        rssi=-50,
-        active=False,
-    )
-
-
 def test_wosensor_passive_and_active():
     """Test parsing wosensor as passive with active data as well."""
     ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
@@ -1684,3 +1647,93 @@ def test_meter_pro_c_passive() -> None:
         rssi=-67,
         active=False,
     )
+
+
+def test_parse_advertisement_data_keypad():
+    """Test parse_advertisement_data for the keypad."""
+    ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
+    adv_data = generate_advertisement_data(
+        manufacturer_data={2409: b"\xeb\x13\x02\xe6#\x0f\x8fd\x00\x00\x00\x00"},
+        service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"y\x00d"},
+        rssi=-67,
+    )
+    result = parse_advertisement_data(ble_device, adv_data, SwitchbotModel.KEYPAD)
+    assert result == SwitchBotAdvertisement(
+        address="aa:bb:cc:dd:ee:ff",
+        data={
+            "data": {"attempt_state": 143, "battery": 100},
+            "isEncrypted": False,
+            "model": "y",
+            "modelFriendlyName": "Keypad",
+            "modelName": SwitchbotModel.KEYPAD,
+            "rawAdvData": b"y\x00d",
+        },
+        device=ble_device,
+        rssi=-67,
+        active=True,
+    )
+
+
+def test_parse_advertisement_data_relay_switch_1pm():
+    """Test parse_advertisement_data for the keypad."""
+    ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
+    adv_data = generate_advertisement_data(
+        manufacturer_data={2409: b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00"},
+        service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b"<\x00\x00\x00"},
+        rssi=-67,
+    )
+    result = parse_advertisement_data(
+        ble_device, adv_data, SwitchbotModel.RELAY_SWITCH_1PM
+    )
+    assert result == SwitchBotAdvertisement(
+        address="aa:bb:cc:dd:ee:ff",
+        data={
+            "data": {
+                "switchMode": True,
+                "sequence_number": 71,
+                "isOn": True,
+                "power": 4.9,
+                "voltage": 0,
+                "current": 0,
+            },
+            "isEncrypted": False,
+            "model": "<",
+            "modelFriendlyName": "Relay Switch 1PM",
+            "modelName": SwitchbotModel.RELAY_SWITCH_1PM,
+            "rawAdvData": b"<\x00\x00\x00",
+        },
+        device=ble_device,
+        rssi=-67,
+        active=True,
+    )
+
+
+def test_parse_advertisement_data_relay_switch_1():
+    """Test parse_advertisement_data for the keypad."""
+    ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
+    adv_data = generate_advertisement_data(
+        manufacturer_data={2409: b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00"},
+        service_data={"0000fd3d-0000-1000-8000-00805f9b34fb": b";\x00\x00\x00"},
+        rssi=-67,
+    )
+    result = parse_advertisement_data(
+        ble_device, adv_data, SwitchbotModel.RELAY_SWITCH_1
+    )
+    assert result == SwitchBotAdvertisement(
+        address="aa:bb:cc:dd:ee:ff",
+        data={
+            "data": {
+                "switchMode": True,
+                "sequence_number": 71,
+                "isOn": True,
+            },
+            "isEncrypted": False,
+            "model": ";",
+            "modelFriendlyName": "Relay Switch 1",
+            "modelName": SwitchbotModel.RELAY_SWITCH_1,
+            "rawAdvData": b";\x00\x00\x00",
+        },
+        device=ble_device,
+        rssi=-67,
+        active=True,
+    )

pyswitchbot-0.54.0/tests/test_relay_switch.py

@@ -0,0 +1,60 @@
+from unittest.mock import AsyncMock
+
+import pytest
+from bleak.backends.device import BLEDevice
+
+from switchbot import SwitchBotAdvertisement, SwitchbotModel
+from switchbot.devices import relay_switch
+
+from .test_adv_parser import generate_ble_device
+
+
+def create_device_for_command_testing(calibration=True, reverse_mode=False):
+    ble_device = generate_ble_device("aa:bb:cc:dd:ee:ff", "any")
+    relay_switch_device = relay_switch.SwitchbotRelaySwitch(
+        ble_device, "ff", "ffffffffffffffffffffffffffffffff"
+    )
+    relay_switch_device.update_from_advertisement(make_advertisement_data(ble_device))
+    return relay_switch_device
+
+
+def make_advertisement_data(ble_device: BLEDevice):
+    """Set advertisement data with defaults."""
+
+    return SwitchBotAdvertisement(
+        address="aa:bb:cc:dd:ee:ff",
+        data={
+            "rawAdvData": b"$X|\x0866G\x81\x00\x00\x001\x00\x00\x00\x00",
+            "data": {
+                "switchMode": True,
+                "sequence_number": 71,
+                "isOn": True,
+                "power": 4.9,
+                "voltage": 0,
+                "current": 0,
+            },
+            "isEncrypted": False,
+            "model": "<",
+            "modelFriendlyName": "Relay Switch 1PM",
+            "modelName": SwitchbotModel.RELAY_SWITCH_1PM,
+        },
+        device=ble_device,
+        rssi=-80,
+        active=True,
+    )
+
+
+@pytest.mark.asyncio
+async def test_turn_on():
+    relay_switch_device = create_device_for_command_testing()
+    relay_switch_device._send_command = AsyncMock(return_value=b"\x01")
+    await relay_switch_device.turn_on()
+    assert relay_switch_device.is_on() is True
+
+
+@pytest.mark.asyncio
+async def test_trun_off():
+    relay_switch_device = create_device_for_command_testing()
+    relay_switch_device._send_command = AsyncMock(return_value=b"\x01")
+    await relay_switch_device.turn_off()
+    assert relay_switch_device.is_on() is False

pyswitchbot-0.51.0/PKG-INFO

@@ -1,20 +0,0 @@
-Metadata-Version: 2.1
-Name: PySwitchbot
-Version: 0.51.0
-Summary: A library to communicate with Switchbot
-Home-page: https://github.com/Danielhiversen/pySwitchbot/
-Author: Daniel Hjelseth Hoyer
-License: MIT
-Classifier: Development Status :: 3 - Alpha
-Classifier: Environment :: Other Environment
-Classifier: Intended Audience :: Developers
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python
-Classifier: Topic :: Home Automation
-Classifier: Topic :: Software Development :: Libraries :: Python Modules
-License-File: LICENSE
-Requires-Dist: aiohttp>=3.9.5
-Requires-Dist: bleak>=0.19.0
-Requires-Dist: bleak-retry-connector>=3.4.0
-Requires-Dist: cryptography>=39.0.0
-Requires-Dist: pyOpenSSL>=23.0.0

pyswitchbot-0.51.0/PySwitchbot.egg-info/PKG-INFO

@@ -1,20 +0,0 @@
-Metadata-Version: 2.1
-Name: PySwitchbot
-Version: 0.51.0
-Summary: A library to communicate with Switchbot
-Home-page: https://github.com/Danielhiversen/pySwitchbot/
-Author: Daniel Hjelseth Hoyer
-License: MIT
-Classifier: Development Status :: 3 - Alpha
-Classifier: Environment :: Other Environment
-Classifier: Intended Audience :: Developers
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python
-Classifier: Topic :: Home Automation
-Classifier: Topic :: Software Development :: Libraries :: Python Modules
-License-File: LICENSE
-Requires-Dist: aiohttp>=3.9.5
-Requires-Dist: bleak>=0.19.0
-Requires-Dist: bleak-retry-connector>=3.4.0
-Requires-Dist: cryptography>=39.0.0
-Requires-Dist: pyOpenSSL>=23.0.0