[db_manager] Fix #14796 - Quote values depending on data type in query builder

arnaud-morvan · slarosa · commit 8e5b28f3bb5d · 2016-05-29T23:59:03.000+02:00
diff --git a/python/plugins/db_manager/db_plugins/data_model.py b/python/plugins/db_manager/db_plugins/data_model.py
@@ -57,11 +57,16 @@ def columnCount(self, parent=None):
         return len(self._header)
 
     def data(self, index, role):
-        if role != Qt.DisplayRole and role != Qt.FontRole:
+        if role not in [Qt.DisplayRole,
+                        Qt.EditRole,
+                        Qt.FontRole]:
             return None
 
         val = self.getData(index.row(), index.column())
 
+        if role == Qt.EditRole:
+            return val
+
         if role == Qt.FontRole:  # draw NULL in italic
             if val is not None:
                 return None
diff --git a/python/plugins/db_manager/dlg_query_builder.py b/python/plugins/db_manager/dlg_query_builder.py
@@ -21,7 +21,7 @@
 Query builder dialog, based on the QSpatialite plugin (GPLv2+) by Romain Riviere
 """
 
-from qgis.PyQt.QtCore import QObject, QEvent
+from qgis.PyQt.QtCore import Qt, QObject, QEvent
 from qgis.PyQt.QtWidgets import QDialog, QMessageBox, QTextEdit
 
 from .ui.ui_DlgQueryBuilder import Ui_DbManagerQueryBuilderDlg as Ui_Dialog
@@ -287,10 +287,17 @@ def list_values(self):
         self.ui.values.setModel(model)
 
     def query_item(self, index):
-        queryWord = index.data()
-        queryWord = ' "%s"' % queryWord
-        if queryWord != '':
-            self.ui.where.insertPlainText(queryWord)
+        value = index.data(Qt.EditRole)
+
+        if value is None:
+            queryWord = u'NULL'
+        elif isinstance(value, (int, float)):
+            queryWord = unicode(value)
+        else:
+            queryWord = self.db.connector.quoteString(value)
+
+        if queryWord.strip() != '':
+            self.ui.where.insertPlainText(u' ' + queryWord)
             self.ui.where.setFocus()
 
     def use_rtree(self):
