cute box

capture.py

@@ -0,0 +1,33 @@
+import cv2
+from datetime import datetime
+from loguru import logger
+from pathlib import Path
+
+BASE_PATH = Path("dumped/cam")
+
+def gen():
+    API = cv2.CAP_AVFOUNDATION
+    cap = cv2.VideoCapture(0, API)
+    while True:
+        ret, frame = cap.read()
+        if not ret:
+            logger.warning("Failed to grab frame")
+            break
+        yield frame
+
+def main():
+    for frame in gen():
+        cv2.imshow("frame", frame)
+        k = cv2.waitKey(1)
+        if k == ord("q"):
+            break
+        elif k == ord("s"):
+            now = datetime.now()
+            filename = BASE_PATH / f"capture_{now.strftime('%Y%m%d%H%M%S')}.jpg"
+            logger.warning(f"Saving to {filename}")
+            cv2.imwrite(str(filename), frame)
+        else:
+            ...
+
+if __name__ == "__main__":
+    main()