Create __init__.py

switchbot/__init__.py

@@ -0,0 +1,49 @@
+"""Library to handle connection with Switchbot"""
+
+import logging
+
+import bluepy
+
+UUID = "cba20d00-224d-11e6-9fb8-0002a5d5c51b"
+HANDLE = "cba20002-224d-11e6-9fb8-0002a5d5c51b"
+ON_KEY = "570101"
+OFF_KEY = "570102"
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class Switchmate:
+    """Representation of a Switchmate."""
+
+    def __init__(self, mac) -> None:
+        self._mac = mac
+
+    def _sendpacket(self, key, retry=2) -> bool:
+        try:
+            device = bluepy.btle.Peripheral(self._mac,
+                                            bluepy.btle.ADDR_TYPE_RANDOM)
+            hand_service = device.getServiceByUUID(UUID)
+            hand = hand_service.getCharacteristics(HANDLE)[0]
+            hand.write(binascii.a2b_hex(key))
+            device.disconnect()
+        except bluepy.btle.BTLEException:
+            _LOGGER.error("Cannot connect to switchbot.", exc_info=True)
+            if retry < 1:
+                return False
+            self._sendpacket(key, retry-1)
+        return True
+
+    def update(self) -> None:
+        """Synchronize state with switch."""
+        try:
+            return self._device.readCharacteristic(HANDLE) == ON_KEY
+        except bluepy.btle.BTLEException:
+            self._connect()
+
+    def turn_on(self, **kwargs) -> None:
+        """Turn device on."""
+        return self._sendpacket(ON_KEY)
+
+    def turn_off(self, **kwargs) -> None:
+        """Turn device off."""
+        return self._sendpacket(OFF_KEY)