moved new classes to separated file

This commit is contained in:
iLLiCiTiT 2021-01-27 14:33:46 +01:00
parent 6e9e7a40f8
commit 142bccef86
3 changed files with 451 additions and 277 deletions

View file

@ -31,7 +31,7 @@ from pype.settings.lib import (
)
from .widgets import UnsavedChangesDialog
from . import lib
from .item_types import create_ui_for_entity
from .item_widgets import create_ui_for_entity
from avalon.mongodb import (
AvalonMongoConnection,
AvalonMongoDB

View file

@ -11,21 +11,6 @@ from .widgets import (
)
from .multiselection_combobox import MultiSelectionComboBox
from .lib import NOT_SET, METADATA_KEY, TypeToKlass, CHILD_OFFSET
from pype.settings.entities import (
GUIEntity,
DictImmutableKeysEntity,
DictMutableKeysEntity,
ListEntity,
PathEntity,
ListStrictEntity,
NumberEntity,
BoolEntity,
EnumEntity,
TextEntity,
PathInput,
RawJsonEntity
)
from pype.api import Logger
from avalon.vendor import qtawesome
@ -4119,267 +4104,6 @@ class SplitterWidget(QtWidgets.QWidget):
self.hidden_by_role = False
class BaseWidget(QtWidgets.QWidget):
def __init__(self, entity, entity_widget):
self.entity = entity
self.entity_widget = entity_widget
super(BaseWidget, self).__init__(entity_widget.content_widget)
self.label_widget = None
self.create_ui()
def show_actions_menu(self, event):
print("Show actions for {}".format(self.entity.path))
class GUIWidget(BaseWidget):
separator_height = 2
def create_ui(self):
entity_type = self.entity["type"]
if entity_type == "label":
self._create_label_ui()
elif entity_type in ("separator", "splitter"):
self._create_separator_ui()
else:
raise KeyError("Unknown GUI type {}".format(entity_type))
self.entity_widget.add_widget_to_layout(self)
def _create_label_ui(self):
self.setObjectName("LabelWidget")
label = self.entity["label"]
label_widget = QtWidgets.QLabel(label, self)
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(0, 5, 0, 5)
layout.addWidget(label_widget)
def _create_separator_ui(self):
splitter_item = QtWidgets.QWidget(self)
splitter_item.setObjectName("SplitterItem")
splitter_item.setMinimumHeight(self.separator_height)
splitter_item.setMaximumHeight(self.separator_height)
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(5, 5, 5, 5)
layout.addWidget(splitter_item)
class DictImmutableKeysWidget(BaseWidget):
def create_ui(self):
self.input_fields = []
if not self.entity.is_dynamic_item and not self.entity.label:
self._ui_item_without_label()
else:
self._ui_item_or_as_widget()
for child_obj in self.entity.children:
self.input_fields.append(
create_ui_for_entity(child_obj, self)
)
self.entity_widget.add_widget_to_layout(self)
# any_visible = False
# for input_field in self.input_fields:
# if not input_field.hidden_by_role:
# any_visible = True
# break
#
# if not any_visible:
# self.hide()
def _ui_item_without_label(self):
self.setObjectName("DictInvisible")
self.body_widget = None
self.content_widget = self
self.content_layout = QtWidgets.QGridLayout(self)
self.content_layout.setContentsMargins(0, 0, 0, 0)
self.content_layout.setSpacing(5)
def _ui_item_or_as_widget(self):
content_widget = QtWidgets.QWidget(self)
if self.entity.is_dynamic_item:
content_widget.setObjectName("DictAsWidgetBody")
show_borders = str(int(self.show_borders))
content_widget.setProperty("show_borders", show_borders)
content_layout_margins = (5, 5, 5, 5)
main_layout_spacing = 5
body_widget = None
label_widget = QtWidgets.QLabel(self.entity.label)
else:
content_widget.setObjectName("ContentWidget")
if self.entity.highlight_content:
content_state = "hightlighted"
bottom_margin = 5
else:
content_state = ""
bottom_margin = 0
content_widget.setProperty("content_state", content_state)
content_layout_margins = (CHILD_OFFSET, 5, 0, bottom_margin)
main_layout_spacing = 0
body_widget = ExpandingWidget(self.entity.label, self)
label_widget = body_widget.label_widget
body_widget.set_content_widget(content_widget)
content_layout = QtWidgets.QGridLayout(content_widget)
content_layout.setContentsMargins(*content_layout_margins)
main_layout = QtWidgets.QHBoxLayout(self)
main_layout.setContentsMargins(0, 0, 0, 0)
main_layout.setSpacing(main_layout_spacing)
if not body_widget:
main_layout.addWidget(content_widget)
else:
main_layout.addWidget(body_widget)
self.label_widget = label_widget
self.body_widget = body_widget
self.content_widget = content_widget
self.content_layout = content_layout
if body_widget:
if len(self.input_fields) == 1 and self.checkbox_widget:
body_widget.hide_toolbox(hide_content=True)
elif self.entity.collapsible:
if not self.entity.collapsed:
body_widget.toggle_content()
else:
body_widget.hide_toolbox(hide_content=False)
def add_widget_to_layout(self, widget, label=None):
row = self.content_layout.rowCount()
if label:
label_widget = GridLabelWidget(label, widget)
label_widget.input_field = widget
widget.label_widget = label_widget
self.content_layout.addWidget(label_widget, row, 0, 1, 1)
self.content_layout.addWidget(widget, row, 1, 1, 1)
else:
self.content_layout.addWidget(widget, row, 0, 1, 2)
class BoolWidget(BaseWidget):
def create_ui(self):
checkbox_height = self.style().pixelMetric(
QtWidgets.QStyle.PM_IndicatorHeight
)
self.input_field = NiceCheckbox(height=checkbox_height, parent=self)
spacer = QtWidgets.QWidget(self)
spacer.setAttribute(QtCore.Qt.WA_TranslucentBackground)
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(5)
layout.addWidget(self.input_field, 0)
layout.addWidget(spacer, 1)
self.setFocusProxy(self.input_field)
self.input_field.stateChanged.connect(self._on_value_change)
self.entity_widget.add_widget_to_layout(self, self.entity.label)
def _on_value_change(self):
print("value changed", self.entity.path)
class TextWidgetTwo(BaseWidget):
def create_ui(self):
multiline = self.entity.multiline
if multiline:
self.input_field = QtWidgets.QPlainTextEdit(self)
else:
self.input_field = QtWidgets.QLineEdit(self)
placeholder_text = self.entity.placeholder_text
if placeholder_text:
self.input_field.setPlaceholderText(placeholder_text)
self.setFocusProxy(self.input_field)
layout_kwargs = {}
if multiline:
layout_kwargs["alignment"] = QtCore.Qt.AlignTop
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(5)
layout.addWidget(self.input_field, 1, **layout_kwargs)
self.input_field.textChanged.connect(self._on_value_change)
self.entity_widget.add_widget_to_layout(self, self.entity.label)
def _on_value_change(self):
print("value changed", self.entity.path)
class NumberWidgetTwo(BaseWidget):
def create_ui(self):
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(5)
kwargs = {
"minimum": self.entity.minimum,
"maximum": self.entity.maximum,
"decimal": self.entity.decimal
}
self.input_field = NumberSpinBox(self, **kwargs)
self.setFocusProxy(self.input_field)
layout.addWidget(self.input_field, 1)
self.input_field.valueChanged.connect(self._on_value_change)
self.entity_widget.add_widget_to_layout(self, self.entity.label)
def _on_value_change(self):
print("value changed", self.entity.path)
def create_ui_for_entity(entity, entity_widget):
if isinstance(entity, GUIEntity):
return GUIWidget(entity, entity_widget)
elif isinstance(entity, DictImmutableKeysEntity):
return DictImmutableKeysWidget(entity, entity_widget)
elif isinstance(entity, BoolEntity):
return BoolWidget(entity, entity_widget)
elif isinstance(entity, TextEntity):
return TextWidgetTwo(entity, entity_widget)
elif isinstance(entity, NumberEntity):
return NumberWidgetTwo(entity, entity_widget)
# DictMutableKeysEntity,
# ListEntity,
# PathEntity,
# ListStrictEntity,
#
# NumberEntity,
# EnumEntity,
# PathInput,
# RawJsonEntity
label = "<{}>: {} ({})".format(entity.__class__.__name__, entity.path, entity.current_value)
widget = QtWidgets.QLabel(label, entity_widget)
entity_widget.add_widget_to_layout(widget)
return widget
TypeToKlass.types["boolean"] = BooleanWidget
TypeToKlass.types["number"] = NumberWidget
TypeToKlass.types["text"] = TextWidget

View file

@ -0,0 +1,450 @@
from Qt import QtWidgets, QtCore, QtGui
from pype.settings.entities import (
GUIEntity,
DictImmutableKeysEntity,
DictMutableKeysEntity,
ListEntity,
PathEntity,
ListStrictEntity,
NumberEntity,
BoolEntity,
EnumEntity,
TextEntity,
PathInput,
RawJsonEntity
)
from .widgets import (
IconButton,
ExpandingWidget,
NumberSpinBox,
GridLabelWidget,
ComboBox,
NiceCheckbox
)
from .lib import CHILD_OFFSET
class BaseWidget(QtWidgets.QWidget):
def __init__(self, entity, entity_widget):
self.entity = entity
self.entity_widget = entity_widget
self.ignore_input_changes = entity_widget.ignore_input_changes
self._style_state = None
super(BaseWidget, self).__init__(entity_widget.content_widget)
self.entity.on_change_callbacks.append(self._on_entity_change)
self.label_widget = None
self.create_ui()
@staticmethod
def get_style_state(
is_invalid, is_modified, has_project_override, has_studio_override
):
"""Return stylesheet state by intered booleans."""
if is_invalid:
return "invalid"
if is_modified:
return "modified"
if has_project_override:
return "overriden"
if has_studio_override:
return "studio"
return ""
def show_actions_menu(self, event):
print("Show actions for {}".format(self.entity.path))
class InputWidget(BaseWidget):
def update_style(self):
state = self.get_style_state(
self.entity.is_invalid,
self.entity.has_unsaved_changes,
self.entity.has_project_override,
self.entity.has_studio_override
)
if self._style_state == state:
return
self._style_state = state
self.input_field.setProperty("input-state", state)
self.input_field.style().polish(self.input_field)
if self.label_widget:
self.label_widget.setProperty("state", state)
self.label_widget.style().polish(self.label_widget)
class GUIWidget(BaseWidget):
separator_height = 2
def create_ui(self):
entity_type = self.entity["type"]
if entity_type == "label":
self._create_label_ui()
elif entity_type in ("separator", "splitter"):
self._create_separator_ui()
else:
raise KeyError("Unknown GUI type {}".format(entity_type))
self.entity_widget.add_widget_to_layout(self)
def _create_label_ui(self):
self.setObjectName("LabelWidget")
label = self.entity["label"]
label_widget = QtWidgets.QLabel(label, self)
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(0, 5, 0, 5)
layout.addWidget(label_widget)
def _create_separator_ui(self):
splitter_item = QtWidgets.QWidget(self)
splitter_item.setObjectName("SplitterItem")
splitter_item.setMinimumHeight(self.separator_height)
splitter_item.setMaximumHeight(self.separator_height)
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(5, 5, 5, 5)
layout.addWidget(splitter_item)
def _on_entity_change(self):
pass
class DictImmutableKeysWidget(BaseWidget):
def create_ui(self):
self.input_fields = []
if not self.entity.is_dynamic_item and not self.entity.label:
self._ui_item_without_label()
else:
self._ui_item_or_as_widget()
for child_obj in self.entity.children:
self.input_fields.append(
create_ui_for_entity(child_obj, self)
)
self.entity_widget.add_widget_to_layout(self)
# any_visible = False
# for input_field in self.input_fields:
# if not input_field.hidden_by_role:
# any_visible = True
# break
#
# if not any_visible:
# self.hide()
def _ui_item_without_label(self):
self.setObjectName("DictInvisible")
self.body_widget = None
self.content_widget = self
self.content_layout = QtWidgets.QGridLayout(self)
self.content_layout.setContentsMargins(0, 0, 0, 0)
self.content_layout.setSpacing(5)
def _ui_item_or_as_widget(self):
content_widget = QtWidgets.QWidget(self)
if self.entity.is_dynamic_item:
content_widget.setObjectName("DictAsWidgetBody")
show_borders = str(int(self.show_borders))
content_widget.setProperty("show_borders", show_borders)
content_layout_margins = (5, 5, 5, 5)
main_layout_spacing = 5
body_widget = None
label_widget = QtWidgets.QLabel(self.entity.label)
else:
content_widget.setObjectName("ContentWidget")
if self.entity.highlight_content:
content_state = "hightlighted"
bottom_margin = 5
else:
content_state = ""
bottom_margin = 0
content_widget.setProperty("content_state", content_state)
content_layout_margins = (CHILD_OFFSET, 5, 0, bottom_margin)
main_layout_spacing = 0
body_widget = ExpandingWidget(self.entity.label, self)
label_widget = body_widget.label_widget
body_widget.set_content_widget(content_widget)
content_layout = QtWidgets.QGridLayout(content_widget)
content_layout.setContentsMargins(*content_layout_margins)
main_layout = QtWidgets.QHBoxLayout(self)
main_layout.setContentsMargins(0, 0, 0, 0)
main_layout.setSpacing(main_layout_spacing)
if not body_widget:
main_layout.addWidget(content_widget)
else:
main_layout.addWidget(body_widget)
self.label_widget = label_widget
self.body_widget = body_widget
self.content_widget = content_widget
self.content_layout = content_layout
if body_widget:
if len(self.input_fields) == 1 and self.checkbox_widget:
body_widget.hide_toolbox(hide_content=True)
elif self.entity.collapsible:
if not self.entity.collapsed:
body_widget.toggle_content()
else:
body_widget.hide_toolbox(hide_content=False)
def add_widget_to_layout(self, widget, label=None):
row = self.content_layout.rowCount()
if label:
label_widget = GridLabelWidget(label, widget)
label_widget.input_field = widget
widget.label_widget = label_widget
self.content_layout.addWidget(label_widget, row, 0, 1, 1)
self.content_layout.addWidget(widget, row, 1, 1, 1)
else:
self.content_layout.addWidget(widget, row, 0, 1, 2)
def _on_entity_change(self):
print("_on_entity_change", self.entity.path)
class BoolWidget(InputWidget):
def create_ui(self):
checkbox_height = self.style().pixelMetric(
QtWidgets.QStyle.PM_IndicatorHeight
)
self.input_field = NiceCheckbox(height=checkbox_height, parent=self)
spacer = QtWidgets.QWidget(self)
spacer.setAttribute(QtCore.Qt.WA_TranslucentBackground)
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(5)
layout.addWidget(self.input_field, 0)
layout.addWidget(spacer, 1)
self.setFocusProxy(self.input_field)
self.input_field.stateChanged.connect(self._on_value_change)
self.entity_widget.add_widget_to_layout(self, self.entity.label)
def _on_value_change(self):
if self.ignore_input_changes:
return
print("value changed", self.entity.path)
def _on_entity_change(self):
self.update_style()
class TextWidget(InputWidget):
def create_ui(self):
multiline = self.entity.multiline
if multiline:
self.input_field = QtWidgets.QPlainTextEdit(self)
else:
self.input_field = QtWidgets.QLineEdit(self)
self.set_value(self.entity.value)
placeholder_text = self.entity.placeholder_text
if placeholder_text:
self.input_field.setPlaceholderText(placeholder_text)
self.setFocusProxy(self.input_field)
layout_kwargs = {}
if multiline:
layout_kwargs["alignment"] = QtCore.Qt.AlignTop
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(5)
layout.addWidget(self.input_field, 1, **layout_kwargs)
self.input_field.textChanged.connect(self._on_value_change)
self.entity_widget.add_widget_to_layout(self, self.entity.label)
def set_value(self, text):
if self.entity.multiline:
self.input_field.setPlainText(text)
else:
self.input_field.setText(text)
def input_value(self):
if self.entity.multiline:
return self.input_field.toPlainText()
else:
return self.input_field.text()
def _on_value_change(self):
if self.ignore_input_changes:
return
text = self.input_value()
self.entity.set_value(text)
def _on_entity_change(self):
self.update_style()
class NumberWidget(InputWidget):
def create_ui(self):
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(5)
kwargs = {
"minimum": self.entity.minimum,
"maximum": self.entity.maximum,
"decimal": self.entity.decimal
}
self.input_field = NumberSpinBox(self, **kwargs)
self.input_field.setValue(self.entity.value)
self.setFocusProxy(self.input_field)
layout.addWidget(self.input_field, 1)
self.input_field.valueChanged.connect(self._on_value_change)
self.entity_widget.add_widget_to_layout(self, self.entity.label)
def _on_value_change(self):
if self.ignore_input_changes:
return
self.entity.set_value(self.input_field.value())
def _on_entity_change(self):
self.update_style()
class RawJsonInput(QtWidgets.QPlainTextEdit):
tab_length = 4
def __init__(self, *args, **kwargs):
super(RawJsonInput, self).__init__(*args, **kwargs)
self.setObjectName("RawJsonInput")
self.setTabStopDistance(
QtGui.QFontMetricsF(
self.font()
).horizontalAdvance(" ") * self.tab_length
)
def sizeHint(self):
document = self.document()
layout = document.documentLayout()
height = document.documentMargin() + 2 * self.frameWidth() + 1
block = document.begin()
while block != document.end():
height += layout.blockBoundingRect(block).height()
block = block.next()
hint = super(RawJsonInput, self).sizeHint()
hint.setHeight(height)
return hint
def set_value(self, value):
if value is NOT_SET:
value = ""
elif not isinstance(value, str):
try:
value = json.dumps(value, indent=4)
except Exception:
value = ""
self.setPlainText(value)
def json_value(self):
return json.loads(self.toPlainText())
def has_invalid_value(self):
try:
self.json_value()
return False
except Exception:
return True
def resizeEvent(self, event):
self.updateGeometry()
super(RawJsonInput, self).resizeEvent(event)
class RawJsonWidget(InputWidget):
def create_ui(self, label_widget=None):
layout = QtWidgets.QHBoxLayout(self)
layout.setContentsMargins(0, 0, 0, 0)
layout.setSpacing(5)
self.input_field = RawJsonInput(self)
self.input_field.setSizePolicy(
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.MinimumExpanding
)
self.setFocusProxy(self.input_field)
if not self.as_widget and not label_widget:
if self.label:
label_widget = QtWidgets.QLabel(self.label)
layout.addWidget(label_widget, 0, alignment=QtCore.Qt.AlignTop)
self.label_widget = label_widget
layout.addWidget(self.input_field, 1, alignment=QtCore.Qt.AlignTop)
self.input_field.textChanged.connect(self._on_value_change)
def _on_entity_change(self):
self.update_style()
def create_ui_for_entity(entity, entity_widget):
if isinstance(entity, GUIEntity):
return GUIWidget(entity, entity_widget)
elif isinstance(entity, DictImmutableKeysEntity):
return DictImmutableKeysWidget(entity, entity_widget)
elif isinstance(entity, BoolEntity):
return BoolWidget(entity, entity_widget)
elif isinstance(entity, TextEntity):
return TextWidget(entity, entity_widget)
elif isinstance(entity, NumberEntity):
return NumberWidget(entity, entity_widget)
elif isinstance(entity, RawJsonEntity):
return RawJsonWidget(entity, entity_widget)
# DictMutableKeysEntity,
# ListEntity,
# PathEntity,
# ListStrictEntity,
#
# EnumEntity,
# PathInput,
# RawJsonEntity
label = "<{}>: {} ({})".format(
entity.__class__.__name__, entity.path, entity.value
)
widget = QtWidgets.QLabel(label, entity_widget)
entity_widget.add_widget_to_layout(widget)
return widget