General: Benevolent context label collector

openpype/plugins/publish/collect_context_label.py

@@ -1,5 +1,6 @@
 """
-Requires:
+Optional:
+ context -> hostName (str)
  context -> currentFile (str)
 Provides:
  context -> label (str)
@@ -16,16 +17,22 @@ class CollectContextLabel(pyblish.api.ContextPlugin):
  label = "Context Label"
 
  def process(self, context):
-
- # Get last registered host
- host = pyblish.api.registered_hosts()[-1]
-
- # Get scene name from "currentFile"
- path = context.data.get("currentFile") or "<Unsaved>"
- base = os.path.basename(path)
+ # Add ability to use custom context label
+ context_label = context.data.get("contextLabel")
+ if context_label:
+ context.data["label"] = context_label
+ return
+
+ host_name = context.data.get("hostName")
+ if not host_name:
+ host_name = pyblish.api.registered_hosts()[-1]
+ # Use host name as base for label
+ label = host_name.title()
+
+ # Get scene name from "currentFile" and use basename as ending of label
+ path = context.data.get("currentFile")
+ if path:
+ label += " - {}".format(os.path.basename(path))
 
  # Set label
- label = "{host} - {scene}".format(host=host.title(), scene=base)
- if host == "standalonepublisher":
- label = host.title()
  context.data["label"] = label