Initial commit

custom_components/plejd/pyplejd/__init__.py

@@ -0,0 +1,68 @@
+import logging
+from datetime import timedelta
+
+from bleak_retry_connector import close_stale_connections
+
+from .mesh import PlejdMesh
+from .api import get_cryptokey, get_devices
+from .plejd_device import PlejdDevice
+
+_LOGGER = logging.getLogger(__name__)
+
+class PlejdManager:
+
+    def __init__(self, credentials):
+        self.credentials = credentials
+        self.mesh = PlejdMesh()
+        self.mesh.statecallback = self._update_device
+        self.devices = { }
+        self.credentials = credentials
+
+    def discover_plejd(self, service_info, *_):
+        _LOGGER.debug("Adding plejd %s", service_info)
+        self.mesh.add_mesh_node(service_info.device)
+
+    async def close_stale(self, device):
+        _LOGGER.info("Closing stale connections for %s", device)
+        await close_stale_connections(device)
+
+    @property
+    def connected(self):
+        return self.mesh is not None and self.mesh.connected
+
+    async def get_devices(self):
+        devices = await get_devices(**self.credentials)
+        self.devices = {k: PlejdDevice(self, **v) for (k,v) in devices.items()}
+        _LOGGER.info("Devices")
+        _LOGGER.info(self.devices)
+        return self.devices
+
+    async def _update_device(self, deviceState):
+        address = deviceState["address"]
+        if address in self.devices:
+            await self.devices[address].new_state(deviceState["state"], deviceState["dim"])
+
+    @property
+    def keepalive_interval(self):
+        if self.mesh.pollonWrite:
+            return timedelta(seconds=10)
+        else:
+            return timedelta(minutes=10)
+
+    async def keepalive(self):
+        if self.mesh.crypto_key is None:
+            self.mesh.set_crypto_key(await get_cryptokey(**self.credentials))
+        if not self.mesh.connected:
+            if not await self.mesh.connect():
+                return False
+        return await self.mesh.ping()
+
+    async def disconnect(self):
+        _LOGGER.info("DISCONNECT")
+        await self.mesh.disconnect()
+
+    async def poll(self):
+        await self.mesh.poll()
+
+    async def ping(self):
+        return await self.mesh.ping()