move houdini shot open post script to a callback (#266)

Author: scottdmilner

pipeline/pipe/h/hipfile/filemanager.py

@@ -40,8 +40,7 @@ def _check_unsaved_changes() -> bool:
                 return False
         return True
 
-    @staticmethod
-    def _open_file(path: Path) -> None:
+    def _open_file(self, path: Path) -> None:
         hou.hipFile.load(str(path), suppress_save_prompt=True)
 
     def _setup_file(self, path: Path, entity: SGEntity) -> None:

pipeline/pipe/h/hipfile/shot.py

@@ -7,9 +7,12 @@
 from typing import cast
 
 import pipe.h
+from pipe.db import DB
 from pipe.glui.dialogs import FilteredListDialog
 from pipe.struct.db import SGEntity, Shot
 
+from env_sg import DB_Config
+
 from .filemanager import HFileManager
 
 
@@ -55,6 +58,26 @@ def _post_open_file(self, entity: SGEntity):
         shot = cast(Shot, entity)
         hou.playbar.setFrameRange(shot.cut_in - 5, shot.cut_out + 5)
         hou.playbar.setPlaybackRange(shot.cut_in - 5, shot.cut_out + 5)
+        hou.setFrame(shot.cut_in)
+
+    def _open_file(self, path):
+        def do_post_open_file(event: hou.hipFileEventType) -> None:
+            if event != hou.hipFileEventType.AfterLoad:
+                return
+            try:
+                shot_code = str(hou.contextOption("SHOT")).split("/").pop()
+                conn = DB.Get(DB_Config)
+                shot = conn.get_shot_by_code(shot_code)
+                self._post_open_file(shot)
+            except Exception:
+                print("Failed to update frame range!")
+
+            hou.hipFile.removeEventCallback(do_post_open_file)
+
+        # hou.hipFile.load interrupts the running of the script so we have to
+        # call _post_open_file with a callback instead
+        hou.hipFile.addEventCallback(do_post_open_file)
+        super()._open_file(path)
 
     def _setup_file(self, path: Path, entity: SGEntity) -> None:
         super(HShotFileManager, HShotFileManager)._setup_file(self, path, entity)

pipeline/pipe/m/shotfile/shotfile_manager.py

@@ -118,8 +118,7 @@ def _generate_filename_ext(self, entity) -> tuple[str, str]:
         shot = cast(Shot, entity)
         return shot.code, "mb"
 
-    @staticmethod
-    def _open_file(path: Path) -> None:
+    def _open_file(self, path: Path) -> None:
         mc.file(str(path), open=True, force=True)
 
     def _post_open_file(self, entity: SGEntity) -> None:

pipeline/pipe/util/filemanager.py

@@ -92,9 +92,8 @@ def _generate_filename_ext(self, entity: SGEntity) -> tuple[str, str]:
     def _get_subpath(self) -> str:
         return ""
 
-    @staticmethod
     @abstractmethod
-    def _open_file(path: Path) -> None:
+    def _open_file(self, path: Path) -> None:
         """Opens the file into the current session"""
         pass