mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
6b82917715
always avoid using temporary file names, and then set this option as enabled by default. Also change the setting key so that existing users will also get the new default value. Fixes #32591
120 lines
5.4 KiB
Python
120 lines
5.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
Postprocessing.py
|
|
---------------------
|
|
Date : August 2012
|
|
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__ = 'August 2012'
|
|
__copyright__ = '(C) 2012, Victor Olaya'
|
|
|
|
import os
|
|
import traceback
|
|
from qgis.PyQt.QtWidgets import QApplication
|
|
from qgis.PyQt.QtCore import QCoreApplication
|
|
from qgis.core import (Qgis,
|
|
QgsProject,
|
|
QgsProcessingFeedback,
|
|
QgsProcessingUtils,
|
|
QgsMapLayerType,
|
|
QgsWkbTypes,
|
|
QgsMessageLog,
|
|
QgsProviderRegistry,
|
|
QgsExpressionContext,
|
|
QgsExpressionContextScope)
|
|
|
|
from processing.core.ProcessingConfig import ProcessingConfig
|
|
from processing.gui.RenderingStyles import RenderingStyles
|
|
|
|
|
|
def handleAlgorithmResults(alg, context, feedback=None, showResults=True, parameters={}):
|
|
wrongLayers = []
|
|
if feedback is None:
|
|
feedback = QgsProcessingFeedback()
|
|
feedback.setProgressText(QCoreApplication.translate('Postprocessing', 'Loading resulting layers'))
|
|
i = 0
|
|
|
|
for l, details in context.layersToLoadOnCompletion().items():
|
|
if feedback.isCanceled():
|
|
return False
|
|
|
|
if len(context.layersToLoadOnCompletion()) > 2:
|
|
# only show progress feedback if we're loading a bunch of layers
|
|
feedback.setProgress(100 * i / float(len(context.layersToLoadOnCompletion())))
|
|
try:
|
|
layer = QgsProcessingUtils.mapLayerFromString(l, context, typeHint=details.layerTypeHint)
|
|
if layer is not None:
|
|
details.setOutputLayerName(layer)
|
|
|
|
'''If running a model, the execution will arrive here when an algorithm that is part of
|
|
that model is executed. We check if its output is a final otuput of the model, and
|
|
adapt the output name accordingly'''
|
|
outputName = details.outputName
|
|
expcontext = QgsExpressionContext()
|
|
scope = QgsExpressionContextScope()
|
|
expcontext.appendScope(scope)
|
|
for out in alg.outputDefinitions():
|
|
if out.name() not in parameters:
|
|
continue
|
|
outValue = parameters[out.name()]
|
|
if hasattr(outValue, "sink"):
|
|
outValue = outValue.sink.valueAsString(expcontext)[0]
|
|
else:
|
|
outValue = str(outValue)
|
|
if outValue == l:
|
|
outputName = out.name()
|
|
break
|
|
style = None
|
|
if outputName:
|
|
style = RenderingStyles.getStyle(alg.id(), outputName)
|
|
if style is None:
|
|
if layer.type() == QgsMapLayerType.RasterLayer:
|
|
style = ProcessingConfig.getSetting(ProcessingConfig.RASTER_STYLE)
|
|
else:
|
|
if layer.geometryType() == QgsWkbTypes.PointGeometry:
|
|
style = ProcessingConfig.getSetting(ProcessingConfig.VECTOR_POINT_STYLE)
|
|
elif layer.geometryType() == QgsWkbTypes.LineGeometry:
|
|
style = ProcessingConfig.getSetting(ProcessingConfig.VECTOR_LINE_STYLE)
|
|
else:
|
|
style = ProcessingConfig.getSetting(ProcessingConfig.VECTOR_POLYGON_STYLE)
|
|
if style:
|
|
layer.loadNamedStyle(style)
|
|
|
|
details.project.addMapLayer(context.temporaryLayerStore().takeMapLayer(layer))
|
|
|
|
if details.postProcessor():
|
|
details.postProcessor().postProcessLayer(layer, context, feedback)
|
|
|
|
else:
|
|
wrongLayers.append(str(l))
|
|
except Exception:
|
|
QgsMessageLog.logMessage(QCoreApplication.translate('Postprocessing',
|
|
"Error loading result layer:") + "\n" + traceback.format_exc(),
|
|
'Processing', Qgis.Critical)
|
|
wrongLayers.append(str(l))
|
|
i += 1
|
|
|
|
feedback.setProgress(100)
|
|
|
|
if wrongLayers:
|
|
msg = QCoreApplication.translate('Postprocessing', "The following layers were not correctly generated.")
|
|
msg += "<ul>" + "".join(["<li>%s</li>" % lay for lay in wrongLayers]) + "</ul>"
|
|
msg += QCoreApplication.translate('Postprocessing',
|
|
"You can check the 'Log Messages Panel' in QGIS main window to find more information about the execution of the algorithm.")
|
|
feedback.reportError(msg)
|
|
|
|
return len(wrongLayers) == 0
|