Merge pull request #1262 from nyalldawson/processing_postgis

Improvements to processing "Import into PostGIS" algorithm

Author: jef-n

python/plugins/processing/admintools/ImportIntoPostGIS.py

@@ -35,6 +35,7 @@
 from processing.parameters.ParameterBoolean import ParameterBoolean
 from processing.parameters.ParameterVector import ParameterVector
 from processing.parameters.ParameterString import ParameterString
+from processing.parameters.ParameterSelection import ParameterSelection
 from processing.tools import dataobjects
 
 from processing.admintools import postgis_utils
@@ -48,15 +49,18 @@ class ImportIntoPostGIS(GeoAlgorithm):
     INPUT = 'INPUT'
     OVERWRITE = 'OVERWRITE'
     CREATEINDEX = 'CREATEINDEX'
+    GEOMETRY_COLUMN = 'GEOMETRY_COLUMN'
+    LOWERCASE_NAMES = 'LOWERCASE_NAMES'
 
     def getIcon(self):
         return QIcon(os.path.dirname(__file__) + '/../images/postgis.png')
 
     def processAlgorithm(self, progress):
-        connection = self.getParameterValue(self.DATABASE)
+        connection = self.DB_CONNECTIONS[self.getParameterValue(self.DATABASE)]
         schema = self.getParameterValue(self.SCHEMA)
         overwrite = self.getParameterValue(self.OVERWRITE)
         createIndex = self.getParameterValue(self.CREATEINDEX)
+        convertLowerCase = self.getParameterValue(self.LOWERCASE_NAMES)
         settings = QSettings()
         mySettings = '/PostgreSQL/connections/' + connection
         try:
@@ -80,14 +84,19 @@ def processAlgorithm(self, progress):
             raise GeoAlgorithmExecutionException(
                     "Couldn't connect to database:\n" + e.message)
 
+        geomColumn = self.getParameterValue(self.GEOMETRY_COLUMN)
+        if not geomColumn:
+            geomColumn = 'the_geom'
+
         uri = QgsDataSourceURI()
         uri.setConnection(host, str(port), database, username, password)
-        uri.setDataSource(schema, table, 'the_geom', '')
+        uri.setDataSource(schema, table, geomColumn, '')
 
         options = {}
         if overwrite:
             options['overwrite'] = True
-
+        if convertLowerCase:
+            options['lowercaseFieldNames'] = True
         layerUri = self.getParameterValue(self.INPUT)
         layer = dataobjects.getObjectFromUri(layerUri)
         (ret, errMsg) = QgsVectorLayerImport.importLayer(
@@ -104,19 +113,31 @@ def processAlgorithm(self, progress):
                     'Error importing to PostGIS\n%s' % errMsg)
 
         if createIndex:
-            db.create_spatial_index(table, schema, 'the_geom')
+            db.create_spatial_index(table, schema, geomColumn)
 
         db.vacuum_analyze(table, schema)
 
+    def dbConnectionNames(self):
+        settings = QSettings()
+        settings.beginGroup('/PostgreSQL/connections/')
+        return settings.childGroups()
+
     def defineCharacteristics(self):
         self.name = 'Import into PostGIS'
         self.group = 'PostGIS management tools'
         self.addParameter(ParameterVector(self.INPUT, 'Layer to import'))
-        self.addParameter(ParameterString(self.DATABASE,
-                          'Database (connection name)'))
+
+        self.DB_CONNECTIONS = self.dbConnectionNames()
+        self.addParameter(ParameterSelection(self.DATABASE, 'Database (connection name)',
+                          self.DB_CONNECTIONS))
+
         self.addParameter(ParameterString(self.SCHEMA, 'Schema (schema name)'))
         self.addParameter(ParameterString(self.TABLENAME, 'Table to import to'
                           ))
+        self.addParameter(ParameterString(self.GEOMETRY_COLUMN, 'Geometry column', 'the_geom'
+                          ))
         self.addParameter(ParameterBoolean(self.OVERWRITE, 'Overwrite', True))
         self.addParameter(ParameterBoolean(self.CREATEINDEX,
                           'Create spatial index', True))
+        self.addParameter(ParameterBoolean(self.LOWERCASE_NAMES,
+                          'Convert field names to lowercase', False))

src/providers/postgres/qgspostgresprovider.cpp

@@ -2844,8 +2844,6 @@ QgsVectorLayerImport::ImportError QgsPostgresProvider::createEmptyLayer(
   QString *errorMessage,
   const QMap<QString, QVariant> *options )
 {
-  Q_UNUSED( options );
-
   // populate members from the uri structure
   QgsDataSourceURI dsUri( uri );
   QString schemaName = dsUri.schema();
@@ -3048,6 +3046,12 @@ QgsVectorLayerImport::ImportError QgsPostgresProvider::createEmptyLayer(
         continue;
       }
 
+      if ( options->contains( "lowercaseFieldNames" ) && options->value( "lowercaseFieldNames" ).toBool() )
+      {
+        //convert field name to lowercase
+        fld.setName( fld.name().toLower() );
+      }
+
       if ( !convertField( fld ) )
       {
         if ( errorMessage )