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

[dbmanager] allow to import GEOMETRY tables

Browse Source 
UI is similar to the browser, where generic geometry tables
are display several times, one for each type.
This commit is contained in:
olivierdalang 2019-06-05 17:43:17 +02:00
parent b9d5cde53e
commit 86f8bdb0aa
1 changed files with 31 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
python/plugins/db_manager/db_plugins/plugin.py
View File

@ -31,7 +31,8 @@ from qgis.core import (
Qgis,
QgsApplication,
QgsSettings,
QgsMapLayerType
QgsMapLayerType,
QgsWkbTypes
)
from ..db_plugins import createDbPlugin
@ -522,8 +523,28 @@ class Database(DbItemObject):
def tables(self, schema=None, sys_tables=False):
tables = self.connector.getTables(schema.name if schema else None, sys_tables)
if tables is not None:
tables = [self.tablesFactory(x, self, schema) for x in tables]
return tables
ret = []
for t in tables:
table = self.tablesFactory(t, self, schema)
ret.append(table)
# Similarly to what to browser does, if the geom type is generic geometry,
# we additionnly add three copies of the layer to allow importing
if isinstance(table, VectorTable):
if table.geomType == 'GEOMETRY':
point_table = self.tablesFactory(t, self, schema)
point_table.geomType = 'POINT'
ret.append(point_table)
line_table = self.tablesFactory(t, self, schema)
line_table.geomType = 'LINESTRING'
ret.append(line_table)
poly_table = self.tablesFactory(t, self, schema)
poly_table.geomType = 'POLYGON'
ret.append(poly_table)
return ret
def createTable(self, table, fields, schema=None):
field_defs = [x.definition() for x in fields]
@ -692,6 +713,13 @@ class Table(DbItemObject):
geomCol = self.geomColumn if self.type in [Table.VectorType, Table.RasterType] else ""
uniqueCol = self.getValidQgisUniqueFields(True) if self.isView else None
uri.setDataSource(schema, self.name, geomCol if geomCol else None, None, uniqueCol.name if uniqueCol else "")
uri.setSrid(str(self.srid))
for f in self.fields():
if f.primaryKey:
uri.setKeyColumn(f.name)
break
uri.setWkbType(QgsWkbTypes.parseType(self.geomType))
return uri
def mimeUri(self):