From 15d1302f182639706d6886fd63e8a5c17946f76b Mon Sep 17 00:00:00 2001
From: Brett Williams <blw1138@mac.com>
Date: Tue, 23 May 2023 15:28:20 -0500
Subject: [PATCH] Refactor AERender to use BaseRenderEngine

---
 lib/render_workers/aerender_worker.py | 37 +++++++++++++++++----------
 1 file changed, 23 insertions(+), 14 deletions(-)

diff --git a/lib/render_workers/aerender_worker.py b/lib/render_workers/aerender_worker.py
index 57f446c..129ef67 100644
--- a/lib/render_workers/aerender_worker.py
+++ b/lib/render_workers/aerender_worker.py
@@ -17,11 +17,32 @@ def aerender_path():
         return paths[0]
 
 
+class AERender(BaseRenderEngine):
+
+    supported_extensions = ['.aep']
+
+    @classmethod
+    def version(cls):
+        version = None
+        try:
+            render_path = cls.renderer_path()
+            if render_path:
+                ver_out = subprocess.check_output([render_path, '-version'])
+                version = ver_out.decode('utf-8').split(" ")[-1].strip()
+        except Exception as e:
+            logging.error(f'Failed to get {cls.name()} version: {e}')
+        return version
+
+    @classmethod
+    def get_formats(cls):
+        # todo: create implementation
+        return []
+
+
 class AERenderWorker(BaseRenderWorker):
 
-    renderer = 'After Effects'
-    render_engine = 'aerender'
     supported_extensions = ['.aep']
+    engine = AERender
 
     def __init__(self, input_path, output_path, args=None):
         super(AERenderWorker, self).__init__(input_path=input_path, output_path=output_path, ignore_extensions=False,
@@ -35,18 +56,6 @@ class AERenderWorker(BaseRenderWorker):
         self.progress_history = []
         self.attributes = {}
 
-    @classmethod
-    def version(cls):
-        version = None
-        try:
-            render_path = cls.renderer_path()
-            if render_path:
-                ver_out = subprocess.check_output([render_path, '-version'])
-                version = ver_out.decode('utf-8').split(" ")[-1].strip()
-        except Exception as e:
-            logging.error(f'Failed to get {cls.renderer} version: {e}')
-        return version
-
     def generate_worker_subprocess(self):
 
         if os.path.exists('nexrender-cli-macos'):