Fix #11951 - support for output directory in batch processing

wonder-sk · wonder-sk · commit c9682e550b7d · 2015-02-20T11:18:00.000+07:00
diff --git a/python/plugins/processing/gui/BatchOutputSelectionPanel.py b/python/plugins/processing/gui/BatchOutputSelectionPanel.py
@@ -38,6 +38,7 @@
 from processing.core.parameters import ParameterBoolean
 from processing.core.parameters import ParameterSelection
 from processing.core.parameters import ParameterFixedTable
+from processing.core.outputs import OutputDirectory
 
 
 class BatchOutputSelectionPanel(QWidget):
@@ -65,6 +66,10 @@ def __init__(self, output, alg, row, col, panel):
         self.setLayout(self.horizontalLayout)
 
     def showSelectionDialog(self):
+        if isinstance(self.output, OutputDirectory):
+            self.selectDirectory()
+            return
+
         filefilter = self.output.getFileFilter(self.alg)
         settings = QSettings()
         if settings.contains('/Processing/LastBatchOutputPath'):
@@ -117,6 +122,21 @@ def showSelectionDialog(self):
                 except:
                     pass
 
+    def selectDirectory(self):
+
+        settings = QSettings()
+        if settings.contains('/Processing/LastBatchOutputPath'):
+            lastDir = unicode(settings.value('/Processing/LastBatchOutputPath'))
+        else:
+            lastDir = ''
+
+        dirName = QFileDialog.getExistingDirectory(self,
+            self.tr('Select directory'), lastDir, QFileDialog.ShowDirsOnly)
+
+        if dirName:
+            self.table.cellWidget(self.row, self.col).setValue(dirName)
+            settings.setValue('/Processing/LastBatchOutputPath', dirName)
+
     def setValue(self, text):
         return self.text.setText(text)
 
