Be smart when adding limit: followup #8092

qgis · Sep 22, 2020 · ef02c54 · ef02c54
1 parent 1c1febc
commit ef02c54
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/plugins/db_manager/db_plugins/postgis/connector.py b/python/plugins/db_manager/db_plugins/postgis/connector.py
@@ -121,7 +121,8 @@ def description(self):
             uri = QgsDataSourceUri(self.connection.uri())
 
             # TODO: make this part provider-agnostic
-            uri.setTable('(SELECT row_number() OVER () AS __rid__, * FROM (' + self.sql + ' LIMIT 1) as foo)')
+            sql = self.sql if self.sql.upper().find(' LIMIT ') >= 0 else self.sql + ' LIMIT 1 '
+            uri.setTable('(SELECT row_number() OVER () AS __rid__, * FROM (' + sql + ') as foo)')
             uri.setKeyColumn('__rid__')
             # TODO: fetch provider name from connection (QgsAbstractConnectionProvider)
             # TODO: re-use the VectorLayer for fetching rows in batch mode