postgres provider:

jef-n · jef-n · commit d2b5ed920404 · 2012-02-08T21:02:40.000+01:00
- pre-select first integer column as primary key for views
- also consider oid type as integer column
diff --git a/src/providers/postgres/qgspgtablemodel.cpp b/src/providers/postgres/qgspgtablemodel.cpp
@@ -91,16 +91,20 @@ void QgsPgTableModel::addTableEntry( QgsPostgresLayerProperty layerProperty )
   sridItem->setEditable( false );
 
   QString pkText, pkCol = "";
-  switch ( layerProperty.pkCols.size() )
+  if ( layerProperty.pkCols.size() == 0 )
   {
-    case 0:   pkText = ""; break;
-    case 1:   pkText = layerProperty.pkCols[0]; pkCol = pkText; break;
-    default:  pkText = tr( "Select..." ); break;
+    pkText = "";
+  }
+  else
+  {
+    pkCol = pkText = layerProperty.pkCols[0];
   }
 
   QStandardItem *pkItem = new QStandardItem( pkText );
-  if ( pkText == tr( "Select..." ) )
+  if ( layerProperty.pkCols.size() > 1 )
     pkItem->setFlags( pkItem->flags() | Qt::ItemIsEditable );
+  else
+    pkItem->setFlags( pkItem->flags() & ~Qt::ItemIsEditable );
 
   pkItem->setData( layerProperty.pkCols, Qt::UserRole + 1 );
   pkItem->setData( pkCol, Qt::UserRole + 2 );
@@ -313,11 +317,11 @@ bool QgsPgTableModel::setData( const QModelIndex &idx, const QVariant &value, in
 
     bool ok = geomType != QGis::UnknownGeometry;
 
-    if ( geomType != QGis::NoGeometry )
+    if ( ok && geomType != QGis::NoGeometry )
       idx.sibling( idx.row(), dbtmSrid ).data().toInt( &ok );
 
     QStringList pkCols = idx.sibling( idx.row(), dbtmPkCol ).data( Qt::UserRole + 1 ).toStringList();
-    if ( pkCols.size() > 0 )
+    if ( ok && pkCols.size() > 0 )
       ok = pkCols.contains( idx.sibling( idx.row(), dbtmPkCol ).data().toString() );
 
     for ( int i = 0; i < dbtmColumns; i++ )
diff --git a/src/providers/postgres/qgspostgresprovider.cpp b/src/providers/postgres/qgspostgresprovider.cpp
@@ -1056,16 +1056,15 @@ bool QgsPostgresProvider::loadFields()
         fieldSize = -1;
         fieldPrec = 0;
       }
-      else if ( fieldTypeName.startsWith( "int" ) ||
-                fieldTypeName == "serial" )
+      else if ( fieldTypeName == "int2" || fieldTypeName == "int4" ||
+                fieldTypeName == "oid" || fieldTypeName == "serial" )
       {
         fieldType = QVariant::Int;
         fieldSize = -1;
         fieldPrec = 0;
       }
-      else if ( fieldTypeName == "real" ||
-                fieldTypeName == "double precision" ||
-                fieldTypeName.startsWith( "float" ) )
+      else if ( fieldTypeName == "real" || fieldTypeName == "double precision" ||
+                fieldTypeName == "float4" || fieldTypeName == "float8" )
       {
         fieldType = QVariant::Double;
         fieldSize = -1;

Original file line number	Diff line number	Diff line change
`@@ -1056,16 +1056,15 @@ bool QgsPostgresProvider::loadFields()`
`1056`	`1056`	`fieldSize = -1;`
`1057`	`1057`	`fieldPrec = 0;`
`1058`	`1058`	`}`
`1059`		`- else if ( fieldTypeName.startsWith( "int" ) \|\|`
`1060`		`- fieldTypeName == "serial" )`
	`1059`	`+ else if ( fieldTypeName == "int2" \|\| fieldTypeName == "int4" \|\|`
	`1060`	`+ fieldTypeName == "oid" \|\| fieldTypeName == "serial" )`
`1061`	`1061`	`{`
`1062`	`1062`	`fieldType = QVariant::Int;`
`1063`	`1063`	`fieldSize = -1;`
`1064`	`1064`	`fieldPrec = 0;`
`1065`	`1065`	`}`
`1066`		`- else if ( fieldTypeName == "real" \|\|`
`1067`		`- fieldTypeName == "double precision" \|\|`
`1068`		`- fieldTypeName.startsWith( "float" ) )`
	`1066`	`+ else if ( fieldTypeName == "real" \|\| fieldTypeName == "double precision" \|\|`
	`1067`	`+ fieldTypeName == "float4" \|\| fieldTypeName == "float8" )`
`1069`	`1068`	`{`
`1070`	`1069`	`fieldType = QVariant::Double;`
`1071`	`1070`	`fieldSize = -1;`