From 6a1979b6b2b852be227ff0b254cc27797ba8b3f5 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Wed, 24 Aug 2022 18:09:28 +0200
Subject: [PATCH 1/3] created aftereffects module

---
 openpype/hosts/aftereffects/__init__.py | 15 ++++++---------
 openpype/hosts/aftereffects/module.py   | 24 ++++++++++++++++++++++++
 2 files changed, 30 insertions(+), 9 deletions(-)
 create mode 100644 openpype/hosts/aftereffects/module.py

diff --git a/openpype/hosts/aftereffects/__init__.py b/openpype/hosts/aftereffects/__init__.py
index deae48d122..c9ad6aaeeb 100644
--- a/openpype/hosts/aftereffects/__init__.py
+++ b/openpype/hosts/aftereffects/__init__.py
@@ -1,9 +1,6 @@
-def add_implementation_envs(env, _app):
-    """Modify environments to contain all required for implementation."""
-    defaults = {
-        "OPENPYPE_LOG_NO_COLORS": "True",
-        "WEBSOCKET_URL": "ws://localhost:8097/ws/"
-    }
-    for key, value in defaults.items():
-        if not env.get(key):
-            env[key] = value
+from .module import AfterEffectsModule
+
+
+__all__ = (
+    "AfterEffectsModule",
+)
diff --git a/openpype/hosts/aftereffects/module.py b/openpype/hosts/aftereffects/module.py
new file mode 100644
index 0000000000..33e42b451b
--- /dev/null
+++ b/openpype/hosts/aftereffects/module.py
@@ -0,0 +1,24 @@
+import os
+from openpype.modules import OpenPypeModule
+from openpype.modules.interfaces import IHostModule
+
+
+class AfterEffectsModule(OpenPypeModule, IHostModule):
+    name = "aftereffects"
+    host_name = "aftereffects"
+
+    def initialize(self, module_settings):
+        self.enabled = True
+
+    def add_implementation_envs(self, env, _app):
+        """Modify environments to contain all required for implementation."""
+        defaults = {
+            "OPENPYPE_LOG_NO_COLORS": "True",
+            "WEBSOCKET_URL": "ws://localhost:8097/ws/"
+        }
+        for key, value in defaults.items():
+            if not env.get(key):
+                env[key] = value
+
+    def get_workfile_extensions(self):
+        return [".aep"]

From 7e1ba966ce2188e5a492255bae7f9216eaac8833 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Wed, 24 Aug 2022 18:09:38 +0200
Subject: [PATCH 2/3] workio is not using HOST_WORKFILE_EXTENSIONS

---
 openpype/hosts/aftereffects/api/workio.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/openpype/hosts/aftereffects/api/workio.py b/openpype/hosts/aftereffects/api/workio.py
index d6c732285a..18b40af5dc 100644
--- a/openpype/hosts/aftereffects/api/workio.py
+++ b/openpype/hosts/aftereffects/api/workio.py
@@ -1,12 +1,11 @@
 """Host API required Work Files tool"""
 import os
 
-from openpype.pipeline import HOST_WORKFILE_EXTENSIONS
 from .launch_logic import get_stub
 
 
 def file_extensions():
-    return HOST_WORKFILE_EXTENSIONS["aftereffects"]
+    return [".aep"]
 
 
 def has_unsaved_changes():

From 18f22f4d0fa522c50bba4797471ebd5ab4446e43 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Wed, 24 Aug 2022 18:12:49 +0200
Subject: [PATCH 3/3] removed unused import

---
 openpype/hosts/aftereffects/module.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/openpype/hosts/aftereffects/module.py b/openpype/hosts/aftereffects/module.py
index 33e42b451b..93d575c186 100644
--- a/openpype/hosts/aftereffects/module.py
+++ b/openpype/hosts/aftereffects/module.py
@@ -1,4 +1,3 @@
-import os
 from openpype.modules import OpenPypeModule
 from openpype.modules.interfaces import IHostModule