def set_combo_with_value(combo: QtWidgets.QComboBox, value):
"""
Searches all items of the given combo for the given value and changes the current index to that one.
:param combo:
:param value:
:return:
"""
combo.setCurrentIndex(combo.findData(value))
class RequirementEditor:
_editor: Union[None, ResourceRequirementEditor, ArrayRequirementEditor,
TemplateRequirementEditor]
def __init__(self,
parent: QWidget,
parent_layout: QVBoxLayout,
resource_database: ResourceDatabase,
*,
on_remove=None):
self.parent = parent
self.parent_layout = parent_layout
self.resource_database = resource_database
self._editor = None
self._last_resource = None
self._last_items = ()
self._last_comment = None
self.line_layout = QHBoxLayout()
self.line_layout.setAlignment(Qt.AlignLeft)
self.parent_layout.addLayout(self.line_layout)
if on_remove is not None:
self.remove_button = QtWidgets.QToolButton(parent)
self.remove_button.setText("X")
self.remove_button.setMaximumWidth(20)
self.remove_button.clicked.connect(on_remove)
self.line_layout.addWidget(self.remove_button)
else:
self.remove_button = None
self.requirement_type_combo = QComboBox(parent)
self.requirement_type_combo.addItem("Resource", ResourceRequirement)
self.requirement_type_combo.addItem("Or", RequirementOr)
self.requirement_type_combo.addItem("And", RequirementAnd)
if resource_database.requirement_template:
self.requirement_type_combo.addItem("Template",
RequirementTemplate)
self.requirement_type_combo.setMaximumWidth(75)
self.requirement_type_combo.activated.connect(
self._on_change_requirement_type)
self.line_layout.addWidget(self.requirement_type_combo)
def create_specialized_editor(self, requirement: Requirement):
self.requirement_type_combo.setCurrentIndex(
self.requirement_type_combo.findData(type(requirement)))
if isinstance(requirement, ResourceRequirement):
self._editor = ResourceRequirementEditor(self.parent,
self.line_layout,
self.resource_database,
requirement)
elif isinstance(requirement, RequirementArrayBase):
self._editor = ArrayRequirementEditor(self.parent,
self.parent_layout,
self.line_layout,
self.resource_database,
requirement)
elif isinstance(requirement, RequirementTemplate):
self._editor = TemplateRequirementEditor(self.parent,
self.line_layout,
self.resource_database,
requirement)
else:
raise RuntimeError(
f"Unknown requirement type: {type(requirement)} - {requirement}"
)
def _on_change_requirement_type(self):
current_requirement = self.current_requirement
self._editor.deleteLater()
if isinstance(current_requirement, ResourceRequirement):
self._last_resource = current_requirement
elif isinstance(current_requirement, RequirementArrayBase):
self._last_items = current_requirement.items
self._last_comment = current_requirement.comment
elif isinstance(current_requirement, RequirementTemplate):
pass
else:
raise RuntimeError(
f"Unknown requirement type: {type(current_requirement)} - {current_requirement}"
)
new_class = self.requirement_type_combo.currentData()
if new_class == ResourceRequirement:
if self._last_resource is None:
new_requirement = _create_default_resource_requirement(
self.resource_database)
else:
new_requirement = self._last_resource
elif new_class == RequirementTemplate:
new_requirement = _create_default_template_requirement(
self.resource_database)
else:
new_requirement = new_class(self._last_items, self._last_comment)
self.create_specialized_editor(new_requirement)
def deleteLater(self):
if self.remove_button is not None:
self.remove_button.deleteLater()
self.requirement_type_combo.deleteLater()
if self._editor is not None:
self._editor.deleteLater()
@property
def current_requirement(self) -> Requirement:
return self._editor.current_requirement
def combo_set_to_value(combo: QtWidgets.QComboBox, item):
combo.setCurrentIndex(combo.findData(item))
