administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 10:45:21 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Clean up exception handling

Browse Source
This commit is contained in:
Nick Mowen 2023-06-17 08:47:13 -06:00
parent 4f78e877cc
commit 0f3bc345e2
1 changed files with 27 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
frigate/util.py
View File

@ -1148,14 +1148,18 @@ def to_relative_box(
def get_video_properties(url, get_duration=False): def get_video_properties(url, get_duration=False):
def calculate_duration(video: Optional[any]) -> float: def calculate_duration(video: Optional[any]) -> float:
# Get the frames per second (fps) of the video stream duration = None
fps = video.get(cv2.CAP_PROP_FPS)
total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
if fps and total_frames: if video is not None:
duration = total_frames / fps # Get the frames per second (fps) of the video stream
else: fps = video.get(cv2.CAP_PROP_FPS)
# if cv2 failed need to use ffprobe total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
if fps and total_frames:
duration = total_frames / fps
# if cv2 failed need to use ffprobe
if duration is None:
ffprobe_cmd = [ ffprobe_cmd = [
"ffprobe", "ffprobe",
"-v", "-v",
@ -1167,6 +1171,7 @@ def get_video_properties(url, get_duration=False):
f"{url}", f"{url}",
] ]
p = sp.run(ffprobe_cmd, capture_output=True) p = sp.run(ffprobe_cmd, capture_output=True)
if p.returncode == 0 and p.stdout.decode(): if p.returncode == 0 and p.stdout.decode():
duration = float(p.stdout.decode().strip()) duration = float(p.stdout.decode().strip())
else: else:
@ -1182,26 +1187,25 @@ def get_video_properties(url, get_duration=False):
# Check if the video stream was opened successfully # Check if the video stream was opened successfully
if not video.isOpened(): if not video.isOpened():
logger.debug(f"Error opening video stream {url}.") video = None
raise Exception()
except Exception: except Exception:
if get_duration: video = None
return {"duration": calculate_duration({})}
return {} result = {}
# Get the width of frames in the video stream
width = video.get(cv2.CAP_PROP_FRAME_WIDTH)
# Get the height of frames in the video stream
height = video.get(cv2.CAP_PROP_FRAME_HEIGHT)
# Release the video stream
video.release()
result = {"width": round(width), "height": round(height)}
if get_duration: if get_duration:
result["duration"] = calculate_duration(video) result["duration"] = calculate_duration(video)
if video is not None:
# Get the width of frames in the video stream
width = video.get(cv2.CAP_PROP_FRAME_WIDTH)
# Get the height of frames in the video stream
height = video.get(cv2.CAP_PROP_FRAME_HEIGHT)
# Release the video stream
video.release()
result = {"width": round(width), "height": round(height)}
return result return result