Extract video time info based on reprostim capture filename format, WiP

#96
ReproNim · Jun 25, 2024 · aa07fda · aa07fda
1 parent 56ad2a6
commit aa07fda
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/Parsing/parse_wQR.py b/Parsing/parse_wQR.py
@@ -3,6 +3,8 @@
 import json
 import logging
 import os
+from datetime import datetime
+
 from pydantic import BaseModel, Field
 from pyzbar.pyzbar import decode, ZBarSymbol
 import cv2
@@ -18,6 +20,12 @@
 logger.debug(f"name={__name__}")
 
 
+# Define class for video time info
+class VideoTimeInfo(BaseModel):
+    start_time: datetime = Field(..., description="Start time of the video")
+    end_time: datetime = Field(..., description="End time of the video")
+
+
 # Define the data model for the QR record
 class QrRecord(BaseModel):
     frame_start: int = Field(..., description="Frame number where QR code starts")