Video rot warning

hovnatan · hovnatan · commit 44390ab7f515 · 2021-03-05T16:27:16.000+04:00
diff --git a/superannotate/db/projects.py b/superannotate/db/projects.py
@@ -237,9 +237,12 @@ def upload_video_to_project(
                 "Frame rotation of %s found. Output images will be rotated accordingly.",
                 rot
             )
-    except:
+    except Exception as e:
+        warning_str = ""
+        if "ffprobe" in str(e):
+            warning_str = "This could be because ffmpeg package is not installed. To install it, run: sudo apt install ffmpeg"
         logger.warning(
-            "Couldn't read video metadata to determine rotation. This could be because ffmpeg package is not installed. To install it, run: sudo apt install ffmpeg"
+            "Couldn't read video metadata to determine rotation. " + warning_str
         )
 
     video = cv2.VideoCapture(str(video_path), cv2.CAP_FFMPEG)

Original file line number	Diff line number	Diff line change
`@@ -237,9 +237,12 @@ def upload_video_to_project(`
`237`	`237`	`"Frame rotation of %s found. Output images will be rotated accordingly.",`
`238`	`238`	`rot`
`239`	`239`	`)`
`240`		`- except:`
	`240`	`+ except Exception as e:`
	`241`	`+ warning_str = ""`
	`242`	`+ if "ffprobe" in str(e):`
	`243`	`+ warning_str = "This could be because ffmpeg package is not installed. To install it, run: sudo apt install ffmpeg"`
`241`	`244`	`logger.warning(`
`242`		`- "Couldn't read video metadata to determine rotation. This could be because ffmpeg package is not installed. To install it, run: sudo apt install ffmpeg"`
	`245`	`+ "Couldn't read video metadata to determine rotation. " + warning_str`
`243`	`246`	`)`
`244`	`247`
`245`	`248`	`video = cv2.VideoCapture(str(video_path), cv2.CAP_FFMPEG)`