sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-10-08 00:05:09 -04:00
Code Issues Packages Projects Releases Wiki Activity

[spatialite] Ensure that encodeUri and decodeUri are lossless

Browse Source
This commit is contained in:
nirvn 2020-11-14 14:47:12 +07:00 committed by Mathieu Pellerin
parent 1dfea2a036
commit c54c2ee353
2 changed files with 20 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/providers/spatialite/qgsspatialiteprovider.cpp
View File

@ -5860,6 +5860,12 @@ QVariantMap QgsSpatiaLiteProviderMetadata::decodeUri( const QString &uri ) const
QVariantMap components;
components.insert( QStringLiteral( "path" ), dsUri.database() );
components.insert( QStringLiteral( "layerName" ), dsUri.table() );
if ( !dsUri.sql().isEmpty() )
components.insert( QStringLiteral( "subset" ), dsUri.sql() );
if ( !dsUri.geometryColumn().isEmpty() )
components.insert( QStringLiteral( "geometryColumn" ), dsUri.geometryColumn() );
if ( !dsUri.keyColumn().isEmpty() )
components.insert( QStringLiteral( "keyColumn" ), dsUri.keyColumn() );
return components;
}
@ -5876,6 +5882,9 @@ QString QgsSpatiaLiteProviderMetadata::encodeUri( const QVariantMap &parts ) con
QgsDataSourceUri dsUri;
dsUri.setDatabase( parts.value( QStringLiteral( "path" ) ).toString() );
dsUri.setTable( parts.value( QStringLiteral( "layerName" ) ).toString() );
dsUri.setSql( parts.value( QStringLiteral( "subset" ) ).toString() );
dsUri.setGeometryColumn( parts.value( QStringLiteral( "geometryColumn" ) ).toString() );
dsUri.setKeyColumn( parts.value( QStringLiteral( "keyColumn" ) ).toString() );
return dsUri.uri();
}

14
tests/src/python/test_provider_spatialite.py
View File

@ -965,10 +965,14 @@ class TestQgsSpatialiteProvider(unittest.TestCase, ProviderTestCase):
"""Check that the provider URI decoding returns expected values"""
filename = '/home/to/path/test.db'
uri = 'dbname=\'{}\' table="test" (geometry) sql='.format(filename)
uri = 'dbname=\'{}\' table="test" (geometry) key=testkey sql=1=1'.format(filename)
registry = QgsProviderRegistry.instance()
components = registry.decodeUri('spatialite', uri)
self.assertEqual(components['path'], filename)
self.assertEqual(components['layerName'], 'test')
self.assertEqual(components['subset'], '1=1')
self.assertEqual(components['geometryColumn'], 'geometry')
self.assertEqual(components['keyColumn'], 'testkey')
def testEncodeUri(self):
"""Check that the provider URI encoding returns expected values"""
@ -976,9 +980,13 @@ class TestQgsSpatialiteProvider(unittest.TestCase, ProviderTestCase):
filename = '/home/to/path/test.db'
registry = QgsProviderRegistry.instance()
parts = {'path': filename, 'layerName': 'test'}
parts = {'path': filename,
'layerName': 'test',
'subset': '1=1',
'geometryColumn': 'geometry',
'keyColumn': 'testkey'}
uri = registry.encodeUri('spatialite', parts)
self.assertEqual(uri, 'dbname=\'{}\' table="test"'.format(filename))
self.assertEqual(uri, 'dbname=\'{}\' key=\'testkey\' table="test" (geometry) sql=1=1'.format(filename))
def testPKNotInt(self):
""" Check when primary key is not an integer """