Merge pull request #64 from bEpic-studio/enhancement/close-ayonmenus-when-dvr-is-shutdown

Close AYON Menus when Resolve is shutdown

Author: rdelillo

client/ayon_resolve/api/menu.py

@@ -8,6 +8,8 @@
 from ayon_core.style import load_stylesheet
 from ayon_core.resources import get_ayon_icon_filepath
 
+from .pulse import PulseThread
+
 MENU_LABEL = os.environ["AYON_MENU_LABEL"]
 
 
@@ -143,7 +145,6 @@ def on_set_resolution_clicked(self):
         print("Clicked Set Resolution")
 
 
-
 def launch_ayon_menu():
     app = (
         QtWidgets.QApplication.instance()
@@ -157,4 +158,9 @@ def launch_ayon_menu():
 
     ayon_menu.show()
 
+    # Force close current process if Resolve is closed
+    pulse = PulseThread(parent=ayon_menu)
+    pulse.no_host_response.connect(app.quit)
+    pulse.start()
+
     sys.exit(app.exec_())

client/ayon_resolve/api/pulse.py

@@ -0,0 +1,46 @@
+"""This module is a copy from ayon-resolve"""
+
+import os
+import sys
+
+from qtpy import QtCore
+
+
+class PulseThread(QtCore.QThread):
+    """A Timer that checks whether host app is still alive.
+
+    This checks whether the Resolve process is still active at a certain
+    interval. This is useful due to how Resolve runs its scripts. Each script
+    runs in its own environment and process (a `fusionscript` process each).
+    If Resolve would go down and we have a UI process running at the same time
+    then it can happen that the `fusionscript.exe` will remain running in the
+    background in limbo due to e.g. a Qt interface's QApplication that keeps
+    running infinitely.
+
+    Warning:
+        When the host is not detected this will automatically exit
+        the current process.
+    """
+    no_host_response = QtCore.Signal()
+
+    def __init__(self, parent=None):
+        super(PulseThread, self).__init__(parent=parent)
+
+    def run(self):
+        app = getattr(sys.modules["__main__"], "app", None)
+
+        # Interval in milliseconds
+        interval = int(os.environ.get("AYON_RESOLVE_PULSE_INTERVAL", 1000))
+
+        while True:
+            if self.isInterruptionRequested():
+                return
+
+            # We don't need to call Test because PyRemoteObject of the app
+            # will actually fail to even resolve the Test function if it has
+            # gone down. So we can actually already just check by confirming
+            # the method is still getting resolved. (Optimization)
+            if app.Test is None:
+                self.no_host_response.emit()
+
+            self.msleep(interval)