added basic app

pype/tools/standalonepublish/app.py

@@ -0,0 +1,155 @@
+import os
+import sys
+import json
+from subprocess import Popen
+from pype import lib as pypelib
+from avalon.vendor.Qt import QtWidgets, QtCore
+from avalon import api, style, schema
+from avalon.tools import lib as parentlib
+from .widgets import *
+# Move this to pype lib?
+from avalon.tools.libraryloader.io_nonsingleton import DbConnector
+
+
+class Window(QtWidgets.QDialog):
+    _db = DbConnector()
+    _jobs = {}
+    WIDTH = 1000
+    HEIGHT = 500
+    NOT_SELECTED = '< Nothing is selected >'
+
+    def __init__(self, parent=None):
+        super(Window, self).__init__(parent)
+        self._db.install()
+
+        self.setWindowTitle("Standalone Publish")
+        self.setFocusPolicy(QtCore.Qt.StrongFocus)
+        self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
+        self.setStyleSheet(style.load_stylesheet())
+
+        # Validators
+        self.valid_parent = False
+
+        # statusbar - added under asset_widget
+        label_message = QtWidgets.QLabel()
+        label_message.setFixedHeight(20)
+
+        # assets widget
+        widget_assets_wrap = QtWidgets.QWidget()
+        widget_assets_wrap.setContentsMargins(0, 0, 0, 0)
+        widget_assets = AssetWidget(self)
+
+        layout_assets = QtWidgets.QVBoxLayout(widget_assets_wrap)
+        layout_assets.addWidget(widget_assets)
+        layout_assets.addWidget(label_message)
+
+
+        # family widget
+        widget_family = FamilyWidget(self)
+
+        # components widget
+        widget_components = DropDataFrame(self)
+
+        # Body
+        body = QtWidgets.QSplitter()
+        body.setContentsMargins(0, 0, 0, 0)
+        body.setSizePolicy(
+            QtWidgets.QSizePolicy.Expanding,
+            QtWidgets.QSizePolicy.Expanding
+        )
+        body.setOrientation(QtCore.Qt.Horizontal)
+        body.addWidget(widget_assets_wrap)
+        body.addWidget(widget_family)
+        body.addWidget(widget_components)
+        body.setStretchFactor(body.indexOf(widget_assets_wrap), 2)
+        body.setStretchFactor(body.indexOf(widget_family), 2)
+        body.setStretchFactor(body.indexOf(widget_components), 3)
+
+        layout = QtWidgets.QVBoxLayout(self)
+        layout.addWidget(body)
+
+        self.resize(self.WIDTH, self.HEIGHT)
+
+        # signals
+        widget_assets.selection_changed.connect(self.on_asset_changed)
+
+        self.label_message = label_message
+        self.widget_assets = widget_assets
+        self.widget_family = widget_family
+
+        self.echo("Connected to Database")
+
+        # on start
+        self.on_start()
+
+    @property
+    def db(self):
+        return self._db
+
+    def on_start(self):
+        # Refresh asset input in Family widget
+        self.on_asset_changed()
+
+    def get_avalon_parent(self, entity):
+        parent_id = entity['data']['visualParent']
+        parents = []
+        if parent_id is not None:
+            parent = self.db.find_one({'_id': parent_id})
+            parents.extend(self.get_avalon_parent(parent))
+            parents.append(parent['name'])
+        return parents
+
+    def echo(self, message):
+        self.label_message.setText(str(message))
+        QtCore.QTimer.singleShot(5000, lambda: self.label_message.setText(""))
+
+    def on_asset_changed(self):
+        """Callback on asset selection changed
+
+        This updates the task view.
+
+        """
+        selected = self.widget_assets.get_selected_assets()
+        if len(selected) == 1:
+            self.valid_parent = True
+            asset = self.db.find_one({"_id": selected[0], "type": "asset"})
+            self.widget_family.change_asset(asset['name'])
+        else:
+            self.valid_parent = False
+            self.widget_family.change_asset(self.NOT_SELECTED)
+        self.widget_family.on_data_changed()
+
+
+def show(parent=None, debug=False, context=None):
+    """Display Loader GUI
+
+    Arguments:
+        debug (bool, optional): Run loader in debug-mode,
+            defaults to False
+
+    """
+
+    try:
+        module.window.close()
+        del module.window
+    except (RuntimeError, AttributeError):
+        pass
+
+    with parentlib.application():
+        window = Window(parent, context)
+        window.show()
+
+        module.window = window
+
+
+def cli(args):
+    import argparse
+    parser = argparse.ArgumentParser()
+    parser.add_argument("project")
+    parser.add_argument("asset")
+
+    args = parser.parse_args(args)
+    # project = args.project
+    # asset = args.asset
+
+    show()