Update .gitignore

[kiosk.git] / camera_trigger.py

diff --git a/camera_trigger.py b/camera_trigger.py
index 74829e32d8b54cd9612e3b9259f515f596be848d..b47a26ed4dfd9023ff45de49503808a757b7be4a 100644 (file)
--- a/camera_trigger.py
+++ b/camera_trigger.py
@@ -1,31 +1,34 @@
+#!/usr/bin/env python3
+
+from datetime import datetime
 import glob
 import os
 import time
+from typing import List, Tuple, Optional
+
 import trigger
 import utils
-from datetime import datetime
+
 
 class any_camera_trigger(trigger.trigger):
     def __init__(self):
         self.triggers_in_the_past_seven_min = {
-            "driveway" :       0,
-            "frontdoor" :      0,
-            "cabin_driveway" : 0,
-            "backyard" :       0,
+            "driveway": 0,
+            "frontdoor": 0,
+            "cabin_driveway": 0,
         }
-        self.last_trigger = {
-            "driveway" :       0,
-            "frontdoor" :      0,
-            "cabin_driveway" : 0,
-            "backyard" :       0,
+        self.last_trigger_timestamp = {
+            "driveway": 0,
+            "frontdoor": 0,
+            "cabin_driveway": 0,
         }
 
-    def choose_priority(self, camera, age):
+    def choose_priority(self, camera: str, age: int) -> int:
+        """Based on the camera name and last trigger age, compute priority."""
         base_priority_by_camera = {
-            "driveway" : 1,
-            "frontdoor" : 2,
-            "cabin_driveway" : 1,
-            "backyard" : 0,
+            "driveway": 1,
+            "frontdoor": 2,
+            "cabin_driveway": 1,
         }
         priority = base_priority_by_camera[camera]
         if age < 10:
@@ -36,50 +39,57 @@ class any_camera_trigger(trigger.trigger):
             priority += trigger.trigger.PRIORITY_LOW
         return priority
 
-    def get_triggered_page_list(self):
+    def get_triggered_page_list(self) -> Optional[List[Tuple[str, int]]]:
+        """Return a list of triggered pages with priorities."""
         triggers = []
-        cameras_with_recent_triggers = 0
-        camera_list = [ "driveway",
-                        "frontdoor",
-                        "cabin_driveway",
-                        "backyard" ]
+        num_cameras_with_recent_triggers = 0
+        camera_list = ["driveway", "frontdoor", "cabin_driveway"]
 
         now = time.time()
         try:
-            # First pass, just see whether each camera is triggered and,
-            # if so, count how many times in the past 7m it has triggered.
+            # First pass, just see whether each camera is triggered
+            # and, if so, count how many times in the past 7m it has
+            # been triggered.
             for camera in camera_list:
-                file = "/timestamps/last_camera_motion_%s" % camera
-                ts = os.stat(file).st_ctime
-                age = now - ts
-                # print "Camera: %s, age %s" % (camera, age)
-                if age < 60:
-                    cameras_with_recent_triggers += 1
-                    time_since_last_trigger = now - self.last_trigger[camera]
-                    self.last_trigger[camera] = now
-                    if time_since_last_trigger < (60 * 7):
-                        self.triggers_in_the_past_seven_min[camera] += 1
-                    else:
-                        self.triggers_in_the_past_seven_min[camera] = 1
+                filename = f"/timestamps/last_camera_motion_{camera}"
+                ts = os.stat(filename).st_ctime
+                if ts != self.last_trigger_timestamp[camera] and (now - ts) < 10:
+                    print("Camera: %s, age %s" % (camera, (now - ts)))
+                    self.last_trigger_timestamp[camera] = ts
+                    num_cameras_with_recent_triggers += 1
+                    self.triggers_in_the_past_seven_min[camera] = 0
+                    filename = f"/timestamps/camera_motion_history_{camera}"
+                    with open(filename, "r") as f:
+                        contents = f.readlines()
+                    for x in contents:
+                        x.strip()
+                        age = now - int(x)
+                        if age < (60 * 7):
+                            self.triggers_in_the_past_seven_min[camera] += 1
 
             # Second pass, see whether we want to trigger due to
             # camera activity we found.  All cameras timestamps were
             # just considered and should be up-to-date.  Some logic to
-            # squelch spammy cameras unless more than one is
-            # triggered at the same time.
+            # squelch spammy cameras unless more than one is triggered
+            # at the same time.
             for camera in camera_list:
-                if self.last_trigger[camera] == now:
-                    ts = utils.timestamp()
-                    if (self.triggers_in_the_past_seven_min[camera] <= 4 or
-                        cameras_with_recent_triggers > 1):
-                        p = self.choose_priority(camera, age)
-                        print(("%s: ****** %s[%d] CAMERA TRIGGER ******" % (
-                            ts, camera, p)))
-                        triggers.append( ( "hidden/%s.html" % camera,
-                                           self.choose_priority(camera, age)) )
+                if (now - self.last_trigger_timestamp[camera]) < 10:
+                    if (
+                        self.triggers_in_the_past_seven_min[camera] <= 4
+                        or num_cameras_with_recent_triggers > 1
+                    ):
+                        priority = self.choose_priority(camera, int(age))
+                        print(
+                            f"{utils.timestamp()}: *** {camera}[{priority}] CAMERA TRIGGER ***"
+                        )
+                        triggers.append(
+                            (
+                                f"hidden/{camera}.html",
+                                priority,
+                            )
+                        )
                     else:
-                        print(("%s: Camera %s too spammy, squelching it" % (
-                            ts, camera)))
+                        print(f"{utils.timestamp()}: Camera {camera} too spammy, squelching it")
         except Exception as e:
             print(e)
             pass
@@ -89,5 +99,6 @@ class any_camera_trigger(trigger.trigger):
         else:
             return triggers
 
-#x = any_camera_trigger()
-#print(x.get_triggered_page_list())
+
+# x = any_camera_trigger()
+# print(x.get_triggered_page_list())