Rough version of asset loader GUI

pyblish_starter/__init__.py

@@ -1,24 +1,26 @@
 from .pipeline import (
+    ls,
     setup,
     register_plugins,
 
 
     # Internal
-    time,# as _time,
-    format_private_dir,# as _format_private_dir,
+    time as _time,
+    format_private_dir as _format_private_dir,
 
-    _families,
-    _defaults
+    _registered_families,
+    _registered_defaults
 )
 
 
 __all__ = [
-    "time",
+    "ls",
     "setup",
     "register_plugins",
-    "format_private_dir",
 
     # Internal
-    "_defaults",
-    "_families",
+    "_time",
+    "_format_private_dir",
+    "_registered_defaults",
+    "_registered_families",
 ]

pyblish_starter/maya/lib.py

@@ -6,8 +6,8 @@
 from ..pipeline import (
     register_default,
     register_family,
-    _defaults,
-    _families,
+    _registered_defaults,
+    _registered_families,
 )
 
 
@@ -52,6 +52,7 @@ def hierarchy_from_string(hierarchy):
         for parent in sorted(parents):
             if parent < padding:
                 cmds.parent(name, parents[parent])
+                break
 
     # Return assembly
     return parents[0]
@@ -172,11 +173,11 @@ def create(name, family, use_selection=False):
     """
 
     try:
-        item = next(i for i in _families if i["name"] == family)
+        item = next(i for i in _registered_families if i["name"] == family)
     except:
         raise RuntimeError("{0} is not a valid family".format(family))
 
-    attrs = _defaults + item.get("attributes", [])
+    attrs = _registered_defaults + item.get("attributes", [])
 
     if not use_selection:
         cmds.select(deselect=True)

pyblish_starter/pipeline.py

@@ -3,14 +3,19 @@
 
 from pyblish import api
 
-_defaults = []
-_families = []
+_registered_defaults = []
+_registered_families = []
 
 
 def setup():
     register_plugins()
 
 
+def ls(root):
+    """List available assets"""
+    return os.listdir(os.path.join(root, "public"))
+
+
 def register_plugins():
     """Register accompanying plugins"""
     from . import plugins
@@ -36,7 +41,7 @@ def register_default(item):
     assert "key" in item
     assert "value" in item
 
-    _defaults.append(item)
+    _registered_defaults.append(item)
 
 
 def register_family(item):
@@ -61,12 +66,12 @@ def register_family(item):
     assert "name" in item
 
     # If family was already registered then overwrite it
-    for i, family in enumerate(_families):
+    for i, family in enumerate(_registered_families):
         if item["name"] == family["name"]:
-            _families[i] = item
+            _registered_families[i] = item
             return
 
-    _families.append(item)
+    _registered_families.append(item)
 
 
 def time():

pyblish_starter/tools/asset_loader/__main__.py

@@ -0,0 +1,4 @@
+from . import app
+
+with app.application():
+    app.show()

pyblish_starter/tools/asset_loader/app.py

@@ -0,0 +1,164 @@
+import sys
+import contextlib
+
+from ...vendor.Qt import QtWidgets, QtCore
+from ...maya.lib import load
+from ... import ls
+
+
+self = sys.modules[__name__]
+self._window = None
+
+
+class Window(QtWidgets.QDialog):
+    def __init__(self, parent=None):
+        super(Window, self).__init__(parent)
+        self.setWindowTitle("Starter Asset Loader")
+        self.setFocusPolicy(QtCore.Qt.StrongFocus)
+
+        body = QtWidgets.QWidget()
+        footer = QtWidgets.QWidget()
+
+        container = QtWidgets.QWidget()
+
+        listing = QtWidgets.QListWidget()
+
+        layout = QtWidgets.QVBoxLayout(container)
+        layout.addWidget(listing)
+        layout.setContentsMargins(0, 0, 0, 0)
+
+        options = QtWidgets.QWidget()
+        layout = QtWidgets.QGridLayout(options)
+        layout.setContentsMargins(0, 0, 0, 0)
+
+        autoclose_checkbox = QtWidgets.QCheckBox("Close after load")
+        autoclose_checkbox.setCheckState(QtCore.Qt.Checked)
+        layout.addWidget(autoclose_checkbox, 1, 0)
+
+        layout = QtWidgets.QVBoxLayout(body)
+        layout.addWidget(container)
+        layout.addWidget(options, 0, QtCore.Qt.AlignLeft)
+        layout.setContentsMargins(0, 0, 0, 0)
+
+        load_button = QtWidgets.QPushButton("Load")
+        error_msg = QtWidgets.QLabel()
+        error_msg.hide()
+
+        layout = QtWidgets.QVBoxLayout(footer)
+        layout.addWidget(load_button)
+        layout.addWidget(error_msg)
+        layout.setContentsMargins(0, 0, 0, 0)
+
+        layout = QtWidgets.QVBoxLayout(self)
+        layout.addWidget(body)
+        layout.addWidget(footer)
+
+        self.data = {
+            "button": {
+                "load": load_button,
+                "autoclose": autoclose_checkbox,
+            },
+            "model": {
+                "listing": listing
+            }
+        }
+
+        load_button.clicked.connect(self.on_load)
+        listing.currentItemChanged.connect(self.on_data_changed)
+
+        # Defaults
+        self.resize(220, 250)
+        load_button.setFocus()
+        load_button.setEnabled(False)
+
+    def on_data_changed(self, *args):
+        button = self.data["button"]["load"]
+        item = self.data["model"]["listing"].currentItem()
+        button.setEnabled(item.data(QtCore.Qt.ItemIsEnabled))
+
+    def keyPressEvent(self, event):
+        """Custom keyPressEvent.
+
+        Override keyPressEvent to do nothing so that Maya's panels won't
+        take focus when pressing "SHIFT" whilst mouse is over viewport or
+        outliner. This way users don't accidently perform Maya commands
+        whilst trying to name an instance.
+
+        """
+
+        pass
+
+    def refresh(self, root):
+        listing = self.data["model"]["listing"]
+
+        assets = ls(root)
+
+        if assets:
+            for asset in assets:
+                item = QtWidgets.QListWidgetItem(asset)
+                item.setData(QtCore.Qt.ItemIsEnabled, True)
+                listing.addItem(item)
+
+        else:
+            item = QtWidgets.QListWidgetItem("No assets found")
+            item.setData(QtCore.Qt.ItemIsEnabled, False)
+            listing.addItem(item)
+
+        listing.setCurrentItem(listing.item(0))
+
+    def on_load(self):
+        listing = self.data["model"]["listing"]
+        autoclose_checkbox = self.data["button"]["autoclose"]
+        error_msg = self.findChild(QtWidgets.QWidget, "Error Message")
+
+        item = listing.currentItem()
+
+        if item is not None:
+            try:
+                load(item.text())
+
+            except NameError as e:
+                error_msg.setText(str(e))
+                error_msg.show()
+                raise
+
+            except (TypeError, RuntimeError, KeyError) as e:
+                error_msg.setText("Program error: %s" % str(e))
+                error_msg.show()
+                raise
+
+        if autoclose_checkbox.checkState():
+            self.close()
+
+
+def show(root):
+    if self._window:
+        self._window.close()
+        del(self._window)
+
+    try:
+        widgets = QtWidgets.QApplication.topLevelWidgets()
+        widgets = dict((w.objectName(), w) for w in widgets)
+        parent = widgets["MayaWindow"]
+    except KeyError:
+        parent = None
+
+    window = Window(parent)
+    window.show()
+    window.refresh(root)
+
+    self._window = window
+
+
+@contextlib.contextmanager
+def application():
+    app = QtWidgets.QApplication(sys.argv)
+    yield
+    app.exec_()
+
+
+if __name__ == '__main__':
+    import os
+
+    with application():
+        show(os.path.expanduser("~"))

pyblish_starter/tools/instance_creator/__init__.py

@@ -1,15 +1,7 @@
-from .lib import (
-    register_default,
-    register_family,
-)
-
 from .app import (
     show,
 )
 
 __all__ = [
-    "register_default",
-    "register_family",
-
     "show",
 ]