"""
***************************************************************************
GdalAlgorithmDialog.py
---------------------
Date : May 2015
Copyright : (C) 2015 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__ = 'May 2015'
__copyright__ = '(C) 2015, Victor Olaya'
import json
import os
from PyQt5.QtWidgets import QFileDialog
from qgis.PyQt.QtCore import QCoreApplication
from qgis.PyQt.QtWidgets import (QWidget,
QVBoxLayout,
QPushButton,
QLabel,
QPlainTextEdit,
QLineEdit,
QComboBox,
QCheckBox,
QTreeView,
QTreeWidget,
QTreeWidgetItem,
QSizePolicy,
QDialogButtonBox)
from qgis.core import (
QgsProcessingException,
QgsProcessingFeedback,
QgsProcessingParameterDefinition
)
from qgis.gui import (QgsMessageBar,
QgsProjectionSelectionWidget,
QgsProcessingAlgorithmDialogBase,
QgsProcessingLayerOutputDestinationWidget)
from processing.gui.AlgorithmDialog import AlgorithmDialog
from processing.gui.AlgorithmDialogBase import AlgorithmDialogBase
from processing.gui.ParametersPanel import ParametersPanel
from processing.gui.MultipleInputPanel import MultipleInputPanel
from processing.gui.NumberInputPanel import NumberInputPanel
from processing.gui.wrappers import WidgetWrapper
from processing.tools.dataobjects import createContext
from processing.tools.Widget.TreeWidget import Tree
from processing.tools.FileListPrintUtils import printStr
class GdalAlgorithmDialog(AlgorithmDialog):
def __init__(self, alg, parent=None):
super().__init__(alg, parent=parent)
self.mainWidget().parametersHaveChanged()
def getParametersPanel(self, alg, parent):
return GdalParametersPanel(parent, alg)
class GdalParametersPanel(ParametersPanel):
def __init__(self, parent, alg):
super().__init__(parent, alg)
self.dialog = parent
algname = alg.name()
w = QWidget()
layout = QVBoxLayout()
layout.setMargin(0)
layout.setSpacing(6)
gdalHidden = True
loop = True
if algname == "postgistogeoserver":
# 资源目录
label = QLabel()
label.setText(self.tr("资源目录树（必选）"))
layout.addWidget(label)
tree = Tree("zyml")
self.curTree = tree.initTreeWidget()
layout.addWidget(self.curTree)
self.curTree.itemSelectionChanged.connect(self.parametersHaveChanged)
gdalHidden = True
elif algname == "importvectorintopostgisdatabaseavailableconnections" or algname == "importrasterintopostgisdatabase" or algname == "gdbtopostgis":
# 行政区划
label = QLabel()
label.setText(self.tr("行政区划（必选）"))
layout.addWidget(label)
tree = Tree("xzqh")
self.curTree = tree.initTreeWidget()
layout.addWidget(self.curTree)
self.curTree.itemSelectionChanged.connect(self.parametersHaveChanged)
# 上传附件
pushBtn = QPushButton()
pushBtn.setText("选择附件")
pushBtn.clicked.connect(self.uploadChange)
layout.addWidget(pushBtn)
self.filelist = QPlainTextEdit()
self.filelist.setReadOnly(True)
layout.addWidget(self.filelist)
gdalHidden = True
elif algname == "postgisupdate":
label = QLabel()
label.setText(self.tr("预分析结果"))
layout.addWidget(label)
self.vectorupdate = QPlainTextEdit()
self.vectorupdate.setReadOnly(True)
layout.addWidget(self.vectorupdate)
gdalHidden = True
loop = False
elif algname == "postgisrestore" or algname == "spotfileupload" or algname == "spotfiledownload" or algname == "spotfiledelete" or algname == "licensemake":
gdalHidden = True
loop = False
label = QLabel()
label.setText(self.tr("控制台"))
label.setHidden(gdalHidden)
layout.addWidget(label)
self.text = QPlainTextEdit()
self.text.setReadOnly(True)
self.text.setHidden(gdalHidden)
layout.addWidget(self.text)
w.setLayout(layout)
self.addExtraWidget(w)
self.connectParameterSignals(loop=loop)
self.parametersHaveChanged()
def uploadChange(self):
file_paths, _ = QFileDialog.getOpenFileNames(self, '选择文件', '', 'All Files (*)')
fileliststr = ",".join(file_paths)
self.filelist.setPlainText(fileliststr)
printStr(fileliststr)
def connectParameterSignals(self, loop):
for wrapper in list(self.wrappers.values()):
wrapper.widgetValueHasChanged.connect(self.parametersHaveChanged)
# TODO - remove when all wrappers correctly emit widgetValueHasChanged!
# For compatibility with 3.x API, we need to check whether the wrapper is
# the deprecated WidgetWrapper class. If not, it's the newer
# QgsAbstractProcessingParameterWidgetWrapper class
# TODO QGIS 4.0 - remove
if issubclass(wrapper.__class__, WidgetWrapper):
w = wrapper.widget
else:
w = wrapper.wrappedWidget()
if loop == True:
self.connectWidgetChangedSignals(w)
for c in w.findChildren(QWidget):
self.connectWidgetChangedSignals(c)
def connectWidgetChangedSignals(self, w):
if isinstance(w, QLineEdit):
w.textChanged.connect(self.parametersHaveChanged)
elif isinstance(w, QComboBox):
w.currentIndexChanged.connect(self.parametersHaveChanged)
elif isinstance(w, QgsProjectionSelectionWidget):
w.crsChanged.connect(self.parametersHaveChanged)
elif isinstance(w, QCheckBox):
w.stateChanged.connect(self.parametersHaveChanged)
elif isinstance(w, MultipleInputPanel):
w.selectionChanged.connect(self.parametersHaveChanged)
elif isinstance(w, NumberInputPanel):
w.hasChanged.connect(self.parametersHaveChanged)
elif isinstance(w, QgsProcessingLayerOutputDestinationWidget):
w.destinationChanged.connect(self.parametersHaveChanged)
elif isinstance(w, QTreeWidget):
w.itemSelectionChanged.connect(self.parametersHaveChanged)
def parametersHaveChanged(self):
context = createContext()
feedback = QgsProcessingFeedback()
try:
# messy as all heck, but we don't want to call the dialog's implementation of
# createProcessingParameters as we want to catch the exceptions raised by the
# parameter panel instead...
parameters = {} if self.dialog.mainWidget() is None else self.dialog.mainWidget().createProcessingParameters()
for output in self.algorithm().destinationParameterDefinitions():
if not output.name() in parameters or parameters[output.name()] is None:
if not output.flags() & QgsProcessingParameterDefinition.FlagOptional:
parameters[output.name()] = self.tr("[temporary file]")
for p in self.algorithm().parameterDefinitions():
if p.flags() & QgsProcessingParameterDefinition.FlagHidden:
continue
if p.flags() & QgsProcessingParameterDefinition.FlagOptional and p.name() not in parameters:
continue
if p.name() not in parameters or not p.checkValueIsAcceptable(parameters[p.name()]):
# not ready yet
self.text.setPlainText('')
return
try:
b = self.contains_keys(self, ["curTree"])
if b == True:
# 获取选中的项
selectedItems = self.curTree.selectedItems()
if len(selectedItems) == 0:
self.algorithm().setSelectedValue("")
else:
for selectedItem in selectedItems:
v = selectedItem.text(1)
self.algorithm().setSelectedValue(v)
commands = self.algorithm().getConsoleCommands(parameters, context, feedback, executing=False)
b = self.contains_keys(self, ["vectorupdate"])
if b == True:
if len(commands) == 0:
self.vectorupdate.setPlainText("更新数据包通过预分析，可进行数据更新！")
else:
self.vectorupdate.setPlainText("\n".join(commands))
commands = [c for c in commands if c not in ['cmd.exe', '/C ']]
self.text.setPlainText(" ".join(commands))
except QgsProcessingException as e:
self.text.setPlainText(str(e))
except AlgorithmDialogBase.InvalidParameterValue as e:
self.text.setPlainText(self.tr("Invalid value for parameter '{0}'").format(e.parameter.description()))
except AlgorithmDialogBase.InvalidOutputExtension as e:
self.text.setPlainText(e.message)
def contains_keys(self, obj, keys):
if isinstance(obj, dict):
return all(key in obj.keys() for key in keys)
elif hasattr(type(obj), '__dict__'):
return all(key in obj.__dict__ for key in keys)
else:
raise ValueError("Invalid object type")