class postgisQueryBuilder: def __init__(self, iface): # Save reference to the QGIS interface self.iface = iface # initialize plugin directory self.plugin_dir = os.path.dirname(__file__) # initialize locale locale = QSettings().value("locale/userLocale")[0:2] localePath = os.path.join(self.plugin_dir, 'i18n', 'postgisquerybuilder_{}.qm'.format(locale)) if os.path.exists(localePath): self.translator = QTranslator() self.translator.load(localePath) if qVersion() > '4.3.3': QCoreApplication.installTranslator(self.translator) #self.dlg = uic.loadUi( os.path.join( os.path.dirname( os.path.abspath( __file__ ) ), "ui_postgisquerybuilder.ui" ) ) self.querySet = querySet() self.PSQL = PSQL(self.iface) def eventsConnect(self): self.dlg.QueryType.activated.connect(self.setQueryType) self.dlg.checkAutoCompiled.stateChanged.connect(self.queryGen) self.dlg.LAYERa.activated.connect(self.setLAYERa) self.dlg.SCHEMAb.activated.connect(self.setSCHEMAb) self.dlg.FILTERSCHEMA.activated.connect(self.setFILTERSCHEMA) self.dlg.LAYERb.activated.connect(self.setLAYERb) self.dlg.FIELD.activated.connect(self.setFIELD) self.dlg.SPATIALREL.activated.connect(self.setSPATIALREL) self.dlg.SPATIALRELNOT.stateChanged.connect(self.setSPATIALRELNOT) self.dlg.checkCreateView.clicked.connect(self.checkCreateView) self.dlg.BUFFERRADIUS.textChanged.connect(self.setBUFFERRADIUS) self.dlg.ButtonRun.clicked.connect(self.runQuery) self.dlg.ButtonReset.clicked.connect(self.resetForm) self.dlg.ButtonHelp.clicked.connect(self.helpDialog) self.dlg.fieldsListA.clicked.connect(self.setFieldsList) self.dlg.fieldsListB.clicked.connect(self.setFieldsList) self.dlg.checkMaterialized.clicked.connect(self.setMaterialized) #self.dlg.AddToMapButton.clicked.connect(self.layerAddToMap) #self.dlg.GetInfoButton.clicked.connect(self.layerGetTable) #self.dlg.DeleteButton.clicked.connect(self.layerDelete) #self.dlg.RefreshButton.clicked.connect(self.layerRefresh) #self.dlg.convertToTableButton.clicked.connect(self.convertToTable) self.dlg.JOIN.activated.connect(self.setJOIN) self.dlg.FIELDb.activated.connect(self.setFIELDb) self.dlg.tabWidget.currentChanged.connect(self.tabChangedHub) self.dlg.KEYFIELD.activated.connect(self.keyGeomFieldsChanged) self.dlg.KEYFIELD.editTextChanged.connect(self.keyGeomFieldsChanged) self.dlg.GEOMETRYFIELD.activated.connect(self.keyGeomFieldsChanged) self.dlg.GEOMETRYFIELD.editTextChanged.connect( self.keyGeomFieldsChanged) self.dlg.LAYERaAllFields.clicked.connect(self.selectAllFields) self.dlg.LayerList.itemDoubleClicked.connect(self.useForQuery) self.dlg.LayerList.customContextMenuRequested.connect( self.layerContextMenu) self.dlg.LayerList.itemSelectionChanged.connect(self.saveForQuery) self.dlg.schemaAdd.clicked.connect(self.addNewSchema) self.dlg.HelpLink.clicked.connect(self.helpLinkOpen) def eventsDisconnect(self): self.dlg.QueryType.activated.disconnect(self.setQueryType) self.dlg.checkAutoCompiled.stateChanged.disconnect(self.queryGen) self.dlg.LAYERa.activated.disconnect(self.setLAYERa) self.dlg.SCHEMAb.activated.disconnect(self.setSCHEMAb) self.dlg.FILTERSCHEMA.activated.disconnect(self.setFILTERSCHEMA) self.dlg.LAYERb.activated.disconnect(self.setLAYERb) self.dlg.FIELD.activated.disconnect(self.setFIELD) self.dlg.SPATIALREL.activated.disconnect(self.setSPATIALREL) self.dlg.SPATIALRELNOT.stateChanged.disconnect(self.setSPATIALRELNOT) self.dlg.checkCreateView.clicked.disconnect(self.checkCreateView) self.dlg.BUFFERRADIUS.textChanged.disconnect(self.setBUFFERRADIUS) self.dlg.ButtonRun.clicked.disconnect(self.runQuery) self.dlg.ButtonReset.clicked.disconnect(self.resetForm) self.dlg.ButtonHelp.clicked.disconnect(self.helpDialog) #self.dlg.AddToMapButton.clicked.disconnect(self.layerAddToMap) #self.dlg.GetInfoButton.clicked.disconnect(self.layerGetTable) #self.dlg.DeleteButton.clicked.disconnect(self.layerDelete) #self.dlg.RefreshButton.clicked.disconnect(self.layerRefresh) #self.dlg.convertToTableButton.clicked.disconnect(self.convertToTable) self.dlg.JOIN.activated.disconnect(self.setJOIN) self.dlg.FIELDb.activated.disconnect(self.setFIELDb) self.dlg.fieldsListA.clicked.disconnect(self.setFieldsList) self.dlg.fieldsListB.clicked.disconnect(self.setFieldsList) self.dlg.tabWidget.currentChanged.disconnect(self.tabChangedHub) self.dlg.KEYFIELD.activated.disconnect(self.keyGeomFieldsChanged) self.dlg.KEYFIELD.editTextChanged.disconnect(self.keyGeomFieldsChanged) self.dlg.GEOMETRYFIELD.activated.disconnect(self.keyGeomFieldsChanged) self.dlg.GEOMETRYFIELD.editTextChanged.disconnect( self.keyGeomFieldsChanged) self.dlg.LAYERaAllFields.clicked.disconnect(self.selectAllFields) self.dlg.LayerList.itemDoubleClicked.disconnect(self.useForQuery) self.dlg.LayerList.customContextMenuRequested.disconnect( self.layerContextMenu) self.dlg.LayerList.itemSelectionChanged.disconnect(self.saveForQuery) self.dlg.schemaAdd.clicked.disconnect(self.addNewSchema) self.dlg.HelpLink.clicked.disconnect(self.helpLinkOpen) def initGui(self): # Create action that will start plugin configuration self.action = QAction( QIcon(":/plugins/postgisquerybuilder/querybuilderlogo.png"), u"postgisQueryBuilder", self.iface.mainWindow()) # connect the action to the run method self.action.triggered.connect(self.run) # Add toolbar button and menu item self.iface.addToolBarIcon(self.action) self.iface.addPluginToDatabaseMenu(u"&postgisQueryBuilder", self.action) #setup Docked widget self.dlg = postgisQueryBuilderDialog() self.PQBdockwidget = QDockWidget("postgisQueryBuilder", self.iface.mainWindow()) self.PQBdockwidget.setObjectName("postgisQueryBuilder") self.PQBdockwidget.setWidget(self.dlg) self.PQBdockwidget.setAllowedAreas(Qt.RightDockWidgetArea | Qt.LeftDockWidgetArea) self.iface.addDockWidget(Qt.RightDockWidgetArea, self.PQBdockwidget) #defaults #self.dlg.GEOMETRYFIELD.setText("the_geom") #self.dlg.KEYFIELD.setText("ogc_fid") self.predefinedLayer = None #hide Temp slots self.dlg.USERFIELD.hide() self.dlg.USERFIELDLabel.hide() self.dlg.PASSWORDFIELD.hide() self.dlg.PASSWORDFIELDLabel.hide() #init self.getPSQLConnections() self.populateGui() self.eventsConnect() self.toTableDlg = convertToTableDialog(self) def populateGui(self): self.populateComboBox(self.dlg.QueryType, self.querySet.getQueryLabels(), "Select query type", True) self.populateComboBox(self.dlg.JOIN, [ "INNER JOIN", "CROSS JOIN", "RIGHT OUTER JOIN", "LEFT OUTER JOIN", "FULL OUTER JOIN" ], "Select", True) self.populateComboBox(self.dlg.SPATIALREL, self.querySet.getSpatialRelationships(), "Select spatial relationship", True) self.populateComboBox( self.dlg.KEYFIELD, ["ogc_fid", "id", "fid", "gid", "FID", "GID", "ID"], "", None) self.populateComboBox(self.dlg.GEOMETRYFIELD, ["the_geom", "geom", "GEOM", "geometry"], "", None) self.dlg.checkAutoCompiled.setChecked(True) self.dlg.tabWidget.setCurrentIndex(0) #self.recurseChild(self.dlg,"") # def layerAddToMap(self): # for rowList in range(0,self.dlg.LayerList.count()): # rowCheckbox = self.dlg.LayerList.item(rowList) # #take only selected attributes by checkbox # if rowCheckbox.checkState() == Qt.Checked: # self.PSQL.loadView(rowCheckbox.text(),self.dlg.GEOMETRYFIELD.text(),self.dlg.KEYFIELD.text()) # self.uncheckList(self.dlg.LayerList) def layerForQuery(self, listItem): print listItem def layerContextMenu(self, listItem): self.predefinedLayer = None for rowSel in (self.dlg.LayerList.selectedItems()): self.selectedLayer = rowSel.text() contextMenu = QMenu() self.layerNameAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ self.selectedLayer) contextMenu.addSeparator() self.countFeatureAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Features Count: "+self.PSQL.getFeatureCount(self.selectedLayer)) self.relationTypeAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Relation Type: "+self.PSQL.getRelationType(self.selectedLayer)) self.geometryTypeAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Geometry Type: "+self.PSQL.getGeometryType(\ self.selectedLayer,suggestion = self.dlg.GEOMETRYFIELD.currentText())) self.autoKeyAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Detected key field: %s" % self.PSQL.guessKeyField(\ self.selectedLayer,suggestion = self.dlg.KEYFIELD.currentText()) or "undefined") self.autoGeometryAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Detected geom field: %s" % self.PSQL.guessGeometryField(\ self.selectedLayer,suggestion = self.dlg.GEOMETRYFIELD.currentText()) or "undefined") self.nativeSRIDAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Native SRID: EPSG:"+self.PSQL.getSRID(\ self.selectedLayer,suggestion = self.dlg.GEOMETRYFIELD.currentText())) contextMenu.addSeparator() self.useForQueryAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Use for query") self.useForQueryAction.triggered.connect(self.useForQuery) self.layerAddToMapAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","addToLayer.png")),\ "Add layer to map canvas") self.layerAddToMapAction.triggered.connect(self.layerAddToMap) self.probeKeyGeomAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","probeKeyGeom.png")),\ "Auto detect primary key and geometry fields") self.probeKeyGeomAction.triggered.connect(self.probeKeyGeom) self.layerGetTableAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","layerGetTable.png")),\ "View as data table") self.layerGetTableAction.triggered.connect(self.layerGetTable) self.renameObjectAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","renameObject.png")),\ "Rename") self.renameObjectAction.triggered.connect(self.renameObject) self.layerDeleteAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","layerDelete.png")),\ "Delete") self.layerDeleteAction.triggered.connect(self.layerDelete) self.moveObjectAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","moveObject.png")),\ "Move to another schema") self.moveObjectAction.triggered.connect(self.moveObject) if self.PSQL.isView( self.selectedLayer) or self.PSQL.isMaterializedView( self.selectedLayer): self.convertToTableAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","convertToTable.png")),\ "Convert view to table") self.convertToTableAction.triggered.connect(self.convertToTable) if self.PSQL.isView(self.selectedLayer): self.editViewAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","layerDelete.png")),\ "Edit view definition") self.editViewAction.triggered.connect(self.editViewDefinition) if self.PSQL.isMaterializedView(self.selectedLayer): self.layerRefreshAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","layerRefresh.png")),\ "Refresh materialized view") self.layerRefreshAction.triggered.connect(self.layerRefresh) contextMenu.exec_(QCursor.pos()) def editViewDefinition(self): sqlView = 'CREATE OR REPLACE VIEW "%s"."%s" AS ' % ( self.PSQL.getSchema(), self.selectedLayer) + self.PSQL.getViewDef( self.selectedLayer) self.resetDialog() self.dlg.QueryName.setText(self.selectedLayer) self.dlg.checkCreateView.setChecked(True) self.dlg.AddToMap.setChecked(False) if self.PSQL.isMaterializedView(self.selectedLayer): self.dlg.checkMaterialized.setChecked(True) self.dlg.checkAutoCompiled.setChecked(False) self.dlg.QueryResult.setPlainText( self.querySet.formatSqlStatement(sqlView)) self.dlg.tabWidget.setCurrentIndex(3) def useForQuery(self): self.predefinedLayer = self.selectedLayer self.resetDialog() self.dlg.tabWidget.setCurrentIndex(1) def saveForQuery(self): for rowSel in (self.dlg.LayerList.selectedItems()): self.selectedLayer = rowSel.text() self.predefinedLayer = self.selectedLayer def layerAddToMap(self): keyGuess = self.PSQL.guessKeyField( self.selectedLayer, suggestion=self.dlg.KEYFIELD.currentText()) geomGuess = self.PSQL.guessGeometryField( self.selectedLayer, suggestion=self.dlg.GEOMETRYFIELD.currentText()) self.PSQL.loadView(self.selectedLayer, geomGuess, keyGuess) def probeKeyGeom(self): self.populateComboBox(self.dlg.KEYFIELD, self.PSQL.getKeyFields(self.selectedLayer), "", None) self.populateComboBox(self.dlg.GEOMETRYFIELD, self.PSQL.getGeometryFields(self.selectedLayer), "", None) def layerGetTable(self): self.PSQL.tableResultGen(self.selectedLayer, "", self.dlg.TableResult) self.dlg.tabWidget.setCurrentIndex(4) def convertToTable(self): if self.PSQL.isMaterializedView( self.selectedLayer) or self.PSQL.isView(self.selectedLayer): self.toTableDlg.ask(self.selectedLayer) def renameObject(self): newLayer = renameDialog.rename(self.selectedLayer) if newLayer: error = self.PSQL.renameLayer(self.selectedLayer, newLayer) if error: QMessageBox.information(None, "ERROR:", error) else: self.updateLayerMenu() def moveObject(self): schemas = self.PSQL.getSchemas() schemas.remove(self.PSQL.getSchema()) schemas.insert(0, self.PSQL.getSchema()) newSchema = moveDialog.getNewSchema(self) if newSchema: error = self.PSQL.moveLayer(self.selectedLayer, newSchema) if error: QMessageBox.information(None, "ERROR:", error) else: self.updateLayerMenu() def layerDelete(self, cascade=None): msg = "Are you sure you want to delete layer '%s' from schema '%s' ?" % ( self.selectedLayer, self.PSQL.getSchema()) reply = QMessageBox.question(None, 'Message', msg, QMessageBox.Yes, QMessageBox.No) if reply == QMessageBox.Yes: result = self.PSQL.deleteLayer(self.selectedLayer) if result: if "DROP ... CASCADE" in result: msg = result + "\n\nLayer '%s' has dependencies. Do you want to remove all recursively ?" % self.selectedLayer reply = QMessageBox.question(None, 'Message', msg, QMessageBox.Yes, QMessageBox.No) if reply == QMessageBox.Yes: result = self.PSQL.deleteLayer(self.selectedLayer, cascade=True) if result: QMessageBox.information(None, "ERROR:", result) else: print "CASCADE DELETED", self.selectedLayer else: QMessageBox.information(None, "ERROR:", result) else: print "DELETED", self.selectedLayer pass self.populateLayerMenu() def layerRefresh(self): if self.PSQL.isMaterializedView(self.selectedLayer): self.PSQL.refreshMaterializedView(self.selectedLayer) def recurseChild(self, slot, tab): # for testing: prints qt object tree for child in slot.children(): #print tab,"|",child.objectName() if child.children() != []: self.recurseChild(child, tab + " ") def uncheckList(self, slot): for row in range(0, slot.count()): self.dlg.LayerList.item(row).setCheckState(Qt.Unchecked) def addNewSchema(self): newSchema = renameDialog.rename("new_schema") error = self.PSQL.addSchema(newSchema) if error: QMessageBox.information(None, "ERROR:", error) else: self.populateComboBox(self.dlg.DBSchema, self.PSQL.getSchemas(), "Select schema", True) def disableDialogSlot(self, slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.setDisabled(True) def hideDialogSlot(self, slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.hide() def showDialogSlot(self, slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.show() def enableDialogSlot(self, slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.setEnabled(True) def clearDialogSlot(self, slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.clear() def checkCreateView(self): #method called when checkbox createview is clicked if self.dlg.checkCreateView.checkState(): self.dlg.QueryName.setEnabled(True) self.dlg.checkMaterialized.setEnabled(True) else: self.dlg.QueryName.setDisabled(True) self.dlg.checkMaterialized.setDisabled(True) if self.querySet.testQueryParametersCheckList(): self.queryGen() def helpDialog(self): print "called help" if self.dlg.QueryType.currentText() != "Select query type": QDesktopServices.openUrl(QUrl(self.querySet.getHelp())) #webbrowser.open_new(self.querySet.getHelp()) def helpLinkOpen(self): QDesktopServices.openUrl( QUrl( "https://geogear.wordpress.com/postgisquerybuilder-v1-6-cheatsheet/" )) def setMaterialized(self): self.queryGen() def keyGeomFieldsChanged(self): self.querySet.setParameter("GEOMETRYFIELD", self.dlg.GEOMETRYFIELD.currentText()) self.querySet.setParameter("KEYFIELD", self.dlg.KEYFIELD.currentText()) test = None try: test = self.querySet.testQueryParametersCheckList() except: pass if test: self.queryGen() def tabChangedHub(self, tab): #print "TAB:",tab if tab == 0: try: self.updateLayerMenu() except: pass try: self.keyGeomFieldsChanged() except: pass if tab == 3: self.queryGen() if tab == 2: self.updateOrderBy() elif tab == 5: self.updateHistoryLog() def updateOrderBy(self): if self.dlg.LAYERa.currentText()[:6] != "Select": try: self.populateComboBox( self.dlg.orderBy, self.PSQL.getFieldsContent(self.dlg.LAYERa.currentText()), " ", True) except: pass def focusOnQuery(self): self.dlg.tabWidget.setCurrentIndex(3) def updateHistoryLog(self): historyFile = os.path.join(os.path.dirname(__file__), "validSql.log") if os.path.exists(historyFile): in_file = open(historyFile, "r") self.dlg.historyLog.setPlainText(in_file.read()) self.dlg.historyLog.moveCursor(QTextCursor.End) def populateComboBox(self, combo, list, predef, sort): #procedure to fill specified combobox with provided list combo.blockSignals(True) combo.clear() model = QStandardItemModel(combo) predefInList = None for elem in list: try: item = QStandardItem(unicode(elem)) except TypeError: item = QStandardItem(str(elem)) model.appendRow(item) if elem == predef: predefInList = elem if sort: model.sort(0) combo.setModel(model) if predef != "": if predefInList: combo.setCurrentIndex(combo.findText(predefInList)) else: combo.insertItem(0, predef) combo.setCurrentIndex(0) combo.blockSignals(False) def loadSVG(self, svgName): self.dlg.DiagPanel.show() svgFile = os.path.join(os.path.dirname(os.path.abspath(__file__)), "svg", svgName + ".svg") #print svgFile item = QGraphicsSvgItem(svgFile) scene = QGraphicsScene() scene.addItem(item) self.dlg.DiagPanel.setScene(scene) self.dlg.DiagPanel.fitInView(item, Qt.KeepAspectRatio) #,Qt.KeepAspectRatio def setLAYERa(self): #called when LAYERa is activated if self.dlg.LAYERa.currentText()[:6] == "Select": return self.querySet.setParameter("LAYERa", self.dlg.LAYERa.currentText()) #try to guess layer geometry field autoGeometry = self.PSQL.guessGeometryField( self.dlg.LAYERa.currentText(), suggestion=self.dlg.GEOMETRYFIELD.currentText()) self.querySet.setParameter("GEOMETRYFIELDa", autoGeometry) if self.querySet.testQueryParametersCheckList(): self.queryGen() self.populateComboBox( self.dlg.FIELD, self.PSQL.getFieldsContent(self.dlg.LAYERa.currentText()), "Select field", True) if not self.PSQL.testIfFieldExist( self.dlg.LAYERa.currentText(), self.querySet.getParameter("KEYFIELD")): self.querySet.setFIDFIELD() self.addListToFieldTable(self.dlg.fieldsListA, self.PSQL.getFieldsContent( self.dlg.LAYERa.currentText()), True, exclude=autoGeometry) self.populateFilterTable() self.setSCHEMAb() def selectAllFields(self): for row in range(0, self.dlg.fieldsListA.count()): if self.dlg.fieldsListA.item(row).flags() & Qt.ItemIsEnabled: self.dlg.fieldsListA.item(row).setCheckState( self.dlg.LAYERaAllFields.checkState()) self.setFieldsList() def setSCHEMAb(self): #called when SCHEMAb is activated self.querySet.setParameter("SCHEMAb", self.dlg.SCHEMAb.currentText()) self.populateComboBox( self.dlg.LAYERb, self.PSQL.getLayers(schema=self.dlg.SCHEMAb.currentText()), "Select layer", True) def setFILTERSCHEMA(self): #called when FILTERSCHEMA is activated self.dlg.filterTable.setCurrentSchema( self.dlg.FILTERSCHEMA.currentText()) def setLAYERb(self): #called when LAYERb is activated if self.dlg.LAYERb.currentText()[:6] == "Select": return self.querySet.setParameter("LAYERb", self.dlg.LAYERb.currentText()) #try to guess layer geometry field autoGeometry = self.PSQL.guessGeometryField( self.dlg.LAYERb.currentText(), suggestion=self.dlg.GEOMETRYFIELD.currentText()) self.querySet.setParameter("GEOMETRYFIELDb", autoGeometry) if self.querySet.testQueryParametersCheckList(): self.queryGen() self.addListToFieldTable( self.dlg.fieldsListB, self.PSQL.getFieldsContent(self.dlg.LAYERb.currentText()), True) self.populateComboBox( self.dlg.FIELDb, self.PSQL.getFieldsContent(self.dlg.LAYERb.currentText()), "Select field", True) def addListToFieldTable(self, fieldSlot, fl, check, exclude=None): #called to populate field list for WHERE statement wdgt = fieldSlot wdgt.clear() #print "LAYERLIST:",check for row in fl: item = QListWidgetItem() if check: item.setFlags(item.flags() | Qt.ItemIsUserCheckable) item.setCheckState(Qt.Unchecked) item.setText(row) #print row if self.PSQL.isTable(row): item.setIcon(QIcon(":/plugins/postgisquerybuilder/iT.png")) elif self.PSQL.isView(row): item.setIcon(QIcon(":/plugins/postgisquerybuilder/iV.png")) elif self.PSQL.isMaterializedView(row): item.setIcon(QIcon(":/plugins/postgisquerybuilder/iM.png")) #disable geometryfield and auto set checked/unchecked wdgt.addItem(item) if check and exclude and row == exclude: #self.geoQuery and if self.geoQuery: item.setCheckState(Qt.Unchecked) else: item.setCheckState(Qt.Checked) item.setFlags(item.flags() & (~Qt.ItemIsEnabled)) #item.setFlags(item.flags() | Qt.ItemIsEnabled) def setFieldsList(self): # procedure to resume selected fields to SELECT statements items = [] for index in xrange(self.dlg.fieldsListA.count()): if self.dlg.fieldsListA.item(index).checkState() == Qt.Checked: items.append('"' + self.dlg.LAYERa.currentText() + '"."' + self.dlg.fieldsListA.item(index).text() + '"') for index in xrange(self.dlg.fieldsListB.count()): if self.dlg.fieldsListB.item(index).checkState() == Qt.Checked: items.append('"' + self.dlg.LAYERb.currentText() + '"."' + self.dlg.fieldsListB.item(index).text() + '"') #print "SELECTED FIELDS",items self.querySet.setFieldsSet(items) if self.querySet.testQueryParametersCheckList(): self.queryGen() def setFIELD(self): if self.dlg.FIELD.currentText()[:6] == "Select": return self.querySet.setParameter( "FIELD", '"' + self.dlg.LAYERa.currentText() + '"."' + self.dlg.FIELD.currentText() + '"') self.querySet.setParameter("SIMPLEFIELD", self.dlg.FIELD.currentText()) fType = self.PSQL.getFieldsType(self.querySet.getParameter("LAYERa"), self.dlg.FIELD.currentText()) fType = fType[:4] #print fType if ((fType == "char") or (fType == "text") or (fType == "varc")): self.tDelimiter = "'" else: self.tDelimiter = "" if self.querySet.testQueryParametersCheckList(): self.queryGen() def setFIELDb(self): if self.dlg.FIELDb.currentText()[:6] == "Select": return self.querySet.setParameter( "FIELDb", '"' + self.dlg.LAYERb.currentText() + '"."' + self.dlg.FIELDb.currentText() + '"') if self.querySet.testQueryParametersCheckList(): self.queryGen() def setSPATIALREL(self): if self.dlg.SPATIALREL.currentText()[:6] == "Select": return self.querySet.setParameter("SPATIALREL", self.dlg.SPATIALREL.currentText()) if self.querySet.testQueryParametersCheckList(): self.queryGen() def setBUFFERRADIUS(self): self.querySet.setParameter("BUFFERRADIUS", self.dlg.BUFFERRADIUS.text()) if self.querySet.testQueryParametersCheckList(): self.queryGen() def setSPATIALRELNOT(self): if self.dlg.SPATIALRELNOT.isChecked(): self.querySet.setParameter("SPATIALRELNOT", "NOT ") else: self.querySet.setParameter("SPATIALRELNOT", " ") if self.querySet.testQueryParametersCheckList(): self.queryGen() def setJOIN(self): self.querySet.setParameter("JOIN", self.dlg.JOIN.currentText()) if self.querySet.testQueryParametersCheckList(): self.queryGen() def setQueryName(self): try: self.querySet.testIfQueryDefined() except: return self.querySet.setParameter("VIEWNAME", self.dlg.QueryName.text()) if self.querySet.testQueryParametersCheckList(): self.dlg.QueryResult.setPlainText( self.querySet.getQueryParsed( self.dlg.checkCreateView.checkState())) def queryGen(self): if self.dlg.checkCreateView.checkState(): self.enableDialogSlot("QueryName") if not self.querySet.testIfQueryDefined(): return if self.dlg.checkAutoCompiled.checkState(): if self.dlg.checkMaterialized.checkState(): self.querySet.setParameter("MATERIALIZED", "MATERIALIZED") else: self.querySet.setParameter("MATERIALIZED", "") self.enableDialogSlot("QueryResult") self.enableDialogSlot("checkCreateView") self.enableDialogSlot("AddToMap") try: qName = self.querySet.getNameParsed() except: qName = self.dlg.QueryName.text() self.dlg.QueryName.setText(qName) self.querySet.setParameter("VIEWNAME", qName) if self.dlg.filterTable.testIfSintaxOk(): self.querySet.setParameter( "WHERE", self.dlg.filterTable.getWhereStatement()) self.querySet.setParameter( "SPATIALFROM", self.dlg.filterTable.getSpatialFilterLayers( schema=self.PSQL.getSchema())) else: self.querySet.setParameter("WHERE", "") if self.dlg.orderBy.currentText() != " ": self.querySet.setParameter( "ORDERBY", 'ORDER BY "' + self.dlg.orderBy.currentText() + '"') else: self.querySet.setParameter("ORDERBY", "") self.dlg.QueryResult.setPlainText( self.querySet.getQueryParsed( self.dlg.checkCreateView.checkState())) self.dlg.QueryName.textChanged.connect(self.setQueryName) def setQueryType(self, line): theQ = self.dlg.QueryType.currentText() if theQ[:6] == "Select": return self.loadSVG(theQ.replace(" ", "_")) #self.querySet.resetParameters() self.resetDialog() self.dlg.summaryBox.show() self.querySet.setCurrentQuery(theQ) if theQ[:2] == "ST": self.geoQuery = True else: self.geoQuery = None for slot in self.querySet.getRequiredSlots(): #print slot #self.enableDialogSlot(slot) self.showDialogSlot(slot) self.showDialogSlot(slot + "Label") self.showDialogSlot(slot + "AllFields") #print self.querySet.testQueryParametersCheckList() #simulate click on checkbox to set required slot self.dlg.SPATIALRELNOT.setCheckState(Qt.Checked) self.dlg.SPATIALRELNOT.setCheckState(Qt.Unchecked) self.dlg.Helper.setText(theQ + ":\n" + self.querySet.getDescription()) self.loadSVG(theQ.replace(" ", "_")) if self.predefinedLayer: self.dlg.LAYERa.setCurrentIndex( self.dlg.LAYERa.findText(self.predefinedLayer)) self.setLAYERa() def resetDialog(self): self.eventsDisconnect() self.clearAllDialogs() self.querySet.resetParameters() #self.disableQueryDefSlot() self.hideQueryDefSlot() self.loadSVG("Select_query_type") self.dlg.summaryBox.hide() try: tables = self.PSQL.getLayers() except AttributeError: pass else: self.populateComboBox(self.dlg.LAYERa, tables, "Select Layer", True) self.populateComboBox(self.dlg.LAYERb, tables, "Select Layer", True) self.geoQuery = None self.populateComboBox(self.dlg.orderBy, [" "], "", True) self.eventsConnect() def hideQueryDefSlot(self): toHide=["BUFFERRADIUS","FIELD",\ "SPATIALREL","SPATIALRELNOT","LAYERaLabel","BUFFERRADIUSLabel",\ "FIELD","FIELDLabel","SPATIALRELLabel", "SPATIALRELNOTLabel","fieldsListALabel","fieldsListBLabel",\ "FIELDb","FIELDbLabel","JOIN","JOINLabel",\ "LAYERa","SCHEMAb","LAYERb","LAYERbLabel","summaryBox","LAYERaAllFields","fieldsListA","fieldsListB"] for slot in toHide: self.hideDialogSlot(slot) def clearQueryDefSlot(self): toClear=["LAYERa","LAYERb","BUFFERRADIUS","FIELD","FIELDb",\ "SPATIALREL","fieldsListA","fieldsListB",\ "JOIN"] for slot in toClear: self.clearDialogSlot(slot) def disableQueryDefSlot(self): toDisable=["LAYERa","LAYERb","BUFFERRADIUS","FIELD","FIELDb",\ "SPATIALREL",\ "SPATIALRELNOT","fieldsListA","fieldsListB", "JOIN"] for slot in toDisable: self.disableDialogSlot(slot) def clearAllDialogs(self): self.dlg.LAYERa.clear() self.dlg.LAYERb.clear() self.dlg.BUFFERRADIUS.clear() self.dlg.FIELD.clear() self.dlg.QueryResult.clear() self.dlg.QueryName.clear() self.dlg.fieldsListA.clear() self.dlg.fieldsListB.clear() self.dlg.TableResult.clear() self.dlg.SPATIALRELNOT.setCheckState(Qt.Unchecked) self.dlg.LAYERaAllFields.setCheckState(Qt.Unchecked) def getPSQLConnections(self): try: self.dlg.PSQLConnection.activated.disconnect(self.setConnection) except: pass conn = self.PSQL.getConnections() self.populateComboBox(self.dlg.PSQLConnection, conn, "Select connection", True) self.hideQueryDefSlot() self.dlg.PSQLConnection.activated.connect(self.setConnection) def closeDialog(self): self.resetDialog() self.dlg.hide() def resetForm(self): self.resetDialog() self.getPSQLConnections() self.populateGui() self.dlg.tabWidget.setCurrentIndex(0) def setConnection(self): if self.dlg.PSQLConnection.currentText()[:6] == "Select": return self.PSQL.setConnection(self.dlg.PSQLConnection.currentText()) #print "SCHEMAS",self.PSQL.getSchemas() schemas = self.PSQL.getSchemas() self.populateComboBox(self.dlg.DBSchema, schemas, "Select schema", True) self.dlg.DBSchema.activated.connect(self.loadPSQLLayers) for r in range(0, self.dlg.DBSchema.count()): if self.dlg.DBSchema.itemText(r) == "public": self.dlg.DBSchema.setCurrentIndex(r) self.dlg.DBSchema.removeItem(0) self.loadPSQLLayers() #clone dbschema in schemab self.populateComboBox(self.dlg.SCHEMAb, [ self.dlg.DBSchema.itemText(i) for i in range(self.dlg.DBSchema.count()) ], self.dlg.DBSchema.currentText(), True) self.querySet.setParameter("SCHEMAb", self.dlg.SCHEMAb.currentText()) self.populateComboBox(self.dlg.FILTERSCHEMA, [ self.dlg.DBSchema.itemText(i) for i in range(self.dlg.DBSchema.count()) ], self.dlg.DBSchema.currentText(), True) def loadPSQLLayers(self): if self.dlg.DBSchema.currentText() != "Select schema": self.PSQL.setSchema(self.dlg.DBSchema.currentText()) self.populateGui() self.resetDialog() #self.dlg.tabWidget.setCurrentIndex(1) self.querySet.setSchema(self.dlg.DBSchema.currentText()) self.populateComboBox(self.dlg.GEOMETRYFIELD, self.PSQL.scanLayersForGeometry(), "", None) self.populateComboBox(self.dlg.KEYFIELD, self.PSQL.scanLayersForPrimaryKey(), "", None) #sync schemab con dbschema self.dlg.SCHEMAb.setCurrentIndex(self.dlg.DBSchema.currentIndex()) self.querySet.setParameter("SCHEMAb", self.dlg.SCHEMAb.currentText()) self.populateLayerMenu() def populateFilterTable(self): self.dlg.filterTable.populateFilterTable(self.PSQL, self.dlg.LAYERa.currentText()) def testSignal(self, v1, v2): #print "catch:", v1,v2 pass def populateLayerMenu(self): self.addListToFieldTable(self.dlg.LayerList, self.PSQL.getLayers(), None) def updateLayerMenu(self): if self.dlg.DBSchema.currentText() != "Select schema": self.addListToFieldTable(self.dlg.LayerList, self.PSQL.getLayers(), None) def runQuery(self): #method to run generated query if self.dlg.filterTable.testIfSintaxOk(): if self.dlg.AddToMap.checkState(): if self.dlg.checkCreateView.checkState( ): #self.querySet.getParameter("VIEWNAME") self.PSQL.submitQuery(self.dlg.QueryName.text(), self.dlg.QueryResult.toPlainText()) self.PSQL.loadView( self.dlg.QueryName.text(), self.querySet.getParameter("GEOMETRYFIELD"), self.querySet.getParameter("KEYFIELD")) else: self.PSQL.loadSql( self.dlg.QueryName.text(), self.dlg.QueryResult.toPlainText(), self.querySet.getParameter("GEOMETRYFIELD"), self.querySet.getParameter("KEYFIELD")) else: if self.dlg.checkCreateView.checkState(): self.PSQL.submitQuery(self.dlg.QueryName.text(), self.dlg.QueryResult.toPlainText()) sql = 'SELECT * FROM "%s"."%s"' % ( self.PSQL.getSchema(), self.dlg.QueryName.text()) print sql rows = self.PSQL.tableResultGen( self.dlg.LAYERa.currentText(), sql, self.dlg.TableResult) self.PSQL.loadedLayerRefresh(self.dlg.QueryName.text()) else: rows = self.PSQL.tableResultGen( self.dlg.LAYERa.currentText(), self.dlg.QueryResult.toPlainText(), self.dlg.TableResult) self.dlg.labelRowsNumber.setText("Total rows: " + str(rows)) self.dlg.tabWidget.setCurrentIndex(4) else: QMessageBox.information(None, "FILTER ERROR:", "The Filter table is malformed") self.dlg.tabWidget.setCurrentIndex(2) def unload(self): # Remove the plugin menu item and icon self.iface.removePluginMenu(u"&postgisQueryBuilder", self.action) self.iface.removeToolBarIcon(self.action) self.iface.removeDockWidget(self.PQBdockwidget) # run method that performs all the real work def run(self): # show/hide the widget if self.PQBdockwidget.isVisible(): self.PQBdockwidget.hide() else: self.PQBdockwidget.show() self.getPSQLConnections()
class postgisQueryBuilder: def __init__(self, iface): # Save reference to the QGIS interface self.iface = iface # initialize plugin directory self.plugin_dir = os.path.dirname(__file__) # initialize locale locale = QSettings().value("locale/userLocale")[0:2] localePath = os.path.join(self.plugin_dir, 'i18n', 'postgisquerybuilder_{}.qm'.format(locale)) if os.path.exists(localePath): self.translator = QTranslator() self.translator.load(localePath) if qVersion() > '4.3.3': QCoreApplication.installTranslator(self.translator) #self.dlg = uic.loadUi( os.path.join( os.path.dirname( os.path.abspath( __file__ ) ), "ui_postgisquerybuilder.ui" ) ) self.querySet = querySet() self.PSQL = PSQL(self.iface) def eventsConnect(self): self.dlg.QueryType.activated.connect(self.setQueryType) self.dlg.checkAutoCompiled.stateChanged.connect(self.queryGen) self.dlg.LAYERa.activated.connect(self.setLAYERa) self.dlg.SCHEMAb.activated.connect(self.setSCHEMAb) self.dlg.FILTERSCHEMA.activated.connect(self.setFILTERSCHEMA) self.dlg.LAYERb.activated.connect(self.setLAYERb) self.dlg.FIELD.activated.connect(self.setFIELD) self.dlg.SPATIALREL.activated.connect(self.setSPATIALREL) self.dlg.SPATIALRELNOT.stateChanged.connect(self.setSPATIALRELNOT) self.dlg.checkCreateView.clicked.connect(self.checkCreateView) self.dlg.BUFFERRADIUS.textChanged.connect(self.setBUFFERRADIUS) self.dlg.ButtonRun.clicked.connect(self.runQuery) self.dlg.ButtonReset.clicked.connect(self.resetForm) self.dlg.ButtonHelp.clicked.connect(self.helpDialog) self.dlg.fieldsListA.clicked.connect(self.setFieldsList) self.dlg.fieldsListB.clicked.connect(self.setFieldsList) self.dlg.checkMaterialized.clicked.connect(self.setMaterialized) #self.dlg.AddToMapButton.clicked.connect(self.layerAddToMap) #self.dlg.GetInfoButton.clicked.connect(self.layerGetTable) #self.dlg.DeleteButton.clicked.connect(self.layerDelete) #self.dlg.RefreshButton.clicked.connect(self.layerRefresh) #self.dlg.convertToTableButton.clicked.connect(self.convertToTable) self.dlg.JOIN.activated.connect(self.setJOIN) self.dlg.FIELDb.activated.connect(self.setFIELDb) self.dlg.tabWidget.currentChanged.connect(self.tabChangedHub) self.dlg.KEYFIELD.activated.connect(self.keyGeomFieldsChanged) self.dlg.KEYFIELD.editTextChanged.connect(self.keyGeomFieldsChanged) self.dlg.GEOMETRYFIELD.activated.connect(self.keyGeomFieldsChanged) self.dlg.GEOMETRYFIELD.editTextChanged.connect(self.keyGeomFieldsChanged) self.dlg.LAYERaAllFields.clicked.connect(self.selectAllFields) self.dlg.LayerList.itemDoubleClicked.connect(self.useForQuery) self.dlg.LayerList.customContextMenuRequested.connect(self.layerContextMenu) self.dlg.LayerList.itemSelectionChanged.connect(self.saveForQuery) self.dlg.schemaAdd.clicked.connect(self.addNewSchema) self.dlg.HelpLink.clicked.connect(self.helpLinkOpen) def eventsDisconnect(self): self.dlg.QueryType.activated.disconnect(self.setQueryType) self.dlg.checkAutoCompiled.stateChanged.disconnect(self.queryGen) self.dlg.LAYERa.activated.disconnect(self.setLAYERa) self.dlg.SCHEMAb.activated.disconnect(self.setSCHEMAb) self.dlg.FILTERSCHEMA.activated.disconnect(self.setFILTERSCHEMA) self.dlg.LAYERb.activated.disconnect(self.setLAYERb) self.dlg.FIELD.activated.disconnect(self.setFIELD) self.dlg.SPATIALREL.activated.disconnect(self.setSPATIALREL) self.dlg.SPATIALRELNOT.stateChanged.disconnect(self.setSPATIALRELNOT) self.dlg.checkCreateView.clicked.disconnect(self.checkCreateView) self.dlg.BUFFERRADIUS.textChanged.disconnect(self.setBUFFERRADIUS) self.dlg.ButtonRun.clicked.disconnect(self.runQuery) self.dlg.ButtonReset.clicked.disconnect(self.resetForm) self.dlg.ButtonHelp.clicked.disconnect(self.helpDialog) #self.dlg.AddToMapButton.clicked.disconnect(self.layerAddToMap) #self.dlg.GetInfoButton.clicked.disconnect(self.layerGetTable) #self.dlg.DeleteButton.clicked.disconnect(self.layerDelete) #self.dlg.RefreshButton.clicked.disconnect(self.layerRefresh) #self.dlg.convertToTableButton.clicked.disconnect(self.convertToTable) self.dlg.JOIN.activated.disconnect(self.setJOIN) self.dlg.FIELDb.activated.disconnect(self.setFIELDb) self.dlg.fieldsListA.clicked.disconnect(self.setFieldsList) self.dlg.fieldsListB.clicked.disconnect(self.setFieldsList) self.dlg.tabWidget.currentChanged.disconnect(self.tabChangedHub) self.dlg.KEYFIELD.activated.disconnect(self.keyGeomFieldsChanged) self.dlg.KEYFIELD.editTextChanged.disconnect(self.keyGeomFieldsChanged) self.dlg.GEOMETRYFIELD.activated.disconnect(self.keyGeomFieldsChanged) self.dlg.GEOMETRYFIELD.editTextChanged.disconnect(self.keyGeomFieldsChanged) self.dlg.LAYERaAllFields.clicked.disconnect(self.selectAllFields) self.dlg.LayerList.itemDoubleClicked.disconnect(self.useForQuery) self.dlg.LayerList.customContextMenuRequested.disconnect(self.layerContextMenu) self.dlg.LayerList.itemSelectionChanged.disconnect(self.saveForQuery) self.dlg.schemaAdd.clicked.disconnect(self.addNewSchema) self.dlg.HelpLink.clicked.disconnect(self.helpLinkOpen) def initGui(self): # Create action that will start plugin configuration self.action = QAction( QIcon(":/plugins/postgisquerybuilder/querybuilderlogo.png"), u"postgisQueryBuilder", self.iface.mainWindow()) # connect the action to the run method self.action.triggered.connect(self.run) # Add toolbar button and menu item self.iface.addToolBarIcon(self.action) self.iface.addPluginToDatabaseMenu(u"&postgisQueryBuilder", self.action) #setup Docked widget self.dlg = postgisQueryBuilderDialog() self.PQBdockwidget=QDockWidget("postgisQueryBuilder" , self.iface.mainWindow() ) self.PQBdockwidget.setObjectName("postgisQueryBuilder") self.PQBdockwidget.setWidget(self.dlg) self.PQBdockwidget.setAllowedAreas(Qt.RightDockWidgetArea | Qt.LeftDockWidgetArea) self.iface.addDockWidget( Qt.RightDockWidgetArea, self.PQBdockwidget) #defaults #self.dlg.GEOMETRYFIELD.setText("the_geom") #self.dlg.KEYFIELD.setText("ogc_fid") self.predefinedLayer = None #hide Temp slots self.dlg.USERFIELD.hide() self.dlg.USERFIELDLabel.hide() self.dlg.PASSWORDFIELD.hide() self.dlg.PASSWORDFIELDLabel.hide() #init self.getPSQLConnections() self.populateGui() self.eventsConnect() self.toTableDlg = convertToTableDialog(self) def populateGui(self): self.populateComboBox(self.dlg.QueryType,self.querySet.getQueryLabels(),"Select query type",True) self.populateComboBox(self.dlg.JOIN,["INNER JOIN","CROSS JOIN","RIGHT OUTER JOIN","LEFT OUTER JOIN","FULL OUTER JOIN"],"Select",True) self.populateComboBox(self.dlg.SPATIALREL,self.querySet.getSpatialRelationships(),"Select spatial relationship",True) self.populateComboBox(self.dlg.KEYFIELD,["ogc_fid","id","fid","gid","FID","GID","ID"],"",None) self.populateComboBox(self.dlg.GEOMETRYFIELD,["the_geom","geom","GEOM","geometry"],"",None) self.dlg.checkAutoCompiled.setChecked(True) self.dlg.tabWidget.setCurrentIndex(0) #self.recurseChild(self.dlg,"") # def layerAddToMap(self): # for rowList in range(0,self.dlg.LayerList.count()): # rowCheckbox = self.dlg.LayerList.item(rowList) # #take only selected attributes by checkbox # if rowCheckbox.checkState() == Qt.Checked: # self.PSQL.loadView(rowCheckbox.text(),self.dlg.GEOMETRYFIELD.text(),self.dlg.KEYFIELD.text()) # self.uncheckList(self.dlg.LayerList) def layerForQuery(self,listItem): print listItem def layerContextMenu(self,listItem): self.predefinedLayer = None for rowSel in (self.dlg.LayerList.selectedItems()): self.selectedLayer = rowSel.text() contextMenu = QMenu() self.layerNameAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ self.selectedLayer) contextMenu.addSeparator() self.countFeatureAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Features Count: "+self.PSQL.getFeatureCount(self.selectedLayer)) self.relationTypeAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Relation Type: "+self.PSQL.getRelationType(self.selectedLayer)) self.geometryTypeAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Geometry Type: "+self.PSQL.getGeometryType(\ self.selectedLayer,suggestion = self.dlg.GEOMETRYFIELD.currentText())) self.autoKeyAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Detected key field: %s" % self.PSQL.guessKeyField(\ self.selectedLayer,suggestion = self.dlg.KEYFIELD.currentText()) or "undefined") self.autoGeometryAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Detected geom field: %s" % self.PSQL.guessGeometryField(\ self.selectedLayer,suggestion = self.dlg.GEOMETRYFIELD.currentText()) or "undefined") self.nativeSRIDAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Native SRID: EPSG:"+self.PSQL.getSRID(\ self.selectedLayer,suggestion = self.dlg.GEOMETRYFIELD.currentText())) contextMenu.addSeparator() self.useForQueryAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","useForQuery.png")),\ "Use for query") self.useForQueryAction.triggered.connect(self.useForQuery) self.layerAddToMapAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","addToLayer.png")),\ "Add layer to map canvas") self.layerAddToMapAction.triggered.connect(self.layerAddToMap) self.probeKeyGeomAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","probeKeyGeom.png")),\ "Auto detect primary key and geometry fields") self.probeKeyGeomAction.triggered.connect(self.probeKeyGeom) self.layerGetTableAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","layerGetTable.png")),\ "View as data table") self.layerGetTableAction.triggered.connect(self.layerGetTable) self.renameObjectAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","renameObject.png")),\ "Rename") self.renameObjectAction.triggered.connect(self.renameObject) self.layerDeleteAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","layerDelete.png")),\ "Delete") self.layerDeleteAction.triggered.connect(self.layerDelete) self.moveObjectAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","moveObject.png")),\ "Move to another schema") self.moveObjectAction.triggered.connect(self.moveObject) if self.PSQL.isView(self.selectedLayer) or self.PSQL.isMaterializedView(self.selectedLayer): self.convertToTableAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","convertToTable.png")),\ "Convert view to table") self.convertToTableAction.triggered.connect(self.convertToTable) if self.PSQL.isView(self.selectedLayer): self.editViewAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","layerDelete.png")),\ "Edit view definition") self.editViewAction.triggered.connect(self.editViewDefinition) if self.PSQL.isMaterializedView(self.selectedLayer): self.layerRefreshAction = contextMenu.addAction(QIcon(os.path.join(self.plugin_dir,"icons","layerRefresh.png")),\ "Refresh materialized view") self.layerRefreshAction.triggered.connect(self.layerRefresh) contextMenu.exec_(QCursor.pos()) def editViewDefinition(self): sqlView = 'CREATE OR REPLACE VIEW "%s"."%s" AS ' % (self.PSQL.getSchema(),self.selectedLayer) + self.PSQL.getViewDef(self.selectedLayer) self.resetDialog() self.dlg.QueryName.setText(self.selectedLayer) self.dlg.checkCreateView.setChecked(True) self.dlg.AddToMap.setChecked(False) if self.PSQL.isMaterializedView(self.selectedLayer): self.dlg.checkMaterialized.setChecked(True) self.dlg.checkAutoCompiled.setChecked(False) self.dlg.QueryResult.setPlainText(self.querySet.formatSqlStatement(sqlView)) self.dlg.tabWidget.setCurrentIndex(3) def useForQuery(self): self.predefinedLayer = self.selectedLayer self.resetDialog() self.dlg.tabWidget.setCurrentIndex(1) def saveForQuery(self): for rowSel in (self.dlg.LayerList.selectedItems()): self.selectedLayer = rowSel.text() self.predefinedLayer = self.selectedLayer def layerAddToMap(self): keyGuess = self.PSQL.guessKeyField(self.selectedLayer,suggestion=self.dlg.KEYFIELD.currentText()) geomGuess = self.PSQL.guessGeometryField(self.selectedLayer,suggestion=self.dlg.GEOMETRYFIELD.currentText()) self.PSQL.loadView(self.selectedLayer,geomGuess,keyGuess) def probeKeyGeom(self): self.populateComboBox(self.dlg.KEYFIELD,self.PSQL.getKeyFields(self.selectedLayer),"",None) self.populateComboBox(self.dlg.GEOMETRYFIELD,self.PSQL.getGeometryFields(self.selectedLayer),"",None) def layerGetTable(self): self.PSQL.tableResultGen(self.selectedLayer,"",self.dlg.TableResult) self.dlg.tabWidget.setCurrentIndex(4) def convertToTable(self): if self.PSQL.isMaterializedView(self.selectedLayer) or self.PSQL.isView(self.selectedLayer): self.toTableDlg.ask(self.selectedLayer) def renameObject(self): newLayer = renameDialog.rename(self.selectedLayer) if newLayer: error = self.PSQL.renameLayer(self.selectedLayer,newLayer) if error: QMessageBox.information(None, "ERROR:", error) else: self.updateLayerMenu() def moveObject(self): schemas = self.PSQL.getSchemas() schemas.remove(self.PSQL.getSchema()) schemas.insert(0,self.PSQL.getSchema()) newSchema = moveDialog.getNewSchema(self) if newSchema: error = self.PSQL.moveLayer(self.selectedLayer,newSchema) if error: QMessageBox.information(None, "ERROR:", error) else: self.updateLayerMenu() def layerDelete(self, cascade = None): msg = "Are you sure you want to delete layer '%s' from schema '%s' ?" % (self.selectedLayer,self.PSQL.getSchema()) reply = QMessageBox.question(None, 'Message', msg, QMessageBox.Yes, QMessageBox.No) if reply == QMessageBox.Yes: result = self.PSQL.deleteLayer(self.selectedLayer) if result: if "DROP ... CASCADE" in result: msg = result+"\n\nLayer '%s' has dependencies. Do you want to remove all recursively ?" % self.selectedLayer reply = QMessageBox.question(None, 'Message', msg, QMessageBox.Yes, QMessageBox.No) if reply == QMessageBox.Yes: result = self.PSQL.deleteLayer(self.selectedLayer,cascade = True) if result: QMessageBox.information(None, "ERROR:", result) else: print "CASCADE DELETED", self.selectedLayer else: QMessageBox.information(None, "ERROR:", result) else: print "DELETED", self.selectedLayer pass self.populateLayerMenu() def layerRefresh(self): if self.PSQL.isMaterializedView(self.selectedLayer): self.PSQL.refreshMaterializedView(self.selectedLayer) def recurseChild(self,slot,tab): # for testing: prints qt object tree for child in slot.children(): #print tab,"|",child.objectName() if child.children() != []: self.recurseChild(child,tab + " ") def uncheckList(self,slot): for row in range(0,slot.count()): self.dlg.LayerList.item(row).setCheckState(Qt.Unchecked) def addNewSchema(self): newSchema = renameDialog.rename("new_schema") error = self.PSQL.addSchema(newSchema) if error: QMessageBox.information(None, "ERROR:", error) else: self.populateComboBox(self.dlg.DBSchema,self.PSQL.getSchemas(),"Select schema",True) def disableDialogSlot(self,slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.setDisabled(True) def hideDialogSlot(self,slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.hide() def showDialogSlot(self,slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.show() def enableDialogSlot(self,slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.setEnabled(True) def clearDialogSlot(self,slot): for child in self.dlg.tabWidget.widget(1).children(): if child.objectName() == slot: child.clear() def checkCreateView(self): #method called when checkbox createview is clicked if self.dlg.checkCreateView.checkState(): self.dlg.QueryName.setEnabled(True) self.dlg.checkMaterialized.setEnabled(True) else: self.dlg.QueryName.setDisabled(True) self.dlg.checkMaterialized.setDisabled(True) if self.querySet.testQueryParametersCheckList(): self.queryGen() def helpDialog(self): print "called help" if self.dlg.QueryType.currentText() != "Select query type": QDesktopServices.openUrl(QUrl(self.querySet.getHelp())) #webbrowser.open_new(self.querySet.getHelp()) def helpLinkOpen(self): QDesktopServices.openUrl(QUrl("https://geogear.wordpress.com/postgisquerybuilder-v1-6-cheatsheet/")) def setMaterialized(self): self.queryGen() def keyGeomFieldsChanged(self): self.querySet.setParameter("GEOMETRYFIELD",self.dlg.GEOMETRYFIELD.currentText()) self.querySet.setParameter("KEYFIELD",self.dlg.KEYFIELD.currentText()) test = None try: test = self.querySet.testQueryParametersCheckList() except: pass if test: self.queryGen() def tabChangedHub(self,tab): #print "TAB:",tab if tab == 0: try: self.updateLayerMenu() except: pass try: self.keyGeomFieldsChanged() except: pass if tab == 3: self.queryGen() if tab == 2: self.updateOrderBy() elif tab == 5: self.updateHistoryLog() def updateOrderBy(self): if self.dlg.LAYERa.currentText()[:6] != "Select": try: self.populateComboBox(self.dlg.orderBy,self.PSQL.getFieldsContent(self.dlg.LAYERa.currentText())," ",True) except: pass def focusOnQuery(self): self.dlg.tabWidget.setCurrentIndex(3) def updateHistoryLog(self): historyFile = os.path.join(os.path.dirname(__file__),"validSql.log") if os.path.exists(historyFile): in_file = open(historyFile,"r") self.dlg.historyLog.setPlainText(in_file.read()) self.dlg.historyLog.moveCursor(QTextCursor.End) def populateComboBox(self,combo,list,predef,sort): #procedure to fill specified combobox with provided list combo.blockSignals (True) combo.clear() model=QStandardItemModel(combo) predefInList = None for elem in list: try: item = QStandardItem(unicode(elem)) except TypeError: item = QStandardItem(str(elem)) model.appendRow(item) if elem == predef: predefInList = elem if sort: model.sort(0) combo.setModel(model) if predef != "": if predefInList: combo.setCurrentIndex(combo.findText(predefInList)) else: combo.insertItem(0,predef) combo.setCurrentIndex(0) combo.blockSignals (False) def loadSVG(self,svgName): self.dlg.DiagPanel.show() svgFile = os.path.join( os.path.dirname( os.path.abspath( __file__ ) ), "svg",svgName + ".svg") #print svgFile item = QGraphicsSvgItem(svgFile) scene= QGraphicsScene() scene.addItem(item) self.dlg.DiagPanel.setScene(scene) self.dlg.DiagPanel.fitInView(item,Qt.KeepAspectRatio) #,Qt.KeepAspectRatio def setLAYERa(self): #called when LAYERa is activated if self.dlg.LAYERa.currentText()[:6] == "Select": return self.querySet.setParameter("LAYERa",self.dlg.LAYERa.currentText()) #try to guess layer geometry field autoGeometry = self.PSQL.guessGeometryField(self.dlg.LAYERa.currentText(),suggestion=self.dlg.GEOMETRYFIELD.currentText()) self.querySet.setParameter("GEOMETRYFIELDa",autoGeometry) if self.querySet.testQueryParametersCheckList(): self.queryGen() self.populateComboBox(self.dlg.FIELD,self.PSQL.getFieldsContent(self.dlg.LAYERa.currentText()),"Select field",True) if not self.PSQL.testIfFieldExist(self.dlg.LAYERa.currentText(),self.querySet.getParameter("KEYFIELD")): self.querySet.setFIDFIELD() self.addListToFieldTable(self.dlg.fieldsListA,self.PSQL.getFieldsContent(self.dlg.LAYERa.currentText()),True, exclude=autoGeometry) self.populateFilterTable() self.setSCHEMAb() def selectAllFields(self): for row in range(0,self.dlg.fieldsListA.count()): if self.dlg.fieldsListA.item(row).flags() & Qt.ItemIsEnabled: self.dlg.fieldsListA.item(row).setCheckState(self.dlg.LAYERaAllFields.checkState()) self.setFieldsList() def setSCHEMAb(self): #called when SCHEMAb is activated self.querySet.setParameter("SCHEMAb",self.dlg.SCHEMAb.currentText()) self.populateComboBox(self.dlg.LAYERb,self.PSQL.getLayers(schema = self.dlg.SCHEMAb.currentText()),"Select layer",True) def setFILTERSCHEMA(self): #called when FILTERSCHEMA is activated self.dlg.filterTable.setCurrentSchema(self.dlg.FILTERSCHEMA.currentText()) def setLAYERb(self): #called when LAYERb is activated if self.dlg.LAYERb.currentText()[:6] == "Select": return self.querySet.setParameter("LAYERb",self.dlg.LAYERb.currentText()) #try to guess layer geometry field autoGeometry = self.PSQL.guessGeometryField(self.dlg.LAYERb.currentText(),suggestion=self.dlg.GEOMETRYFIELD.currentText()) self.querySet.setParameter("GEOMETRYFIELDb",autoGeometry) if self.querySet.testQueryParametersCheckList(): self.queryGen() self.addListToFieldTable(self.dlg.fieldsListB,self.PSQL.getFieldsContent(self.dlg.LAYERb.currentText()),True) self.populateComboBox(self.dlg.FIELDb,self.PSQL.getFieldsContent(self.dlg.LAYERb.currentText()),"Select field",True) def addListToFieldTable(self,fieldSlot,fl,check,exclude = None): #called to populate field list for WHERE statement wdgt=fieldSlot wdgt.clear() #print "LAYERLIST:",check for row in fl: item=QListWidgetItem() if check: item.setFlags(item.flags() | Qt.ItemIsUserCheckable) item.setCheckState(Qt.Unchecked) item.setText(row) #print row if self.PSQL.isTable(row): item.setIcon(QIcon(":/plugins/postgisquerybuilder/iT.png")) elif self.PSQL.isView(row): item.setIcon(QIcon(":/plugins/postgisquerybuilder/iV.png")) elif self.PSQL.isMaterializedView(row): item.setIcon(QIcon(":/plugins/postgisquerybuilder/iM.png")) #disable geometryfield and auto set checked/unchecked wdgt.addItem(item) if check and exclude and row == exclude: #self.geoQuery and if self.geoQuery: item.setCheckState(Qt.Unchecked) else: item.setCheckState(Qt.Checked) item.setFlags(item.flags() & (~Qt.ItemIsEnabled)) #item.setFlags(item.flags() | Qt.ItemIsEnabled) def setFieldsList(self): # procedure to resume selected fields to SELECT statements items = [] for index in xrange(self.dlg.fieldsListA.count()): if self.dlg.fieldsListA.item(index).checkState() == Qt.Checked: items.append('"'+self.dlg.LAYERa.currentText()+'"."'+self.dlg.fieldsListA.item(index).text()+'"') for index in xrange(self.dlg.fieldsListB.count()): if self.dlg.fieldsListB.item(index).checkState() == Qt.Checked: items.append('"'+self.dlg.LAYERb.currentText()+'"."'+self.dlg.fieldsListB.item(index).text()+'"') #print "SELECTED FIELDS",items self.querySet.setFieldsSet(items) if self.querySet.testQueryParametersCheckList(): self.queryGen() def setFIELD(self): if self.dlg.FIELD.currentText()[:6] == "Select": return self.querySet.setParameter("FIELD",'"'+self.dlg.LAYERa.currentText()+'"."'+self.dlg.FIELD.currentText()+'"') self.querySet.setParameter("SIMPLEFIELD",self.dlg.FIELD.currentText()) fType = self.PSQL.getFieldsType(self.querySet.getParameter("LAYERa"),self.dlg.FIELD.currentText()) fType = fType[:4] #print fType if ((fType == "char") or (fType == "text") or (fType == "varc")): self.tDelimiter = "'" else: self.tDelimiter = "" if self.querySet.testQueryParametersCheckList(): self.queryGen() def setFIELDb(self): if self.dlg.FIELDb.currentText()[:6] == "Select": return self.querySet.setParameter("FIELDb",'"'+self.dlg.LAYERb.currentText()+'"."'+self.dlg.FIELDb.currentText()+'"') if self.querySet.testQueryParametersCheckList(): self.queryGen() def setSPATIALREL(self): if self.dlg.SPATIALREL.currentText()[:6] == "Select": return self.querySet.setParameter("SPATIALREL",self.dlg.SPATIALREL.currentText()) if self.querySet.testQueryParametersCheckList(): self.queryGen() def setBUFFERRADIUS(self): self.querySet.setParameter("BUFFERRADIUS",self.dlg.BUFFERRADIUS.text()) if self.querySet.testQueryParametersCheckList(): self.queryGen() def setSPATIALRELNOT(self): if self.dlg.SPATIALRELNOT.isChecked(): self.querySet.setParameter("SPATIALRELNOT","NOT ") else: self.querySet.setParameter("SPATIALRELNOT"," ") if self.querySet.testQueryParametersCheckList(): self.queryGen() def setJOIN(self): self.querySet.setParameter("JOIN",self.dlg.JOIN.currentText()) if self.querySet.testQueryParametersCheckList(): self.queryGen() def setQueryName(self): try: self.querySet.testIfQueryDefined() except: return self.querySet.setParameter("VIEWNAME",self.dlg.QueryName.text()) if self.querySet.testQueryParametersCheckList(): self.dlg.QueryResult.setPlainText(self.querySet.getQueryParsed(self.dlg.checkCreateView.checkState())) def queryGen(self): if self.dlg.checkCreateView.checkState(): self.enableDialogSlot("QueryName") if not self.querySet.testIfQueryDefined(): return if self.dlg.checkAutoCompiled.checkState(): if self.dlg.checkMaterialized.checkState(): self.querySet.setParameter("MATERIALIZED","MATERIALIZED") else: self.querySet.setParameter("MATERIALIZED","") self.enableDialogSlot("QueryResult") self.enableDialogSlot("checkCreateView") self.enableDialogSlot("AddToMap") try: qName = self.querySet.getNameParsed() except: qName = self.dlg.QueryName.text() self.dlg.QueryName.setText(qName) self.querySet.setParameter("VIEWNAME", qName) if self.dlg.filterTable.testIfSintaxOk(): self.querySet.setParameter("WHERE", self.dlg.filterTable.getWhereStatement()) self.querySet.setParameter("SPATIALFROM", self.dlg.filterTable.getSpatialFilterLayers(schema = self.PSQL.getSchema())) else: self.querySet.setParameter("WHERE", "") if self.dlg.orderBy.currentText() != " ": self.querySet.setParameter("ORDERBY", 'ORDER BY "'+self.dlg.orderBy.currentText()+'"') else: self.querySet.setParameter("ORDERBY", "") self.dlg.QueryResult.setPlainText(self.querySet.getQueryParsed(self.dlg.checkCreateView.checkState())) self.dlg.QueryName.textChanged.connect(self.setQueryName) def setQueryType(self,line): theQ = self.dlg.QueryType.currentText() if theQ[:6] == "Select": return self.loadSVG(theQ.replace(" ","_")) #self.querySet.resetParameters() self.resetDialog() self.dlg.summaryBox.show() self.querySet.setCurrentQuery(theQ) if theQ[:2]=="ST": self.geoQuery = True else: self.geoQuery = None for slot in self.querySet.getRequiredSlots(): #print slot #self.enableDialogSlot(slot) self.showDialogSlot(slot) self.showDialogSlot(slot+"Label") self.showDialogSlot(slot+"AllFields") #print self.querySet.testQueryParametersCheckList() #simulate click on checkbox to set required slot self.dlg.SPATIALRELNOT.setCheckState(Qt.Checked) self.dlg.SPATIALRELNOT.setCheckState(Qt.Unchecked) self.dlg.Helper.setText(theQ+":\n"+self.querySet.getDescription()) self.loadSVG(theQ.replace(" ","_")) if self.predefinedLayer: self.dlg.LAYERa.setCurrentIndex(self.dlg.LAYERa.findText(self.predefinedLayer)) self.setLAYERa() def resetDialog(self): self.eventsDisconnect() self.clearAllDialogs() self.querySet.resetParameters() #self.disableQueryDefSlot() self.hideQueryDefSlot() self.loadSVG("Select_query_type") self.dlg.summaryBox.hide() try: tables = self.PSQL.getLayers() except AttributeError: pass else: self.populateComboBox(self.dlg.LAYERa,tables,"Select Layer",True) self.populateComboBox(self.dlg.LAYERb,tables,"Select Layer",True) self.geoQuery = None self.populateComboBox(self.dlg.orderBy,[" "],"",True) self.eventsConnect() def hideQueryDefSlot(self): toHide=["BUFFERRADIUS","FIELD",\ "SPATIALREL","SPATIALRELNOT","LAYERaLabel","BUFFERRADIUSLabel",\ "FIELD","FIELDLabel","SPATIALRELLabel", "SPATIALRELNOTLabel","fieldsListALabel","fieldsListBLabel",\ "FIELDb","FIELDbLabel","JOIN","JOINLabel",\ "LAYERa","SCHEMAb","LAYERb","LAYERbLabel","summaryBox","LAYERaAllFields","fieldsListA","fieldsListB"] for slot in toHide: self.hideDialogSlot(slot) def clearQueryDefSlot(self): toClear=["LAYERa","LAYERb","BUFFERRADIUS","FIELD","FIELDb",\ "SPATIALREL","fieldsListA","fieldsListB",\ "JOIN"] for slot in toClear: self.clearDialogSlot(slot) def disableQueryDefSlot(self): toDisable=["LAYERa","LAYERb","BUFFERRADIUS","FIELD","FIELDb",\ "SPATIALREL",\ "SPATIALRELNOT","fieldsListA","fieldsListB", "JOIN"] for slot in toDisable: self.disableDialogSlot(slot) def clearAllDialogs(self): self.dlg.LAYERa.clear() self.dlg.LAYERb.clear() self.dlg.BUFFERRADIUS.clear() self.dlg.FIELD.clear() self.dlg.QueryResult.clear() self.dlg.QueryName.clear() self.dlg.fieldsListA.clear() self.dlg.fieldsListB.clear() self.dlg.TableResult.clear() self.dlg.SPATIALRELNOT.setCheckState(Qt.Unchecked) self.dlg.LAYERaAllFields.setCheckState(Qt.Unchecked) def getPSQLConnections(self): try: self.dlg.PSQLConnection.activated.disconnect(self.setConnection) except: pass conn = self.PSQL.getConnections() self.populateComboBox(self.dlg.PSQLConnection,conn,"Select connection",True) self.hideQueryDefSlot() self.dlg.PSQLConnection.activated.connect(self.setConnection) def closeDialog(self): self.resetDialog() self.dlg.hide() def resetForm(self): self.resetDialog() self.getPSQLConnections() self.populateGui() self.dlg.tabWidget.setCurrentIndex(0) def setConnection(self): if self.dlg.PSQLConnection.currentText()[:6] == "Select": return self.PSQL.setConnection(self.dlg.PSQLConnection.currentText()) #print "SCHEMAS",self.PSQL.getSchemas() schemas = self.PSQL.getSchemas() self.populateComboBox(self.dlg.DBSchema,schemas,"Select schema",True) self.dlg.DBSchema.activated.connect(self.loadPSQLLayers) for r in range (0,self.dlg.DBSchema.count()): if self.dlg.DBSchema.itemText(r) == "public": self.dlg.DBSchema.setCurrentIndex(r) self.dlg.DBSchema.removeItem(0) self.loadPSQLLayers() #clone dbschema in schemab self.populateComboBox(self.dlg.SCHEMAb,[self.dlg.DBSchema.itemText(i) for i in range(self.dlg.DBSchema.count())],self.dlg.DBSchema.currentText(),True) self.querySet.setParameter("SCHEMAb",self.dlg.SCHEMAb.currentText()) self.populateComboBox(self.dlg.FILTERSCHEMA,[self.dlg.DBSchema.itemText(i) for i in range(self.dlg.DBSchema.count())],self.dlg.DBSchema.currentText(),True) def loadPSQLLayers(self): if self.dlg.DBSchema.currentText() != "Select schema": self.PSQL.setSchema(self.dlg.DBSchema.currentText()) self.populateGui() self.resetDialog() #self.dlg.tabWidget.setCurrentIndex(1) self.querySet.setSchema(self.dlg.DBSchema.currentText()) self.populateComboBox(self.dlg.GEOMETRYFIELD,self.PSQL.scanLayersForGeometry(),"",None) self.populateComboBox(self.dlg.KEYFIELD,self.PSQL.scanLayersForPrimaryKey(),"",None) #sync schemab con dbschema self.dlg.SCHEMAb.setCurrentIndex(self.dlg.DBSchema.currentIndex()) self.querySet.setParameter("SCHEMAb",self.dlg.SCHEMAb.currentText()) self.populateLayerMenu() def populateFilterTable(self): self.dlg.filterTable.populateFilterTable(self.PSQL,self.dlg.LAYERa.currentText()) def testSignal(self,v1,v2): #print "catch:", v1,v2 pass def populateLayerMenu(self): self.addListToFieldTable(self.dlg.LayerList,self.PSQL.getLayers(),None) def updateLayerMenu(self): if self.dlg.DBSchema.currentText() != "Select schema": self.addListToFieldTable(self.dlg.LayerList,self.PSQL.getLayers(),None) def runQuery(self): #method to run generated query if self.dlg.filterTable.testIfSintaxOk(): if self.dlg.AddToMap.checkState(): if self.dlg.checkCreateView.checkState(): #self.querySet.getParameter("VIEWNAME") self.PSQL.submitQuery(self.dlg.QueryName.text(),self.dlg.QueryResult.toPlainText()) self.PSQL.loadView(self.dlg.QueryName.text(),self.querySet.getParameter("GEOMETRYFIELD"),self.querySet.getParameter("KEYFIELD")) else: self.PSQL.loadSql(self.dlg.QueryName.text(),self.dlg.QueryResult.toPlainText(),self.querySet.getParameter("GEOMETRYFIELD"),self.querySet.getParameter("KEYFIELD")) else: if self.dlg.checkCreateView.checkState(): self.PSQL.submitQuery(self.dlg.QueryName.text(),self.dlg.QueryResult.toPlainText()) sql = 'SELECT * FROM "%s"."%s"' % (self.PSQL.getSchema(),self.dlg.QueryName.text()) print sql rows = self.PSQL.tableResultGen(self.dlg.LAYERa.currentText(),sql,self.dlg.TableResult) self.PSQL.loadedLayerRefresh(self.dlg.QueryName.text()) else: rows = self.PSQL.tableResultGen(self.dlg.LAYERa.currentText(),self.dlg.QueryResult.toPlainText(),self.dlg.TableResult) self.dlg.labelRowsNumber.setText("Total rows: "+str(rows)) self.dlg.tabWidget.setCurrentIndex(4) else: QMessageBox.information(None, "FILTER ERROR:", "The Filter table is malformed") self.dlg.tabWidget.setCurrentIndex(2) def unload(self): # Remove the plugin menu item and icon self.iface.removePluginMenu(u"&postgisQueryBuilder", self.action) self.iface.removeToolBarIcon(self.action) self.iface.removeDockWidget(self.PQBdockwidget) # run method that performs all the real work def run(self): # show/hide the widget if self.PQBdockwidget.isVisible(): self.PQBdockwidget.hide() else: self.PQBdockwidget.show() self.getPSQLConnections()