mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
108 lines
4.4 KiB
Python
108 lines
4.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
PreconfiguredAlgorithmDialog.py
|
|
---------------------
|
|
Date : April 2016
|
|
Copyright : (C) 2012 by Victor Olaya
|
|
Email : volayaf at gmail dot com
|
|
***************************************************************************
|
|
* *
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation; either version 2 of the License, or *
|
|
* (at your option) any later version. *
|
|
* *
|
|
***************************************************************************
|
|
"""
|
|
|
|
__author__ = 'Victor Olaya'
|
|
__date__ = 'April 2016'
|
|
__copyright__ = '(C) 2016, Victor Olaya'
|
|
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
|
|
__revision__ = '$Format:%H$'
|
|
|
|
|
|
import os
|
|
import json
|
|
|
|
from processing.preconfigured.PreconfiguredUtils import algAsDict
|
|
from processing.preconfigured.PreconfiguredUtils import preconfiguredAlgorithmsFolder
|
|
from processing.gui.AlgorithmDialogBase import AlgorithmDialogBase
|
|
from processing.gui.AlgorithmDialog import AlgorithmDialog
|
|
from processing.tools import dataobjects
|
|
from qgis.PyQt.QtWidgets import QMessageBox, QVBoxLayout, QLabel, QLineEdit, QWidget
|
|
from qgis.PyQt.QtGui import QPalette, QColor
|
|
|
|
from qgis.core import QgsApplication
|
|
from qgis.gui import QgsMessageBar
|
|
|
|
|
|
class PreconfiguredAlgorithmDialog(AlgorithmDialog):
|
|
|
|
def __init__(self, alg, toolbox):
|
|
AlgorithmDialog.__init__(self, alg)
|
|
self.toolbox = toolbox
|
|
self.runButton().setText(self.tr("OK"))
|
|
self.tabWidget().removeTab(1)
|
|
self.settingsPanel = SettingsPanel()
|
|
self.tabWidget().addTab(self.settingsPanel, "Description")
|
|
|
|
def accept(self):
|
|
context = dataobjects.createContext()
|
|
try:
|
|
parameters = self.getParameterValues()
|
|
self.setOutputValues()
|
|
ok, msg = self.algorithm().checkParameterValues(parameters, context)
|
|
if not ok:
|
|
QMessageBox.warning(
|
|
self, self.tr('Unable to execute algorithm'), msg)
|
|
return
|
|
description = algAsDict(self.algorithm())
|
|
description["name"] = self.settingsPanel.txtName.text().strip()
|
|
description["group"] = self.settingsPanel.txtGroup.text().strip()
|
|
if not (description["name"] and description["group"]):
|
|
self.tabWidget().setCurrentIndex(self.tabWidget().count() - 1)
|
|
return
|
|
validChars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789:'
|
|
filename = ''.join(c for c in description["name"] if c in validChars).lower() + ".json"
|
|
filepath = os.path.join(preconfiguredAlgorithmsFolder(), filename)
|
|
with open(filepath, "w") as f:
|
|
json.dump(description, f)
|
|
QgsApplication.processingRegistry().providerById('preconfigured').refreshAlgorithms()
|
|
except AlgorithmDialogBase.InvalidParameterValue as e:
|
|
try:
|
|
self.buttonBox().accepted.connect(lambda: e.widget.setPalette(QPalette()))
|
|
palette = e.widget.palette()
|
|
palette.setColor(QPalette.Base, QColor(255, 255, 0))
|
|
e.widget.setPalette(palette)
|
|
self.messageBar().pushMessage("", self.tr('Missing parameter value: {0}').format(
|
|
e.parameter.description()),
|
|
level=Qgis.Warning, duration=5)
|
|
return
|
|
except:
|
|
QMessageBox.critical(self,
|
|
self.tr('Unable to execute algorithm'),
|
|
self.tr('Wrong or missing parameter values'))
|
|
self.close()
|
|
|
|
|
|
class SettingsPanel(QWidget):
|
|
|
|
def __init__(self):
|
|
QWidget.__init__(self)
|
|
layout = QVBoxLayout()
|
|
labelName = QLabel("Name")
|
|
labelGroup = QLabel("Group")
|
|
self.txtName = QLineEdit()
|
|
self.txtGroup = QLineEdit()
|
|
layout.addWidget(labelName)
|
|
layout.addWidget(self.txtName)
|
|
layout.addWidget(labelGroup)
|
|
layout.addWidget(self.txtGroup)
|
|
layout.addStretch()
|
|
self.setLayout(layout)
|