implemented CreateProjectDialog to be able create projects

openpype/tools/project_manager/project_manager/widgets.py

@@ -1,8 +1,16 @@
 import re
+
 from .constants import (
     NAME_ALLOWED_SYMBOLS,
     NAME_REGEX
 )
+from openpype.lib import (
+    create_project,
+    PROJECT_NAME_ALLOWED_SYMBOLS,
+    PROJECT_NAME_REGEX
+)
+from avalon.api import AvalonMongoDB
+
 from Qt import QtWidgets, QtCore
 
 
@@ -80,3 +88,177 @@ class FilterComboBox(QtWidgets.QComboBox):
         self._completer.setCompletionColumn(column)
         self._filter_proxy_model.setFilterKeyColumn(column)
         super(FilterComboBox, self).setModelColumn(column)
+
+
+class CreateProjectDialog(QtWidgets.QDialog):
+    def __init__(self, parent=None, dbcon=None):
+        super(CreateProjectDialog, self).__init__(parent)
+
+        self.setWindowTitle("Create Project")
+
+        self.allowed_regex = "[^{}]+".format(PROJECT_NAME_ALLOWED_SYMBOLS)
+
+        if dbcon is None:
+            dbcon = AvalonMongoDB()
+
+        self.dbcon = dbcon
+        self._ignore_code_change = False
+        self._project_name_is_valid = False
+        self._project_code_is_valid = False
+        self._project_code_value = None
+
+        project_names, project_codes = self._get_existing_projects()
+
+        inputs_widget = QtWidgets.QWidget(self)
+        project_name_input = QtWidgets.QLineEdit(inputs_widget)
+        project_code_input = QtWidgets.QLineEdit(inputs_widget)
+        library_project_input = QtWidgets.QCheckBox(inputs_widget)
+
+        inputs_layout = QtWidgets.QFormLayout(inputs_widget)
+        inputs_layout.setContentsMargins(0, 0, 0, 0)
+        inputs_layout.addRow("Project name:", project_name_input)
+        inputs_layout.addRow("Project code:", project_code_input)
+        inputs_layout.addRow("Library project:", library_project_input)
+
+        project_name_label = QtWidgets.QLabel(self)
+        project_code_label = QtWidgets.QLabel(self)
+
+        btns_widget = QtWidgets.QWidget(self)
+        ok_btn = QtWidgets.QPushButton("Ok", btns_widget)
+        ok_btn.setEnabled(False)
+        cancel_btn = QtWidgets.QPushButton("Cancel", btns_widget)
+        btns_layout = QtWidgets.QHBoxLayout(btns_widget)
+        btns_layout.setContentsMargins(0, 0, 0, 0)
+        btns_layout.addStretch(1)
+        btns_layout.addWidget(ok_btn)
+        btns_layout.addWidget(cancel_btn)
+
+        main_layout = QtWidgets.QVBoxLayout(self)
+        main_layout.addWidget(inputs_widget, 0)
+        main_layout.addWidget(project_name_label, 1)
+        main_layout.addWidget(project_code_label, 1)
+        main_layout.addStretch(1)
+        main_layout.addWidget(btns_widget, 0)
+
+        project_name_input.textChanged.connect(self._on_project_name_change)
+        project_code_input.textChanged.connect(self._on_project_code_change)
+        ok_btn.clicked.connect(self._on_ok_clicked)
+        cancel_btn.clicked.connect(self._on_cancel_clicked)
+
+        self.invalid_project_names = project_names
+        self.invalid_project_codes = project_codes
+
+        self.project_name_label = project_name_label
+        self.project_code_label = project_code_label
+
+        self.project_name_input = project_name_input
+        self.project_code_input = project_code_input
+        self.library_project_input = library_project_input
+
+        self.ok_btn = ok_btn
+
+    @property
+    def project_name(self):
+        return self.project_name_input.text()
+
+    def _on_project_name_change(self, value):
+        if self._project_code_value is None:
+            self._ignore_code_change = True
+            self.project_code_input.setText(value.lower())
+            self._ignore_code_change = False
+
+        self._update_valid_project_name(value)
+
+    def _on_project_code_change(self, value):
+        if not value:
+            value = None
+
+        self._update_valid_project_code(value)
+
+        if not self._ignore_code_change:
+            self._project_code_value = value
+
+    def _update_valid_project_name(self, value):
+        message = ""
+        is_valid = True
+        if not value:
+            message = "Project name is empty"
+            is_valid = False
+
+        elif value in self.invalid_project_names:
+            message = "Project name \"{}\" already exist".format(value)
+            is_valid = False
+
+        elif not PROJECT_NAME_REGEX.match(value):
+            message = (
+                "Project name \"{}\" contain not supported symbols"
+            ).format(value)
+            is_valid = False
+
+        self._project_name_is_valid = is_valid
+        self.project_name_label.setText(message)
+        self._enable_button()
+
+    def _update_valid_project_code(self, value):
+        message = ""
+        is_valid = True
+        if not value:
+            message = "Project code is empty"
+            is_valid = False
+
+        elif value in self.invalid_project_names:
+            message = "Project code \"{}\" already exist".format(value)
+            is_valid = False
+
+        elif not PROJECT_NAME_REGEX.match(value):
+            message = (
+                "Project code \"{}\" contain not supported symbols"
+            ).format(value)
+            is_valid = False
+
+        self._project_code_is_valid = is_valid
+        self.project_code_label.setText(message)
+        self._enable_button()
+
+    def _enable_button(self):
+        self.ok_btn.setEnabled(
+            self._project_name_is_valid and self._project_code_is_valid
+        )
+
+    def _on_cancel_clicked(self):
+        self.done(0)
+
+    def _on_ok_clicked(self):
+        if not self._project_name_is_valid or not self._project_code_is_valid:
+            return
+
+        project_name = self.project_name_input.text()
+        project_code = self.project_code_input.text()
+        library_project = self.library_project_input.isChecked()
+        create_project(project_name, project_code, library_project, self.dbcon)
+
+        self.done(1)
+
+    def _get_existing_projects(self):
+        project_names = set()
+        project_codes = set()
+        for project_name in self.dbcon.database.collection_names():
+            # Each collection will have exactly one project document
+            project_doc = self.dbcon.database[project_name].find_one(
+                {"type": "project"},
+                {"name": 1, "data.code": 1}
+            )
+            if not project_doc:
+                continue
+
+            project_name = project_doc.get("name")
+            if not project_name:
+                continue
+
+            project_names.add(project_name)
+            project_code = project_doc.get("data", {}).get("code")
+            if not project_code:
+                project_code = project_name
+
+            project_codes.add(project_code)
+        return project_names, project_codes