Move get_current_comp and comp_lock_and_undo_chunk to lib

+ fix import of host tools

openpype/hosts/fusion/api/__init__.py

@@ -5,10 +5,7 @@ from .pipeline import (
     ls,
 
     imprint_container,
-    parse_container,
-
-    get_current_comp,
-    comp_lock_and_undo_chunk
+    parse_container
 )
 
 from .workio import (
@@ -24,7 +21,9 @@ from .lib import (
     maintained_selection,
     get_additional_data,
     update_frame_range,
-    set_asset_framerange
+    set_asset_framerange,
+    get_current_comp,
+    comp_lock_and_undo_chunk
 )
 
 from .menu import launch_openpype_menu
@@ -39,9 +38,6 @@ __all__ = [
     "imprint_container",
     "parse_container",
 
-    "get_current_comp",
-    "comp_lock_and_undo_chunk",
-
     # workio
     "open_file",
     "save_file",
@@ -55,6 +51,8 @@ __all__ = [
     "get_additional_data",
     "update_frame_range",
     "set_asset_framerange",
+    "get_current_comp",
+    "comp_lock_and_undo_chunk",
 
     # menu
     "launch_openpype_menu",

openpype/hosts/fusion/api/lib.py

@@ -19,8 +19,6 @@ from openpype.pipeline import (
 )
 from openpype.pipeline.context_tools import get_current_project_asset
 
-from .pipeline import get_current_comp, comp_lock_and_undo_chunk
-
 self = sys.modules[__name__]
 self._project = None
 
@@ -232,3 +230,21 @@ def get_frame_path(path):
         padding = 4  # default Fusion padding
 
     return filename, padding, ext
+
+
+def get_current_comp():
+    """Hack to get current comp in this session"""
+    fusion = getattr(sys.modules["__main__"], "fusion", None)
+    return fusion.CurrentComp if fusion else None
+
+
+@contextlib.contextmanager
+def comp_lock_and_undo_chunk(comp, undo_queue_name="Script CMD"):
+    """Lock comp and open an undo chunk during the context"""
+    try:
+        comp.Lock()
+        comp.StartUndo(undo_queue_name)
+        yield
+    finally:
+        comp.Unlock()
+        comp.EndUndo()

openpype/hosts/fusion/api/pipeline.py

@@ -23,6 +23,12 @@ from openpype.pipeline import (
 )
 from openpype.pipeline.load import any_outdated_containers
 from openpype.hosts.fusion import FUSION_HOST_DIR
+from openpype.tools.utils import host_tools
+
+from .lib import (
+    get_current_comp,
+    comp_lock_and_undo_chunk
+)
 
 log = Logger.get_logger(__name__)
 
@@ -247,19 +253,3 @@ def parse_container(tool):
     return container
 
 
-def get_current_comp():
-    """Hack to get current comp in this session"""
-    fusion = getattr(sys.modules["__main__"], "fusion", None)
-    return fusion.CurrentComp if fusion else None
-
-
-@contextlib.contextmanager
-def comp_lock_and_undo_chunk(comp, undo_queue_name="Script CMD"):
-    """Lock comp and open an undo chunk during the context"""
-    try:
-        comp.Lock()
-        comp.StartUndo(undo_queue_name)
-        yield
-    finally:
-        comp.Unlock()
-        comp.EndUndo()

openpype/hosts/fusion/api/workio.py

@@ -2,7 +2,7 @@
 import sys
 import os
 
-from .pipeline import get_current_comp
+from .lib import get_current_comp
 
 
 def file_extensions():