Merged in feature/change-content-on-switch-context (pull request #511)

PYPE-532 switching context will now change content to new asset

Approved-by: Milan Kolar <milan@orbi.tools>

pype/maya/__init__.py

@@ -229,3 +229,10 @@ def on_task_changed(*args):
 
     # Run
     maya.pipeline._on_task_changed()
+    with maya.suspended_refresh():
+        lib.set_context_settings()
+        lib.update_content_on_context_change()
+
+    lib.show_message("Context was changed",
+                     ("Context was changed to {}".format(
+                        avalon.Session["AVALON_ASSET"])))

pype/maya/lib.py

@@ -2585,3 +2585,44 @@ def remove_render_layer_observer():
     except ValueError:
         # no observer set yet
         pass
+
+
+def update_content_on_context_change():
+    """
+    This will update scene content to match new asset on context change
+    """
+    scene_sets = cmds.listSets(allSets=True)
+    new_asset = api.Session["AVALON_ASSET"]
+    new_data = lib.get_asset()["data"]
+    for s in scene_sets:
+        try:
+            if cmds.getAttr("{}.id".format(s)) == "pyblish.avalon.instance":
+                attr = cmds.listAttr(s)
+                print(s)
+                if "asset" in attr:
+                    print("  - setting asset to: [ {} ]".format(new_asset))
+                    cmds.setAttr("{}.asset".format(s),
+                                 new_asset, type="string")
+                if "frameStart" in attr:
+                    cmds.setAttr("{}.frameStart".format(s),
+                                 new_data["frameStart"])
+                if "frameEnd" in attr:
+                    cmds.setAttr("{}.frameEnd".format(s),
+                                 new_data["frameEnd"],)
+        except ValueError:
+            pass
+
+
+def show_message(title, msg):
+    from avalon.vendor.Qt import QtWidgets
+    from ..widgets import message_window
+
+    # Find maya main window
+    top_level_widgets = {w.objectName(): w for w in
+                         QtWidgets.QApplication.topLevelWidgets()}
+
+    parent = top_level_widgets.get("MayaWindow", None)
+    if parent is None:
+        pass
+    else:
+        message_window.message(title=title, message=msg, parent=parent)

pype/widgets/message_window.py

@@ -7,7 +7,7 @@ log = logging.getLogger(__name__)
 
 class Window(QtWidgets.QWidget):
     def __init__(self, parent, title, message, level):
-        super().__init__()
+        super(Window, self).__init__()
         self.parent = parent
         self.title = title
         self.message = message
@@ -48,9 +48,10 @@ class Window(QtWidgets.QWidget):
         return
 
 
-def message(title=None, message=None, level="info"):
-    global app
-    app = QtWidgets.QApplication(sys.argv)
+def message(title=None, message=None, level="info", parent=None):
+    app = parent
+    if not app:
+        app = QtWidgets.QApplication(sys.argv)
     ex = Window(app, title, message, level)
     ex.show()
     # sys.exit(app.exec_())