mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
222 lines
7.1 KiB
Python
222 lines
7.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
/***************************************************************************
|
|
Name : DB Manager
|
|
Description : Database manager plugin for QuantumGIS
|
|
Date : Oct 13, 2011
|
|
copyright : (C) 2011 by Giuseppe Sucameli
|
|
email : brush.tyler@gmail.com
|
|
|
|
The content of this file is based on
|
|
- PG_Manager by Martin Dobias (GPLv2 license)
|
|
***************************************************************************/
|
|
|
|
/***************************************************************************
|
|
* *
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation; either version 2 of the License, or *
|
|
* (at your option) any later version. *
|
|
* *
|
|
***************************************************************************/
|
|
"""
|
|
|
|
from PyQt4.QtCore import *
|
|
from PyQt4.QtGui import *
|
|
|
|
import qgis.core
|
|
|
|
from .db_plugins.plugin import DbError
|
|
from .dlg_db_error import DlgDbError
|
|
|
|
from .ui.ui_DlgImportVector import Ui_DlgImportVector
|
|
|
|
class DlgImportVector(QDialog, Ui_DlgImportVector):
|
|
|
|
def __init__(self, inLayer, outDb, outUri, parent=None):
|
|
QDialog.__init__(self, parent)
|
|
self.inLayer = inLayer
|
|
self.db = outDb
|
|
self.outUri = outUri
|
|
self.setupUi(self)
|
|
|
|
self.default_pk = "id"
|
|
self.default_geom = "geom"
|
|
|
|
# updates of UI
|
|
for widget in [self.radCreate, self.chkDropTable, self.radAppend,
|
|
self.chkPrimaryKey, self.chkGeomColumn, self.chkSpatialIndex,
|
|
self.chkSourceSrid, self.chkTargetSrid, self.chkEncoding]:
|
|
self.connect(widget, SIGNAL("clicked()"), self.updateUi)
|
|
|
|
self.connect(self.cboSchema, SIGNAL("currentIndexChanged(int)"), self.populateTables)
|
|
self.connect(self.buttonBox, SIGNAL("accepted()"), self.importLayer)
|
|
|
|
self.populateSchemas()
|
|
self.populateTables()
|
|
self.populateEncodings()
|
|
self.updateUi()
|
|
|
|
self.cboTable.setEditText(self.outUri.table())
|
|
pk = self.outUri.keyColumn()
|
|
self.editPrimaryKey.setText(pk if pk != "" else self.default_pk)
|
|
geom = self.outUri.geometryColumn()
|
|
self.editGeomColumn.setText(geom if geom != "" else self.default_geom)
|
|
inCrs = self.inLayer.crs()
|
|
srid = inCrs.postgisSrid() if inCrs.isValid() else 4236
|
|
self.editSourceSrid.setText( "%s" % srid )
|
|
self.editTargetSrid.setText( "%s" % srid )
|
|
|
|
self.checkSupports()
|
|
|
|
|
|
def checkSupports(self):
|
|
allowSpatial = self.db.connector.hasSpatialSupport()
|
|
self.chkGeomColumn.setEnabled(allowSpatial)
|
|
self.chkSourceSrid.setEnabled(allowSpatial)
|
|
self.chkTargetSrid.setEnabled(allowSpatial)
|
|
self.chkSpatialIndex.setEnabled(allowSpatial)
|
|
|
|
|
|
def populateSchemas(self):
|
|
if not self.db:
|
|
return
|
|
|
|
self.cboSchema.clear()
|
|
schemas = self.db.schemas()
|
|
if schemas == None:
|
|
self.hideSchemas()
|
|
return
|
|
|
|
index = -1
|
|
for schema in schemas:
|
|
self.cboSchema.addItem(schema.name)
|
|
if schema.name == self.outUri.schema():
|
|
index = self.cboSchema.count()-1
|
|
self.cboSchema.setCurrentIndex(index)
|
|
|
|
def hideSchemas(self):
|
|
self.cboSchema.setEnabled(False)
|
|
|
|
def populateTables(self):
|
|
if not self.db:
|
|
return
|
|
|
|
currentText = self.cboTable.currentText()
|
|
|
|
schemas = self.db.schemas()
|
|
if schemas != None:
|
|
schema_name = self.cboSchema.currentText()
|
|
matching_schemas = filter(lambda x: x.name == schema_name, schemas)
|
|
tables = matching_schemas[0].tables() if len(matching_schemas) > 0 else []
|
|
else:
|
|
tables = self.db.tables()
|
|
|
|
self.cboTable.clear()
|
|
for table in tables:
|
|
self.cboTable.addItem(table.name)
|
|
|
|
self.cboTable.setEditText(currentText)
|
|
|
|
def populateEncodings(self):
|
|
encodings = ['ISO-8859-1', 'ISO-8859-2', 'UTF-8', 'CP1250']
|
|
for enc in encodings:
|
|
self.cboEncoding.addItem(enc)
|
|
self.cboEncoding.setCurrentIndex(2)
|
|
|
|
def updateUi(self):
|
|
allowDropTable = self.radCreate.isChecked()
|
|
self.chkDropTable.setEnabled(allowDropTable)
|
|
|
|
allowSetPrimaryKey = self.chkPrimaryKey.isChecked()
|
|
self.editPrimaryKey.setEnabled(allowSetPrimaryKey)
|
|
|
|
allowSetGeomColumn = self.chkGeomColumn.isChecked()
|
|
self.editGeomColumn.setEnabled(allowSetGeomColumn)
|
|
|
|
allowSetSourceSrid = self.chkSourceSrid.isChecked()
|
|
self.editSourceSrid.setEnabled(allowSetSourceSrid)
|
|
|
|
allowSetTargetSrid = self.chkTargetSrid.isChecked()
|
|
self.editTargetSrid.setEnabled(allowSetTargetSrid)
|
|
|
|
allowSetEncoding = self.chkEncoding.isChecked()
|
|
self.cboEncoding.setEnabled(allowSetEncoding)
|
|
|
|
|
|
def importLayer(self):
|
|
# sanity checks
|
|
if self.cboTable.currentText().isEmpty():
|
|
QMessageBox.information(self, "Import to database", "Table name is required")
|
|
return
|
|
|
|
if self.chkSourceSrid.isChecked():
|
|
sourceSrid, ok = self.editSourceSrid.text().toInt()
|
|
if not ok:
|
|
QMessageBox.information(self, "Import to database", "Invalid source srid: must be an integer")
|
|
return
|
|
|
|
if self.chkTargetSrid.isChecked():
|
|
targetSrid, ok = self.editTargetSrid.text().toInt()
|
|
if not ok:
|
|
QMessageBox.information(self, "Import to database", "Invalid target srid: must be an integer")
|
|
return
|
|
|
|
QApplication.setOverrideCursor(QCursor(Qt.WaitCursor))
|
|
|
|
schema = self.outUri.schema() if not self.cboSchema.isEnabled() else self.cboSchema.currentText()
|
|
table = self.cboTable.currentText()
|
|
|
|
pk = self.outUri.keyColumn() if not self.chkPrimaryKey.isChecked() else self.editPrimaryKey.text()
|
|
pk = pk if pk != "" else self.default_pk
|
|
|
|
geom = self.outUri.geometryColumn() if not self.chkGeomColumn.isChecked() else self.editGeomColumn.text()
|
|
if self.inLayer.hasGeometryType():
|
|
geom = geom if geom != "" else self.default_geom
|
|
|
|
self.outUri.setDataSource( schema, table, geom, QString(), pk )
|
|
uri = self.outUri.uri()
|
|
|
|
providerName = self.db.dbplugin().providerName()
|
|
|
|
if self.chkSourceSrid.isChecked():
|
|
sourceSrid = self.editSourceSrid.text().toInt()[0]
|
|
inCrs = qgis.core.QgsCoordinateReferenceSystem(sourceSrid)
|
|
self.inLayer.setCrs( inCrs )
|
|
|
|
outCrs = None
|
|
if self.chkTargetSrid.isChecked():
|
|
targetSrid = self.editTargetSrid.text().toInt()[0]
|
|
outCrs = qgis.core.QgsCoordinateReferenceSystem(targetSrid)
|
|
|
|
if self.chkEncoding.isChecked():
|
|
enc = self.cboEncoding.currentText()
|
|
self.inLayer.setProviderEncoding( enc )
|
|
|
|
options = {}
|
|
if self.radCreate.isChecked() and self.chkDropTable.isChecked():
|
|
options['overwrite'] = True
|
|
elif self.radAppend.isChecked():
|
|
options['append'] = True
|
|
|
|
ret, errMsg = qgis.core.QgsVectorLayerImport.importLayer( self.inLayer, uri, providerName, outCrs, False, False, options )
|
|
QApplication.restoreOverrideCursor()
|
|
if ret != 0:
|
|
QMessageBox.warning(self, "Import to database", u"Error %d\n%s" % (ret, errMsg) )
|
|
return
|
|
|
|
if self.chkSpatialIndex.isChecked():
|
|
self.db.connector.createSpatialIndex( (schema, table), geom )
|
|
|
|
QMessageBox.information(self, "Import to database", "Import was successful.")
|
|
return self.accept()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
a = QApplication(sys.argv)
|
|
dlg = DlgLoadData()
|
|
dlg.show()
|
|
sys.exit(a.exec_())
|