From 2b9d3bded2fb9b2af2a589c186760f884fdcb752 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Wed, 24 Aug 2022 15:00:43 +0200
Subject: [PATCH] created nuke module

---
 openpype/hosts/nuke/__init__.py | 47 +++++-----------------------
 openpype/hosts/nuke/module.py   | 54 +++++++++++++++++++++++++++++++++
 2 files changed, 62 insertions(+), 39 deletions(-)
 create mode 100644 openpype/hosts/nuke/module.py

diff --git a/openpype/hosts/nuke/__init__.py b/openpype/hosts/nuke/__init__.py
index 134a6621c4..718307583e 100644
--- a/openpype/hosts/nuke/__init__.py
+++ b/openpype/hosts/nuke/__init__.py
@@ -1,41 +1,10 @@
-import os
-import platform
+from .module import (
+    NUKE_ROOT_DIR,
+    NukeModule,
+)
 
 
-def add_implementation_envs(env, _app):
-    # Add requirements to NUKE_PATH
-    pype_root = os.environ["OPENPYPE_REPOS_ROOT"]
-    new_nuke_paths = [
-        os.path.join(pype_root, "openpype", "hosts", "nuke", "startup")
-    ]
-    old_nuke_path = env.get("NUKE_PATH") or ""
-    for path in old_nuke_path.split(os.pathsep):
-        if not path:
-            continue
-
-        norm_path = os.path.normpath(path)
-        if norm_path not in new_nuke_paths:
-            new_nuke_paths.append(norm_path)
-
-    env["NUKE_PATH"] = os.pathsep.join(new_nuke_paths)
-    env.pop("QT_AUTO_SCREEN_SCALE_FACTOR", None)
-
-    # Try to add QuickTime to PATH
-    quick_time_path = "C:/Program Files (x86)/QuickTime/QTSystem"
-    if platform.system() == "windows" and os.path.exists(quick_time_path):
-        path_value = env.get("PATH") or ""
-        path_paths = [
-            path
-            for path in path_value.split(os.pathsep)
-            if path
-        ]
-        path_paths.append(quick_time_path)
-        env["PATH"] = os.pathsep.join(path_paths)
-
-    # Set default values if are not already set via settings
-    defaults = {
-        "LOGLEVEL": "DEBUG"
-    }
-    for key, value in defaults.items():
-        if not env.get(key):
-            env[key] = value
+__all__ = (
+    "NUKE_ROOT_DIR",
+    "NukeModule",
+)
diff --git a/openpype/hosts/nuke/module.py b/openpype/hosts/nuke/module.py
new file mode 100644
index 0000000000..a50444f817
--- /dev/null
+++ b/openpype/hosts/nuke/module.py
@@ -0,0 +1,54 @@
+import os
+import platform
+from openpype.modules import OpenPypeModule
+from openpype.modules.interfaces import IHostModule
+
+NUKE_ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
+
+
+class NukeModule(OpenPypeModule, IHostModule):
+    name = "nuke"
+    host_name = "nuke"
+
+    def initialize(self, module_settings):
+        self.enabled = True
+
+    def add_implementation_envs(self, env, _app):
+        # Add requirements to NUKE_PATH
+        new_nuke_paths = [
+            os.path.join(NUKE_ROOT_DIR, "startup")
+        ]
+        old_nuke_path = env.get("NUKE_PATH") or ""
+        for path in old_nuke_path.split(os.pathsep):
+            if not path:
+                continue
+
+            norm_path = os.path.normpath(path)
+            if norm_path not in new_nuke_paths:
+                new_nuke_paths.append(norm_path)
+
+        env["NUKE_PATH"] = os.pathsep.join(new_nuke_paths)
+        env.pop("QT_AUTO_SCREEN_SCALE_FACTOR", None)
+
+        # Set default values if are not already set via settings
+        defaults = {
+            "LOGLEVEL": "DEBUG"
+        }
+        for key, value in defaults.items():
+            if not env.get(key):
+                env[key] = value
+
+        # Try to add QuickTime to PATH
+        quick_time_path = "C:/Program Files (x86)/QuickTime/QTSystem"
+        if platform.system() == "windows" and os.path.exists(quick_time_path):
+            path_value = env.get("PATH") or ""
+            path_paths = [
+                path
+                for path in path_value.split(os.pathsep)
+                if path
+            ]
+            path_paths.append(quick_time_path)
+            env["PATH"] = os.pathsep.join(path_paths)
+
+    def get_workfile_extensions(self):
+        return [".nk"]