pySwitchbot/switchbot/__init__.py

"""Library to handle connection with Switchbot"""

import binascii
import logging

import bluepy

UUID = "cba20d00-224d-11e6-9fb8-0002a5d5c51b"
HANDLE = "cba20002-224d-11e6-9fb8-0002a5d5c51b"

PRESS_KEY = "570100"
ON_KEY = "570101"
OFF_KEY = "570102"

_LOGGER = logging.getLogger(__name__)


class Switchbot:
    """Representation of a Switchbot."""

    def __init__(self, mac) -> None:
        self._mac = mac
        self._device = None
        self._connect()

    def _connect(self) -> bool:
        if self._device is not None:
            _LOGGER.debug("Disconnecting")
            try:
                self._device.disconnect()
            except bluepy.btle.BTLEException:
                pass
        try:
            _LOGGER.debug("Connecting")
            self._device = bluepy.btle.Peripheral(self._mac,
                                                  bluepy.btle.ADDR_TYPE_RANDOM)
        except bluepy.btle.BTLEException:
            _LOGGER.warning("Failed to connect to Switchbot", exc_info=True)
            return False
        return True

    def _sendpacket(self, key, retry=2) -> bool:
        try:
            _LOGGER.debug("Prepare to send")
            hand_service = self._device.getServiceByUUID(UUID)
            hand = hand_service.getCharacteristics(HANDLE)[0]
            _LOGGER.debug("Sending command, %s", key)
            hand.write(binascii.a2b_hex(key))
        except bluepy.btle.BTLEException:
            if retry < 1 or not self._connect():
                _LOGGER.error("Cannot connect to switchbot.", exc_info=True)
                return False
            _LOGGER.warning("Cannot connect to switchbot. Retrying")
            return self._sendpacket(key, retry-1)
        return True

    def turn_on(self) -> None:
        """Turn device on."""
        return self._sendpacket(ON_KEY)

    def turn_off(self) -> None:
        """Turn device off."""
        return self._sendpacket(OFF_KEY)

    def press(self) -> None:
        """Press command to device."""
        return self._sendpacket(PRESS_KEY)