Skip to content

Commit 06f9be1

Browse files
Gustrynyalldawson
authored andcommitted
keep comments when save as preset or file, remove lines starting by "--" in DBManager
1 parent 2ed6c67 commit 06f9be1

File tree

1 file changed

+16
-5
lines changed

1 file changed

+16
-5
lines changed

python/plugins/db_manager/dlg_sql_window.py

+16-5
Original file line numberDiff line numberDiff line change
@@ -313,7 +313,7 @@ def executeSqlCompleted(self):
313313

314314
def executeSql(self):
315315

316-
sql = self._getSqlQuery()
316+
sql = self._getExecutableSqlQuery()
317317
if sql == "":
318318
return
319319

@@ -355,7 +355,7 @@ def _getSqlLayer(self, _filter):
355355
else:
356356
geomFieldName = None
357357

358-
query = self._getSqlQuery()
358+
query = self._getSqlExecutableQuery()
359359
if query == "":
360360
return None
361361

@@ -400,7 +400,7 @@ def loadSqlLayer(self):
400400
QgsProject.instance().addMapLayers([layer], True)
401401

402402
def fillColumnCombos(self):
403-
query = self._getSqlQuery()
403+
query = self._getExecutableSqlQuery()
404404
if query == "":
405405
return
406406

@@ -555,16 +555,27 @@ def createView(self):
555555
name, ok = QInputDialog.getText(None, self.tr("View Name"), self.tr("View name"))
556556
if ok:
557557
try:
558-
self.db.connector.createSpatialView(name, self._getSqlQuery())
558+
self.db.connector.createSpatialView(name, self._getExecutableSqlQuery())
559559
except BaseError as e:
560560
DlgDbError.showError(e, self)
561561

562562
def _getSqlQuery(self):
563563
sql = self.editSql.selectedText()
564564
if len(sql) == 0:
565-
sql = self.editSql.text().replace('\n', ' ').strip()
565+
sql = self.editSql.text()
566566
return sql
567567

568+
def _getExecutableSqlQuery(self):
569+
sql = self._getSqlQuery()
570+
571+
# Clean it up!
572+
lines = []
573+
for line in sql.split('\n'):
574+
if not line.strip().startswith('--'):
575+
lines.append(line)
576+
sql = ' '.join(lines)
577+
return sql.strip()
578+
568579
def uniqueChanged(self):
569580
# when an item is (un)checked, simply trigger an update of the combobox text
570581
self.uniqueTextChanged(None)

0 commit comments

Comments
 (0)