uidocumenttools.py

Go to the documentation of this file.

# SPDX-License-Identifier: CC0-1.0
 
from . import documenttoolsdialog
try:
    from PyQt6.QtCore import Qt
    from PyQt6.QtWidgets import (QFormLayout, QListWidget, QAbstractItemView,
                                 QDialogButtonBox, QVBoxLayout, QFrame, QTabWidget,
                                 QPushButton, QAbstractScrollArea, QMessageBox)
except:
    from PyQt5.QtCore import Qt
    from PyQt5.QtWidgets import (QFormLayout, QListWidget, QAbstractItemView,
                                 QDialogButtonBox, QVBoxLayout, QFrame, QTabWidget,
                                 QPushButton, QAbstractScrollArea, QMessageBox)
from krita import Krita
import importlib
from builtins import i18n
 
class UIDocumentTools(object):
 
    def __init__(self):
        self.mainDialog = documenttoolsdialog.DocumentToolsDialog()
        self.mainLayout = QVBoxLayout(self.mainDialog)
        self.formLayout = QFormLayout()
        self.documentLayout = QVBoxLayout()
        self.refreshButton = QPushButton(i18n("Refresh"))
        self.widgetDocuments = QListWidget()
        self.tabTools = QTabWidget()
        self.buttonBox = QDialogButtonBox(
            QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel)
 
        self.kritaInstance = Krita.instance()
        self.documentsList = []
 
        self.refreshButton.clicked.connect(self.refreshButtonClickedrefreshButtonClicked)
        self.buttonBox.accepted.connect(self.confirmButtonconfirmButton)
        self.buttonBox.rejected.connect(self.mainDialog.close)
 
        self.mainDialog.setWindowModality(Qt.WindowModality.NonModal)
        self.widgetDocuments.setSelectionMode(QAbstractItemView.SelectionMode.MultiSelection)
        self.widgetDocuments.setSizeAdjustPolicy(
            QAbstractScrollArea.SizeAdjustPolicy.AdjustToContents)
 
    def initialize(self):
        self.loadDocuments()
        self.loadTools()
 
        self.documentLayout.addWidget(self.widgetDocuments)
        self.documentLayout.addWidget(self.refreshButton)
 
        self.formLayout.addRow(i18n("Documents:"), self.documentLayout)
        self.formLayout.addRow(self.tabTools)
 
        self.line = QFrame()
        self.line.setFrameShape(QFrame.Shape.HLine)
        self.line.setFrameShadow(QFrame.Shadow.Sunken)
 
        self.mainLayout.addLayout(self.formLayout)
        self.mainLayout.addWidget(self.line)
        self.mainLayout.addWidget(self.buttonBox)
 
        self.mainDialog.resize(500, 300)
        self.mainDialog.setWindowTitle(i18n("Document Tools"))
        self.mainDialog.setSizeGripEnabled(True)
        self.mainDialog.show()
        self.mainDialog.activateWindow()
 
    def loadTools(self):
        modulePath = 'documenttools.tools'
        toolsModule = importlib.import_module(modulePath)
        modules = []
 
        for classPath in toolsModule.ToolClasses:
            _module = classPath[:classPath.rfind(".")]
            _klass = classPath[classPath.rfind(".") + 1:]
            modules.append(dict(module='{0}.{1}'.format(modulePath, _module),
                                klass=_klass))
 
        for module in modules:
            m = importlib.import_module(module['module'])
            toolClass = getattr(m, module['klass'])
            obj = toolClass(self.mainDialog)
            self.tabTools.addTab(obj, obj.objectName())
 
    def loadDocuments(self):
        self.widgetDocuments.clear()
 
        self.documentsList = [
            document for document in self.kritaInstance.documents()
            if document.fileName()
        ]
 
        for document in self.documentsList:
            self.widgetDocuments.addItem(document.fileName())
 
    def refreshButtonClicked(self):
        self.loadDocuments()
 
    def confirmButton(self):
        selectedPaths = [
            item.text() for item in self.widgetDocuments.selectedItems()]
        selectedDocuments = [
            document for document in self.documentsList
            for path in selectedPaths if path == document.fileName()]
 
        self.msgBox = QMessageBox(self.mainDialog)
        if selectedDocuments:
            widget = self.tabTools.currentWidget()
            widget.adjust(selectedDocuments)
            self.msgBox.setText(
                i18n("The selected documents has been modified."))
        else:
            self.msgBox.setText(i18n("Select at least one document."))
        self.msgBox.exec()