From 3a2d831ca632fdbfc93bf2789bdc7501424d5262 Mon Sep 17 00:00:00 2001
From: Petr Kalis <petr.kalis@gmail.com>
Date: Fri, 4 Apr 2025 17:20:09 +0200
Subject: [PATCH] Added Loader Hooks to switch_container

---
 client/ayon_core/pipeline/load/utils.py       | 27 ++++++++++++++++---
 .../sceneinventory/switch_dialog/dialog.py    |  8 +++++-
 2 files changed, 31 insertions(+), 4 deletions(-)

diff --git a/client/ayon_core/pipeline/load/utils.py b/client/ayon_core/pipeline/load/utils.py
index abef1bc500..94e7ad53a6 100644
--- a/client/ayon_core/pipeline/load/utils.py
+++ b/client/ayon_core/pipeline/load/utils.py
@@ -632,15 +632,22 @@ def update_container(container, version=-1, hooks_by_identifier=None):
     return update_return
 
 
-def switch_container(container, representation, loader_plugin=None):
+def switch_container(
+    container,
+    representation,
+    loader_plugin=None,
+    hooks_by_identifier=None
+):
     """Switch a container to representation
 
     Args:
         container (dict): container information
         representation (dict): representation entity
+        loader_plugin (LoaderPlugin)
+        hooks_by_identifier (dict): {"pre": [PreHookPlugin1], "post":[]}
 
     Returns:
-        function call
+        return from function call
     """
     from ayon_core.pipeline import get_current_project_name
 
@@ -679,7 +686,21 @@ def switch_container(container, representation, loader_plugin=None):
 
     loader = loader_plugin(context)
 
-    return loader.switch(container, context)
+    loader_identifier = get_loader_identifier(loader)
+    hooks = hooks_by_identifier.get(loader_identifier, {})
+    for hook_plugin in hooks.get("pre", []):
+        hook_plugin.switch(
+            context,
+            container
+        )
+    switch_return = loader.switch(container, context)
+    for hook_plugin in hooks.get("post", []):
+        hook_plugin.switch(
+            context,
+            container
+        )
+
+    return switch_return
 
 
 def _fix_representation_context_compatibility(repre_context):
diff --git a/client/ayon_core/tools/sceneinventory/switch_dialog/dialog.py b/client/ayon_core/tools/sceneinventory/switch_dialog/dialog.py
index a6d88ed44a..c878cad079 100644
--- a/client/ayon_core/tools/sceneinventory/switch_dialog/dialog.py
+++ b/client/ayon_core/tools/sceneinventory/switch_dialog/dialog.py
@@ -1339,8 +1339,14 @@ class SwitchAssetDialog(QtWidgets.QDialog):
                 repre_entity = repres_by_name[container_repre_name]
 
         error = None
+        hook_loaders_by_id = self._controller.get_hook_loaders_by_identifier()
         try:
-            switch_container(container, repre_entity, loader)
+            switch_container(
+                container,
+                repre_entity,
+                loader,
+                hook_loaders_by_id
+            )
         except (
             LoaderSwitchNotImplementedError,
             IncompatibleLoaderError,