[processing] prevent division by zero in save selected features

algorithm (fix #16431)
qgis · Apr 27, 2017 · 1ac053f · 1ac053f · vagvaf · Apr 27, 2017
1 parent 459f486
commit 1ac053f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/plugins/processing/algs/qgis/SaveSelectedFeatures.py b/python/plugins/processing/algs/qgis/SaveSelectedFeatures.py
@@ -25,8 +25,9 @@
 
 __revision__ = '$Format:%H$'
 
-from qgis.core import (QgsApplication)
+from qgis.core import QgsApplication
 from processing.core.GeoAlgorithm import GeoAlgorithm
+from processing.core.GeoAlgorithmExecutionException import GeoAlgorithmExecutionException
 from processing.core.parameters import ParameterVector
 from processing.core.outputs import OutputVector
 from processing.tools import dataobjects
@@ -68,7 +69,10 @@ def processAlgorithm(self, context, feedback):
         writer = output.getVectorWriter(vectorLayer.fields(), vectorLayer.wkbType(), vectorLayer.crs(), context)
 
         features = vectorLayer.getSelectedFeatures()
-        total = 100.0 / int(vectorLayer.selectedFeatureCount())
+        if len(features) == 0:
+            raise GeoAlgorithmExecutionException(self.tr('There are no selected features in the input layer.'))
+
+        total = 100.0 / len(features)
         for current, feat in enumerate(features):
             writer.addFeature(feat)
             feedback.setProgress(int(current * total))