dbmanager: minor fixes

brushtyler · brushtyler · commit c3c79b939e1f · 2012-12-05T20:06:04.000+01:00
diff --git a/python/plugins/db_manager/db_manager.py b/python/plugins/db_manager/db_manager.py
@@ -150,8 +150,13 @@ def importActionSlot(self):
 			QMessageBox.information(self, "Sorry", "No database selected or you are not connected to it.")
 			return
 
+		outUri = db.uri()
+		schema = self.tree.currentSchema()
+		if schema:
+			outUri.setDataSource( schema.name, "", "", "" )
+
 		from .dlg_import_vector import DlgImportVector
-		dlg = DlgImportVector(None, db, db.uri(), self)
+		dlg = DlgImportVector(None, db, outUri, self)
 		dlg.exec_()
 
 	def exportActionSlot(self):
diff --git a/python/plugins/db_manager/dlg_import_vector.py b/python/plugins/db_manager/dlg_import_vector.py
@@ -44,26 +44,27 @@ def __init__(self, inLayer, outDb, outUri, parent=None):
 		self.default_pk = "id"
 		self.default_geom = "geom"
 
-		# updates of UI
-		self.setupWorkingMode()
+		self.mode = self.ASK_FOR_INPUT_MODE if self.inLayer is None else self.HAS_INPUT_MODE
 
-		self.cboTable.setEditText(self.outUri.table())
-		
-		self.populateLayers()
+		# updates of UI
+		self.setupWorkingMode( self.mode )
 
 		self.connect(self.cboSchema, SIGNAL("currentIndexChanged(int)"), self.populateTables)
 		self.populateSchemas()
 		self.populateTables()
-
+		self.populateLayers()
 		self.populateEncodings()
 
 
-	def setupWorkingMode(self):
-		""" display the widget to select a layer/file if there's no input layer """
-		self.mode = self.ASK_FOR_INPUT_MODE if self.inLayer is None else self.HAS_INPUT_MODE
-		self.wdgInput.setVisible( self.mode == self.ASK_FOR_INPUT_MODE )
+	def setupWorkingMode(self, mode):
+		""" hide the widget to select a layer/file if the input layer
+			is already set """
+		self.wdgInput.setVisible( mode == self.ASK_FOR_INPUT_MODE )
+		self.resize( 200, 200 )
+
+		self.cboTable.setEditText(self.outUri.table())
 
-		if not self.inLayer:
+		if mode == self.ASK_FOR_INPUT_MODE:
 			QObject.connect( self.btnChooseInputFile, SIGNAL("clicked()"), self.chooseInputFile )
 			#QObject.connect( self.cboInputLayer.lineEdit(), SIGNAL("editingFinished()"), self.updateInputLayer )
 			QObject.connect( self.cboInputLayer, SIGNAL("editTextChanged(const QString &)"), self.inputPathChanged )
@@ -114,7 +115,7 @@ def deleteInputLayer(self):
 		return False
 
 	def chooseInputFile(self):
-		vectorFormats = qgis.core.QgsVectorFileWriter.fileFilterString()
+		vectorFormats = qgis.core.QgsProviderRegistry.instance().fileVectorFilters()
 		# get last used dir and format
 		settings = QSettings()
                 lastDir = settings.value("/db_manager/lastUsedDir", "").toString()
@@ -216,12 +217,19 @@ def populateEncodings(self):
 		self.cboEncoding.setCurrentIndex(2)
 		
 	def accept(self):
+		if self.mode == self.ASK_FOR_INPUT_MODE:
+			# create the input layer (if not already done) and 
+			# update available options w/o changing the tablename!
+			self.cboTable.blockSignals(True)
+			table = self.cboTable.currentText()
+			self.updateInputLayer()
+			self.cboTable.setEditText(table)
+			self.cboTable.blockSignals(False)
+
 		# sanity checks
 		if self.inLayer is None:
-			# create the input layer and update available options
-			if not self.updateInputLayer():
-				QMessageBox.information(self, "Import to database", "Input layer missing or not valid")
-				return
+			QMessageBox.information(self, "Import to database", "Input layer missing or not valid")
+			return
 
 		if self.cboTable.currentText() == "":
 			QMessageBox.information(self, "Import to database", "Output table name is required")
