ffmpeg_worker refacoring

lib/render_workers/ffmpeg_worker.py

@@ -1,27 +1,9 @@
 #!/usr/bin/env python3
 import re
-import time
 from .render_worker import *
 
 
-class FFMPEGRenderWorker(BaseRenderWorker):
-
-    renderer = 'ffmpeg'
-    render_engine = 'ffmpeg'
-
-    def __init__(self, input_path, output_path, args=None):
-        super(FFMPEGRenderWorker, self).__init__(input_path=input_path, output_path=output_path, ignore_extensions=True,
-                                                 args=args)
-
-        stream_info = subprocess.check_output([self.renderer_path(), "-i",  # https://stackoverflow.com/a/61604105
-                                                     input_path, "-map", "0:v:0", "-c", "copy", "-f", "null", "-y",
-                                                     "/dev/null"], stderr=subprocess.STDOUT).decode('utf-8')
-        found_frames = re.findall('frame=\s*(\d+)', stream_info)
-        self.total_frames = found_frames[-1] if found_frames else '-1'
-        self.frame = 0
-
-        # Stats
-        self.current_frame = -1
+class FFMPEG(BaseRenderEngine):
 
     @classmethod
     def version(cls):
@@ -35,18 +17,58 @@ class FFMPEGRenderWorker(BaseRenderWorker):
             logger.error("Failed to get FFMPEG version: {}".format(e))
         return version
 
+    @classmethod
+    def get_encoders(cls):
+        encoders_raw = subprocess.check_output([cls.renderer_path(), '-encoders'], stderr=subprocess.DEVNULL).decode('utf-8')
+        pattern = '(?P<type>[VASFXBD.]{6})\s+(?P<name>\S{2,})\s+(?P<description>.*)'
+        encoders = [m.groupdict() for m in re.finditer(pattern, encoders_raw)]
+        return encoders
+
+    @classmethod
+    def get_formats(cls):
+        formats_raw = subprocess.check_output([cls.renderer_path(), '-formats'], stderr=subprocess.DEVNULL).decode('utf-8')
+        pattern = '(?P<type>[DE]{1,2})\s+(?P<name>\S{2,})\s+(?P<description>.*)'
+        formats = [m.groupdict() for m in re.finditer(pattern, formats_raw)]
+        return formats
+
+    @classmethod
+    def full_report(cls):
+        return {'version': cls.version(),
+                'help_text': cls.get_help(),
+                'encoders': cls.get_encoders(),
+                'formats': cls.get_formats()}
+
+
+class FFMPEGRenderWorker(BaseRenderWorker):
+
+    engine = FFMPEG
+
+    def __init__(self, input_path, output_path, args=None):
+        super(FFMPEGRenderWorker, self).__init__(input_path=input_path, output_path=output_path, ignore_extensions=True,
+                                                 args=args)
+
+        stream_info = subprocess.check_output([self.engine.renderer_path(), "-i",  # https://stackoverflow.com/a/61604105
+                                                     input_path, "-map", "0:v:0", "-c", "copy", "-f", "null", "-y",
+                                                     "/dev/null"], stderr=subprocess.DEVNULL).decode('utf-8')
+        found_frames = re.findall('frame=\s*(\d+)', stream_info)
+        self.total_frames = found_frames[-1] if found_frames else '-1'
+        self.frame = 0
+
+        # Stats
+        self.current_frame = -1
+
     def generate_worker_subprocess(self):
 
-        cmd = [self.renderer_path(), '-y', '-stats', '-i', self.input_path]
+        cmd = [self.engine.renderer_path(), '-y', '-stats', '-i', self.input_path]
 
         # Resize frame
         if self.args.get('x_resolution', None) and self.args.get('y_resolution', None):
             cmd.extend(['-vf', f"scale={self.args['x_resolution']}:{self.args['y_resolution']}"])
 
         # Convert raw args from string if available
-        raw_args = self.get_raw_args()
+        raw_args = self.args.get('raw', None)
         if raw_args:
-            cmd.extend(raw_args)
+            cmd.extend(raw_args.split(' '))
 
         # Close with output path
         cmd.append(self.output_path)
@@ -67,12 +89,15 @@ class FFMPEGRenderWorker(BaseRenderWorker):
 
 
 if __name__ == '__main__':
-    logging.basicConfig(format='%(asctime)s - %(message)s', datefmt='%d-%b-%y %H:%M:%S', level=logging.DEBUG)
 
-    test_movie = '/Users/brettwilliams/Desktop/dark_knight_rises.mp4'
+    print(FFMPEG.full_report())
 
-    r = FFMPEGRenderWorker(test_movie, '/Users/brettwilliams/Desktop/test-ffmpeg.mp4', args=['-c:v', 'libx265', '-vtag', 'hvc1'])
-    # r = FFMPEGRenderer(test_movie, '/Users/brettwilliams/Desktop/dark_knight_rises-output.mp4')
-    r.start()
-    while r.is_running():
-        time.sleep(1)
+    # logging.basicConfig(format='%(asctime)s - %(message)s', datefmt='%d-%b-%y %H:%M:%S', level=logging.DEBUG)
+    #
+    # test_movie = '/Users/brettwilliams/Desktop/dark_knight_rises.mp4'
+    #
+    # r = FFMPEGRenderWorker(test_movie, '/Users/brettwilliams/Desktop/test-ffmpeg.mp4', args=['-c:v', 'libx265', '-vtag', 'hvc1'])
+    # # r = FFMPEGRenderer(test_movie, '/Users/brettwilliams/Desktop/dark_knight_rises-output.mp4')
+    # r.start()
+    # while r.is_running():
+    #     time.sleep(1)