Merge branch 'master' of ch1p.io:homekit

1 files changed, 87 insertions, 0 deletions

diff --git a/bin/esp32cam_capture_diff_node.py b/bin/esp32cam_capture_diff_node.py
new file mode 100755
index 0000000..d664c6d
--- /dev/null
+++ b/bin/esp32cam_capture_diff_node.py
@@ -0,0 +1,87 @@
+#!/usr/bin/env python3
+import asyncio
+import logging
+import os.path
+import tempfile
+import __py_include
+import homekit.telegram.aio as telegram
+
+from homekit.config import config
+from homekit.camera.esp32 import WebClient
+from homekit.util import Addr, send_datagram, stringify
+from apscheduler.schedulers.asyncio import AsyncIOScheduler
+from typing import Optional
+
+logger = logging.getLogger(__name__)
+cam: Optional[WebClient] = None
+
+
+async def pyssim(fn1: str, fn2: str) -> float:
+    args = [config['pyssim']['bin']]
+    if 'width' in config['pyssim']:
+        args.extend(['--width', str(config['pyssim']['width'])])
+    if 'height' in config['pyssim']:
+        args.extend(['--height', str(config['pyssim']['height'])])
+    args.extend([fn1, fn2])
+    proc = await asyncio.create_subprocess_exec(*args,
+                                                stdout=asyncio.subprocess.PIPE,
+                                                stderr=asyncio.subprocess.PIPE)
+    stdout, stderr = await proc.communicate()
+    if proc.returncode != 0:
+        logger.error(f'pyssim({fn1}, {fn2}): pyssim returned {proc.returncode}, stderr: {stderr.decode().strip()}')
+
+    return float(stdout.decode().strip())
+
+
+class ESP32CamCaptureDiffNode:
+    def __init__(self):
+        self.client = WebClient(Addr.fromstring(config['esp32cam_web_addr']))
+        self.directory = tempfile.gettempdir()
+        self.nextpic = 1
+        self.first = True
+        self.server_addr = Addr.fromstring(config['node']['server_addr'])
+
+        self.scheduler = AsyncIOScheduler()
+        self.scheduler.add_job(self.capture, 'interval', seconds=config['node']['interval'])
+        self.scheduler.start()
+
+    async def capture(self):
+        logger.debug('capture: start')
+
+        filename = self.getfilename()
+        if not await self.client.capture(os.path.join(self.directory, filename)):
+            logger.error('failed to capture')
+            return
+
+        self.nextpic = 1 if self.nextpic == 2 else 2
+        if not self.first:
+            second_filename = os.path.join(self.directory, self.getfilename())
+            score = await pyssim(filename, second_filename)
+            logger.debug(f'pyssim: score={score}')
+            if score < config['pyssim']['threshold']:
+                logger.info(f'score = {score}, informing central server')
+                send_datagram(stringify([config['node']['name'], 2]), self.server_addr)
+
+                # send to telegram
+                if 'telegram' in config:
+                    await telegram.send_message(f'pyssim: score={score}')
+                    await telegram.send_photo(filename)
+                    await telegram.send_photo(second_filename)
+
+        self.first = False
+
+        logger.debug('capture: done')
+
+    def getfilename(self):
+        return os.path.join(self.directory, f'{self.nextpic}.jpg')
+
+
+if __name__ == '__main__':
+    config.load_app('esp32cam_capture_diff_node')
+
+    loop = asyncio.get_event_loop()
+    ESP32CamCaptureDiffNode()
+    try:
+        loop.run_forever()
+    except KeyboardInterrupt:
+        pass