extract metadata

gbstack · gbstack · commit bfaec2a6bcb5 · 2019-11-05T09:07:04.000+08:00
diff --git a/ffprobe3/ffprobe.py b/ffprobe3/ffprobe.py
@@ -54,9 +54,25 @@ def __init__(self, path_to_video):
                 elif stream:
                     data_lines.append(line)
 
+            self.metadata = {}
+            is_metadata = False
+            stream_metadata_met = False
+
             for line in iter(p.stderr.readline, b''):
                 line = line.decode('UTF-8')
 
+                if 'Metadata:' in line and not stream_metadata_met:
+                    is_metadata = True
+                elif 'Stream #' in line:
+                    is_metadata = False
+                    stream_metadata_met = True
+                elif is_metadata:
+                    splits = line.split(',')
+                    for s in splits:
+                        m = re.search(r'(\w+)\s*:\s*(.*)$', s)
+                        # print(m.groups())
+                        self.metadata[m.groups()[0]] = m.groups()[1].strip()
+
                 if '[STREAM]' in line:
                     stream = True
                     data_lines = []
