summaryrefslogtreecommitdiff
path: root/src/esp32cam_capture_diff_node.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/esp32cam_capture_diff_node.py')
-rwxr-xr-xsrc/esp32cam_capture_diff_node.py86
1 files changed, 0 insertions, 86 deletions
diff --git a/src/esp32cam_capture_diff_node.py b/src/esp32cam_capture_diff_node.py
deleted file mode 100755
index 59482f7..0000000
--- a/src/esp32cam_capture_diff_node.py
+++ /dev/null
@@ -1,86 +0,0 @@
-#!/usr/bin/env python3
-import asyncio
-import logging
-import os.path
-import tempfile
-import home.telegram.aio as telegram
-
-from home.config import config
-from home.camera.esp32 import WebClient
-from home.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