Video rot warning

hovnatan · hovnatan · commit 51e833f245a0 · 2021-03-15T11:30:22.000+04:00
diff --git a/superannotate/db/projects.py b/superannotate/db/projects.py
@@ -251,9 +251,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
`@@ -251,9 +251,12 @@ def upload_video_to_project(`
`251`	`251`	`"Frame rotation of %s found. Output images will be rotated accordingly.",`
`252`	`252`	`rot`
`253`	`253`	`)`
`254`		`- except:`
	`254`	`+ except Exception as e:`
	`255`	`+ warning_str = ""`
	`256`	`+ if "ffprobe" in str(e):`
	`257`	`+ warning_str = "This could be because ffmpeg package is not installed. To install it, run: sudo apt install ffmpeg"`
`255`	`258`	`logger.warning(`
`256`		`- "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"`
	`259`	`+ "Couldn't read video metadata to determine rotation. " + warning_str`
`257`	`260`	`)`
`258`	`261`
`259`	`262`	`video = cv2.VideoCapture(str(video_path), cv2.CAP_FFMPEG)`