Merge pull request #458 from ynput/feature/AY-742_copy-to-breakdown-project

Nuke: Push to project - AY-742

client/ayon_core/tools/context_dialog/__init__.py

@@ -1,7 +1,8 @@
-from .window import ContextDialog, main
+from .window import ContextDialog, main, ask_for_context
 
 
 __all__ = (
     "ContextDialog",
     "main",
+    "ask_for_context"
 )

client/ayon_core/tools/context_dialog/window.py

@@ -791,3 +791,12 @@ def main(
     window.show()
     app.exec_()
     controller.store_output()
+
+
+def ask_for_context(strict=True):
+    controller = ContextDialogController()
+    controller.set_strict(strict)
+    window = ContextDialog(controller=controller)
+    window.exec_()
+
+    return controller.get_selected_context()

server_addon/nuke/client/ayon_nuke/api/pipeline.py

@@ -37,8 +37,6 @@ from .lib import (
     INSTANCE_DATA_KNOB,
     get_main_window,
     WorkfileSettings,
-    # TODO: remove this once workfile builder will be removed
-    process_workfile_builder,
     start_workfile_template_builder,
     launch_workfiles_app,
     check_inventory_versions,
@@ -67,6 +65,7 @@ from .workio import (
     current_file
 )
 from .constants import ASSIST
+from . import push_to_project
 
 log = Logger.get_logger(__name__)
 
@@ -159,9 +158,6 @@ def add_nuke_callbacks():
     # template builder callbacks
     nuke.addOnCreate(start_workfile_template_builder, nodeClass="Root")
 
-    # TODO: remove this callback once workfile builder will be removed
-    nuke.addOnCreate(process_workfile_builder, nodeClass="Root")
-
     # fix ffmpeg settings on script
     nuke.addOnScriptLoad(on_script_load)
 
@@ -332,6 +328,11 @@ def _install_menu():
             lambda: update_placeholder()
         )
 
+    menu.addCommand(
+        "Push to Project",
+        lambda: push_to_project.main()
+    )
+
     menu.addSeparator()
     menu.addCommand(
         "Experimental tools...",

server_addon/nuke/client/ayon_nuke/api/push_to_project.py

@@ -0,0 +1,118 @@
+from collections import defaultdict
+import shutil
+import os
+
+from ayon_api import get_project, get_folder_by_id, get_task_by_id
+from ayon_core.settings import get_project_settings
+from ayon_core.pipeline import Anatomy, registered_host
+from ayon_core.pipeline.template_data import get_template_data
+from ayon_core.pipeline.workfile import get_workdir_with_workdir_data
+from ayon_core.tools import context_dialog
+
+from .utils import bake_gizmos_recursively
+from .lib import MENU_LABEL
+
+import nuke
+
+
+def bake_container(container):
+    """Bake containers to read nodes."""
+
+    node = container["node"]
+
+    # Fetch knobs to remove in order.
+    knobs_to_remove = []
+    remove = False
+    for count in range(0, node.numKnobs()):
+        knob = node.knob(count)
+
+        # All knobs from "AYON" tab knob onwards.
+        if knob.name() == MENU_LABEL:
+            remove = True
+
+        if remove:
+            knobs_to_remove.append(knob)
+
+        # Dont remove knobs from "containerId" onwards.
+        if knob.name() == "containerId":
+            remove = False
+
+    # Knobs needs to be remove in reverse order, because child knobs needs to
+    # be remove first.
+    for knob in reversed(knobs_to_remove):
+        node.removeKnob(knob)
+
+    node["tile_color"].setValue(0)
+
+
+def main():
+    context = context_dialog.ask_for_context()
+
+    if context is None:
+        return
+
+    # Get workfile path to save to.
+    project_name = context["project_name"]
+    project = get_project(project_name)
+    folder = get_folder_by_id(project_name, context["folder_id"])
+    task = get_task_by_id(project_name, context["task_id"])
+    host = registered_host()
+    project_settings = get_project_settings(project_name)
+    anatomy = Anatomy(project_name)
+
+    workdir_data = get_template_data(
+        project, folder, task, host.name, project_settings
+    )
+
+    workdir = get_workdir_with_workdir_data(
+        workdir_data,
+        project_name,
+        anatomy,
+        project_settings=project_settings
+    )
+    # Save current workfile.
+    current_file = host.current_file()
+    host.save_file(current_file)
+
+    for container in host.ls():
+        bake_container(container)
+
+    # Bake gizmos.
+    bake_gizmos_recursively()
+
+    # Copy all read node files to "resources" folder next to workfile and
+    # change file path.
+    first_frame = int(nuke.root()["first_frame"].value())
+    last_frame = int(nuke.root()["last_frame"].value())
+    files_by_node_name = defaultdict(set)
+    nodes_by_name = {}
+    for count in range(first_frame, last_frame + 1):
+        nuke.frame(count)
+        for node in nuke.allNodes(filter="Read"):
+            files_by_node_name[node.name()].add(
+                nuke.filename(node, nuke.REPLACE)
+            )
+            nodes_by_name[node.name()] = node
+
+    resources_dir = os.path.join(workdir, "resources")
+    for name, files in files_by_node_name.items():
+        dir = os.path.join(resources_dir, name)
+        if not os.path.exists(dir):
+            os.makedirs(dir)
+
+        for f in files:
+            shutil.copy(f, os.path.join(dir, os.path.basename(f)))
+
+        node = nodes_by_name[name]
+        path = node["file"].value().replace(os.path.dirname(f), dir)
+        node["file"].setValue(path.replace("\\", "/"))
+
+    # Save current workfile to new context.
+    pushed_workfile = os.path.join(
+        workdir, os.path.basename(current_file))
+    host.save_file(pushed_workfile)
+
+    # Open current context workfile.
+    host.open_file(current_file)
+
+    nuke.message(f"Pushed to project: \n{pushed_workfile}")