[processing] repair non-working random extract algorithms (fix #11571)

qgis · Nov 19, 2014 · 3c2d45f · 3c2d45f
1 parent 5e6f4c2
commit 3c2d45f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/python/plugins/processing/algs/qgis/RandomExtract.py b/python/plugins/processing/algs/qgis/RandomExtract.py
@@ -85,9 +85,8 @@ def processAlgorithm(self, progress):
 
         selran = random.sample(xrange(0, featureCount), value)
 
-        output = self.getOutputFromName(self.OUTPUT)
-        writer = output.getVectorWriter(layer.fields(),
-                layer.geometryType(), layer.crs())
+        writer = self.getOutputFromName(self.OUTPUT).getVectorWriter(
+            layer.pendingFields().toList(), layer.wkbType(), layer.crs())
 
         for (i, feat) in enumerate(features):
             if i in selran:

diff --git a/python/plugins/processing/algs/qgis/RandomExtractWithinSubsets.py b/python/plugins/processing/algs/qgis/RandomExtractWithinSubsets.py
@@ -91,10 +91,8 @@ def processAlgorithm(self, progress):
                         value and try again.")
             value = value / 100.0
 
-
-        output = self.getOutputFromName(self.OUTPUT)
-        writer = output.getVectorWriter(layer.fields(),
-                layer.geometryType(), layer.crs())
+        writer = self.getOutputFromName(self.OUTPUT).getVectorWriter(
+            layer.pendingFields().toList(), layer.wkbType(), layer.crs())
 
         selran = []
         current = 0