ayon-core/igniter/install_dialog.py
2020-09-16 09:56:16 +02:00

294 lines
10 KiB
Python

# -*- coding: utf-8 -*-
"""Show dialog for choosing central pype repository."""
import sys
import os
from Qt import QtCore, QtGui, QtWidgets
from .install_thread import InstallThread
class InstallDialog(QtWidgets.QDialog):
_size_w = 400
_size_h = 300
_path = None
_controls_disabled = False
def __init__(self, parent=None):
super(InstallDialog, self).__init__(parent)
self.setWindowTitle("Pype - Configure Pype repository path")
self._icon_path = os.path.join(
os.path.dirname(__file__), 'pype_icon.png')
icon = QtGui.QIcon(self._icon_path)
self.setWindowIcon(icon)
self.setWindowFlags(
QtCore.Qt.WindowCloseButtonHint |
QtCore.Qt.WindowMinimizeButtonHint
)
self.setMinimumSize(
QtCore.QSize(self._size_w, self._size_h))
self.setMaximumSize(
QtCore.QSize(self._size_w + 100, self._size_h + 100))
# style for normal console text
self.default_console_style = QtGui.QTextCharFormat()
self.default_console_style.setFontPointSize(0.1)
self.default_console_style.setForeground(
QtGui.QColor.fromRgb(72, 200, 150))
# style for error console text
self.error_console_style = QtGui.QTextCharFormat()
self.error_console_style.setFontPointSize(0.1)
self.error_console_style.setForeground(
QtGui.QColor.fromRgb(184, 54, 19))
self._init_ui()
def _init_ui(self):
self.setStyleSheet("background-color: rgb(23, 23, 23);")
main = QtWidgets.QVBoxLayout(self)
# Main info
# --------------------------------------------------------------------
self.main_label = QtWidgets.QLabel(
"""Welcome to <b>Pype</b>
<p>
We've detected <b>Pype</b> is not configured yet. But don't worry,
this is as easy as setting one path.
<p>
""")
self.main_label.setWordWrap(True)
self.main_label.setStyleSheet("color: rgb(200, 200, 200);")
# Pype path info
# --------------------------------------------------------------------
self.pype_path_label = QtWidgets.QLabel(
"""Set this path to your studio <b>Pype repository</b> to keep in
sync with your studio environment. This can be path or url.
Leave it empty if you want to use Pype version that come with this
installation.
"""
)
self.pype_path_label.setWordWrap(True)
self.pype_path_label.setStyleSheet("color: rgb(150, 150, 150);")
# Path/Url box | Select button
# --------------------------------------------------------------------
input_layout = QtWidgets.QHBoxLayout()
input_layout.setContentsMargins(0, 10, 0, 10)
self.user_input = QtWidgets.QLineEdit()
self.user_input.setPlaceholderText("Pype repository path or url")
self.user_input.textChanged.connect(self._path_changed)
self.user_input.setStyleSheet(
("color: rgb(233, 233, 233);"
"background-color: rgb(64, 64, 64);"
"padding: 0.5em;"
"border: 1px solid rgb(32, 32, 32);")
)
self._btn_select = QtWidgets.QPushButton("Select")
self._btn_select.setToolTip(
"Select Pype repository"
)
self._btn_select.setStyleSheet(
("color: rgb(64, 64, 64);"
"background-color: rgb(72, 200, 150);"
"padding: 0.5em;")
)
self._btn_select.setMaximumSize(100, 140)
self._btn_select.clicked.connect(self._on_select_clicked)
input_layout.addWidget(self.user_input)
input_layout.addWidget(self._btn_select)
# Bottom button bar
# --------------------------------------------------------------------
bottom_widget = QtWidgets.QWidget()
bottom_layout = QtWidgets.QHBoxLayout()
pype_logo_label = QtWidgets.QLabel("pype logo")
pype_logo = QtGui.QPixmap(self._icon_path)
# pype_logo.scaled(
# pype_logo_label.width(),
# pype_logo_label.height(), QtCore.Qt.KeepAspectRatio)
pype_logo_label.setPixmap(pype_logo)
pype_logo_label.setContentsMargins(10, 0, 0, 10)
self._ok_button = QtWidgets.QPushButton("OK")
self._ok_button.setStyleSheet(
("color: rgb(64, 64, 64);"
"background-color: rgb(72, 200, 150);"
"padding: 0.5em;")
)
self._ok_button.setMinimumSize(64, 24)
self._ok_button.setToolTip("Save and continue")
self._ok_button.clicked.connect(self._on_ok_clicked)
self._exit_button = QtWidgets.QPushButton("Exit")
self._exit_button.setStyleSheet(
("color: rgb(64, 64, 64);"
"background-color: rgb(128, 128, 128);"
"padding: 0.5em;")
)
self._exit_button.setMinimumSize(64, 24)
self._exit_button.setToolTip("Exit without saving")
self._exit_button.clicked.connect(self._on_exit_clicked)
bottom_layout.setContentsMargins(0, 10, 0, 0)
bottom_layout.addWidget(pype_logo_label)
bottom_layout.addStretch(1)
bottom_layout.addWidget(self._ok_button)
bottom_layout.addWidget(self._exit_button)
bottom_widget.setLayout(bottom_layout)
bottom_widget.setStyleSheet("background-color: rgb(32, 32, 32);")
# Status label
# --------------------------------------------------------------------
self._status_label = QtWidgets.QLabel("Console:")
self._status_label.setContentsMargins(0, 10, 0, 10)
self._status_label.setStyleSheet("color: rgb(61, 115, 97);")
# Console
# --------------------------------------------------------------------
self._status_box = QtWidgets.QPlainTextEdit()
self._status_box.setReadOnly(True)
self._status_box.setCurrentCharFormat(self.default_console_style)
self._status_box.setStyleSheet(
"""QPlainTextEdit {
background-color: rgb(32, 32, 32);
color: rgb(72, 200, 150);
font-family: Courier;
font-size: 3pt;
}
QScrollBar:vertical {
border: 1px solid rgb(61, 115, 97);
background: #000;
width:5px;
margin: 0px 0px 0px 0px;
}
QScrollBar::handle:vertical {
background: rgb(72, 200, 150);
min-height: 0px;
}
QScrollBar::sub-page:vertical {
background: rgb(31, 62, 50);
}
QScrollBar::add-page:vertical {
background: rgb(31, 62, 50);
}
QScrollBar::add-line:vertical {
background: rgb(72, 200, 150);
height: 0px;
subcontrol-position: bottom;
subcontrol-origin: margin;
}
QScrollBar::sub-line:vertical {
background: rgb(72, 200, 150);
height: 0 px;
subcontrol-position: top;
subcontrol-origin: margin;
}
"""
)
# Progress bar
# --------------------------------------------------------------------
self._progress_bar = QtWidgets.QProgressBar()
self._progress_bar.setValue(0)
self._progress_bar.setAlignment(QtCore.Qt.AlignCenter)
self._progress_bar.setTextVisible(False)
# setting font and the size
self._progress_bar.setFont(QtGui.QFont('Arial', 7))
self._progress_bar.setStyleSheet(
"""QProgressBar:horizontal {
height: 5px;
border: 1px solid rgb(31, 62, 50);
color: rgb(72, 200, 150);
}
QProgressBar::chunk:horizontal {
background-color: rgb(72, 200, 150);
}
"""
)
# add all to main
main.addWidget(self.main_label)
main.addWidget(self.pype_path_label)
main.addLayout(input_layout)
main.addStretch(1)
main.addWidget(self._status_label)
main.addWidget(self._status_box)
main.addWidget(self._progress_bar)
main.addWidget(bottom_widget)
self.setLayout(main)
def _on_select_clicked(self):
fname = QtWidgets.QFileDialog.getExistingDirectory(
self, 'Select path')
if fname:
fname = QtCore.QDir.toNativeSeparators(fname)
if os.path.isdir(fname):
self.user_input.setText(fname)
def _on_ok_clicked(self):
self._disable_buttons()
self._install_thread = InstallThread(self)
self._install_thread.message.connect(self._update_console)
self._install_thread.progress.connect(self._update_progress)
self._install_thread.finished.connect(self._enable_buttons)
self._install_thread.set_path(self._path)
self._install_thread.start()
def _update_progress(self, progress: int):
self._progress_bar.setValue(progress)
def _on_exit_clicked(self):
self.close()
def _path_changed(self, path: str) -> None:
self._path = path
self._status_label.setText(f"selected <b>{path}</b>")
def _update_console(self, msg: str, error: bool = False) -> None:
"""Display message.
Args:
msg (str): message.
error (bool): if True, print it red.
"""
if not error:
self._status_box.setCurrentCharFormat(self.default_console_style)
else:
self._status_box.setCurrentCharFormat(self.error_console_style)
self._status_box.appendPlainText(msg)
def _disable_buttons(self):
self._btn_select.setEnabled(False)
self._exit_button.setEnabled(False)
self._ok_button.setEnabled(False)
self._controls_disabled = True
def _enable_buttons(self):
self._btn_select.setEnabled(True)
self._exit_button.setEnabled(True)
self._ok_button.setEnabled(True)
self._controls_disabled = False
def closeEvent(self, event):
if self._controls_disabled:
return event.ignore()
return super(InstallDialog, self).closeEvent(event)
if __name__ == "__main__":
app = QtWidgets.QApplication(sys.argv)
d = InstallDialog()
d.show()
sys.exit(app.exec_())