Update utils/utils.py
Browse files- utils/utils.py +18 -1
utils/utils.py
CHANGED
|
@@ -978,4 +978,21 @@ def get_file_manager(base_dir: Optional[str] = None) -> FileManager:
|
|
| 978 |
global _default_file_manager
|
| 979 |
if _default_file_manager is None or base_dir is not None:
|
| 980 |
_default_file_manager = FileManager(base_dir)
|
| 981 |
-
return _default_file_manager
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 978 |
global _default_file_manager
|
| 979 |
if _default_file_manager is None or base_dir is not None:
|
| 980 |
_default_file_manager = FileManager(base_dir)
|
| 981 |
+
return _default_file_manager
|
| 982 |
+
|
| 983 |
+
def validate_video_file(file_path: str) -> bool:
|
| 984 |
+
"""Validate if file is a valid video file."""
|
| 985 |
+
import os
|
| 986 |
+
import cv2
|
| 987 |
+
|
| 988 |
+
if not os.path.exists(file_path):
|
| 989 |
+
return False
|
| 990 |
+
|
| 991 |
+
try:
|
| 992 |
+
cap = cv2.VideoCapture(file_path)
|
| 993 |
+
ret = cap.isOpened()
|
| 994 |
+
cap.release()
|
| 995 |
+
return ret
|
| 996 |
+
except:
|
| 997 |
+
return False
|
| 998 |
+
|