######################################################################################################################
# Copyright (C) 2017-2021 Spine project consortium
# This file is part of Spine Toolbox.
# Spine Toolbox is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General
# Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option)
# any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
# without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General
# Public License for more details. You should have received a copy of the GNU Lesser General Public License along with
# this program. If not, see <http://www.gnu.org/licenses/>.
######################################################################################################################
"""
Classes for custom QDialogs to add edit and remove database items.
:author: M. Marin (KTH)
:date: 13.5.2018
"""
from functools import reduce
from PySide2.QtWidgets import QDialog, QDialogButtonBox, QHeaderView, QGridLayout
from PySide2.QtCore import Slot, Qt
from ...widgets.custom_editors import IconColorEditor
from ...widgets.custom_qtableview import CopyPasteTableView
from ...helpers import busy_effect
[docs]class ManageItemsDialogBase(QDialog):
def __init__(self, parent, db_mngr):
"""Init class.
Args:
parent (SpineDBEditor): data store widget
db_mngr (SpineDBManager)
"""
super().__init__(parent)
self.db_mngr = db_mngr
self.table_view = self.make_table_view()
self.table_view.horizontalHeader().setSectionResizeMode(QHeaderView.Interactive)
self.table_view.horizontalHeader().setStretchLastSection(True)
self.table_view.horizontalHeader().setMinimumSectionSize(120)
self.table_view.verticalHeader().setDefaultSectionSize(parent.default_row_height)
self.button_box = QDialogButtonBox(self)
self.button_box.setStandardButtons(QDialogButtonBox.Cancel | QDialogButtonBox.Ok)
layout = QGridLayout(self)
layout.addWidget(self.table_view)
layout.addWidget(self.button_box)
self.setAttribute(Qt.WA_DeleteOnClose)
[docs] def make_table_view(self):
return CopyPasteTableView(self)
[docs] def connect_signals(self):
"""Connect signals to slots."""
self.button_box.accepted.connect(self.accept)
self.button_box.rejected.connect(self.reject)
[docs] def resize_window_to_columns(self, height=None):
if height is None:
height = self.sizeHint().height()
slack = 64
margins = self.layout().contentsMargins()
self.resize(
slack
+ margins.left()
+ margins.right()
+ self.table_view.frameWidth() * 2
+ self.table_view.verticalHeader().width()
+ self.table_view.horizontalHeader().length(),
height,
)
[docs]class ManageItemsDialog(ManageItemsDialogBase):
"""A dialog with a CopyPasteTableView and a QDialogButtonBox. Base class for all
dialogs to query user's preferences for adding/editing/managing data items.
"""
def __init__(self, parent, db_mngr):
"""Init class.
Args:
parent (SpineDBEditor): data store widget
db_mngr (SpineDBManager)
"""
super().__init__(parent, db_mngr)
self.model = None
[docs] def connect_signals(self):
"""Connect signals to slots."""
super().connect_signals()
try:
self.table_view.itemDelegate().data_committed.connect(self.set_model_data)
except AttributeError:
pass
self.model.dataChanged.connect(self._handle_model_data_changed)
self.model.modelReset.connect(self._handle_model_reset)
@Slot("QModelIndex", "QModelIndex", "QVector")
[docs] def _handle_model_data_changed(self, top_left, bottom_right, roles):
"""Reimplement in subclasses to handle changes in model data."""
@Slot("QModelIndex", "QVariant")
[docs] def set_model_data(self, index, data):
"""Update model data."""
if data is None:
return
self.model.setData(index, data, Qt.EditRole)
@Slot()
[docs] def _handle_model_reset(self):
"""Resize columns and form."""
self.table_view.resizeColumnsToContents()
self.resize_window_to_columns()
[docs]class GetObjectClassesMixin:
"""Provides a method to retrieve object classes for AddObjectsDialog and AddRelationshipClassesDialog.
"""
[docs] def make_db_map_obj_cls_lookup(self):
return {
db_map: {x["name"]: x for x in self.db_mngr.get_items(db_map, "object_class")} for db_map in self.db_maps
}
[docs] def object_class_name_list(self, row):
"""Return a list of object_class names present in all databases selected for given row.
Used by `ManageObjectsDelegate`.
"""
db_column = self.model.header.index('databases')
db_names = self.model._main_data[row][db_column]
db_maps = iter(self.keyed_db_maps[x] for x in db_names.split(",") if x in self.keyed_db_maps)
db_map = next(db_maps, None)
if not db_map:
return []
# Initalize list from first db_map
object_class_name_list = list(self.db_map_obj_cls_lookup[db_map])
# Update list from remaining db_maps
for db_map in db_maps:
object_class_name_list = [x for x in self.db_map_obj_cls_lookup[db_map] if x in object_class_name_list]
return object_class_name_list
[docs]class GetObjectsMixin:
"""Provides a method to retrieve objects for AddRelationshipsDialog and EditRelationshipsDialog.
"""
[docs] def make_db_map_obj_lookup(self):
return {
db_map: {(x["class_id"], x["name"]): x for x in self.db_mngr.get_items(db_map, "object")}
for db_map in self.db_maps
}
[docs] def object_name_list(self, row, column):
"""Return a list of object names present in all databases selected for given row.
Used by `ManageRelationshipsDelegate`.
"""
db_column = self.model.header.index('databases')
db_names = self.model._main_data[row][db_column]
db_maps = [self.keyed_db_maps[x] for x in db_names.split(",") if x in self.keyed_db_maps]
rel_cls_key = (self.class_name, self.object_class_name_list)
object_name_lists = []
for db_map in db_maps:
relationship_classes = self.db_map_rel_cls_lookup[db_map]
if rel_cls_key not in relationship_classes:
continue
rel_cls = relationship_classes[rel_cls_key]
object_class_id_list = rel_cls["object_class_id_list"]
object_class_id_list = [int(x) for x in object_class_id_list.split(",")]
object_class_id = object_class_id_list[column]
objects = self.db_map_obj_lookup[db_map]
object_name_lists.append([name for (class_id, name) in objects if class_id == object_class_id])
if not object_name_lists:
return []
return list(reduce(lambda x, y: set(x) & set(y), object_name_lists))
[docs]class GetRelationshipClassesMixin:
"""Provides a method to retrieve relationships for AddRelationshipsDialog and EditRelationshipsDialog.
"""
[docs] def make_db_map_rel_cls_lookup(self):
return {
db_map: {
(x["name"], x["object_class_name_list"]): x
for x in self.db_mngr.get_items(db_map, "relationship_class")
}
for db_map in self.db_maps
}
[docs]class ShowIconColorEditorMixin:
"""Provides methods to show an `IconColorEditor` upon request.
"""
@busy_effect
[docs] def show_icon_color_editor(self, index):
editor = IconColorEditor(self)
editor.set_data(index.data(Qt.DisplayRole))
editor.accepted.connect(lambda index=index, editor=editor: self.set_model_data(index, editor.data()))
editor.show()