1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
#!/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 parse_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(parse_addr(config['esp32cam_web_addr']))
self.directory = tempfile.gettempdir()
self.nextpic = 1
self.first = True
self.server_addr = parse_addr(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('esp32cam_capture_diff_node')
loop = asyncio.get_event_loop()
ESP32CamCaptureDiffNode()
try:
loop.run_forever()
except KeyboardInterrupt:
pass
|
