Add a dialect select to ogrsql

om-henners · om-henners · commit 524f6ab460c5 · 2015-10-08T23:03:36.000+11:00
Add the ability to select sqlite or ogrsql dialects when executing SQL

Signed-off-by: Henry Walshaw &lt;henry.walshaw@gmail.com&gt;
diff --git a/python/plugins/processing/algs/gdal/ogrsql.py b/python/plugins/processing/algs/gdal/ogrsql.py
@@ -28,17 +28,21 @@
 from processing.core.GeoAlgorithmExecutionException import GeoAlgorithmExecutionException
 from processing.core.parameters import ParameterVector
 from processing.core.parameters import ParameterString
+from processing.core.parameters import ParameterSelection
 from processing.core.outputs import OutputVector
 
 from processing.algs.gdal.GdalUtils import GdalUtils
 from processing.algs.gdal.OgrAlgorithm import OgrAlgorithm
 
 
+DIALECTS = [None, 'ogrsql', 'sqlite']
+
 class OgrSql(OgrAlgorithm):
 
     INPUT = 'INPUT'
     OUTPUT = 'OUTPUT'
     SQL = 'SQL'
+    DIALECT = 'DIALECT'
 
     def defineCharacteristics(self):
         self.name, self.i18n_name = self.trAlgorithm('Execute SQL')
@@ -48,6 +52,12 @@ def defineCharacteristics(self):
                           [ParameterVector.VECTOR_TYPE_ANY], False))
         self.addParameter(ParameterString(self.SQL, self.tr('SQL'), ''))
 
+        self.addParameter(ParameterSelection(
+            self.DIALECT,
+            self.tr('Dialect'),
+            DIALECTS)
+        )
+
         self.addOutput(OutputVector(self.OUTPUT, self.tr('SQL result')))
 
     def getConsoleCommands(self):
@@ -60,6 +70,13 @@ def getConsoleCommands(self):
         arguments.append('-sql')
         arguments.append(sql)
 
+        dialectIdx = self.getParameterValue(self.DIALECT)
+        dialect = DIALECTS[dialectIdx]
+
+        if dialect:
+            arguments.append("-dialect")
+            arguments.append(dialect)
+
         output = self.getOutputFromName(self.OUTPUT)
         outFile = output.value
         arguments.append(outFile)
