From 1c164ad5804af23f519255e3e9677602e21976cb Mon Sep 17 00:00:00 2001
From: Victor W Allen <vwallen@antlab.net>
Date: Wed, 5 May 2021 15:35:48 -0700
Subject: [PATCH] Adds interval image capture to samples

---
 sample-code/interval_image_capture.py | 78 +++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100644 sample-code/interval_image_capture.py

diff --git a/sample-code/interval_image_capture.py b/sample-code/interval_image_capture.py
new file mode 100644
index 0000000..088b0de
--- /dev/null
+++ b/sample-code/interval_image_capture.py
@@ -0,0 +1,78 @@
+#!/usr/bin/env python3
+
+# Standard library modules
+import argparse
+import io
+import os
+import time
+import pathlib
+from datetime import datetime
+
+# 3rd party modules that
+# may require installation
+import picamera
+from PIL import Image
+
+
+def main(interval, save_path):
+
+    # Confirm that the provide save path is a valid path
+    try:
+        path = pathlib.Path(save_path)
+        if not path.exists():
+            raise FileNotFoundError("not a valid path")
+
+        if not path.is_dir():
+            raise NotADirectoryError("path is not a directory")
+
+    except Exception as e:
+        print(f"Unable to save to {save_path}, {e}")
+        exit()
+
+    with picamera.PiCamera(resolution=(224, 224), framerate=30) as camera:
+
+        stream = io.BytesIO()
+        camera.start_preview()
+
+        # Camera warm-up time
+        time.sleep(2)
+
+        while True:
+            camera.annotate_text = "Ready..."
+            stream.seek(0)
+
+            # Use current time stamp for image file name
+            # and append to the save location's path
+            time_stamp = datetime.now().strftime('%Y-%m-%d_%H:%M:%S')
+            save_filename = path.joinpath(f"{time_stamp}.jpg")
+
+            camera.annotate_text = None
+            camera.capture(stream, format='jpeg')
+            img = Image.open(stream)
+            img.save(save_filename)
+
+            # Update display with last timestamp
+            # and print save confirmation
+            camera.annotate_text = f"Saved at:\n{time_stamp}"
+            print(f"Saved image to: {save_filename}")
+
+            time.sleep(interval)
+
+
+if __name__ == '__main__':
+    try:
+        # Accept arguments on the command line for the
+        # interval between image captures, and the path
+        # to save captured images to. Default to saving
+        # in the current directory every 10 seconds
+        parser = argparse.ArgumentParser()
+        parser.add_argument("path", nargs="?", help="Path to image save location", default=os.getcwd())
+        parser.add_argument("-i", "--interval", required=False, help="Seconds between image capture", default=10, type=int)
+        args = parser.parse_args()
+
+        print(f"Capture starting, to stop press \"CTRL+C\"")
+        main(args.interval, args.path)
+
+    except KeyboardInterrupt:
+        print("")
+        print(f"Caught interrupt, exiting...")