Mercurial > dive4elements > river
view flys-backend/contrib/shpimporter/importer.py @ 4284:7a94d5e7fc3d
Write the "hidden" attribute of a style into the collection's attribute when using the theme editor.
Otherwise, the next time using the theme editor, the hidden attributes that should not be changed
by the user are visible.
author | Ingo Weinzierl <ingo.weinzierl@intevation.de> |
---|---|
date | Mon, 29 Oct 2012 07:18:42 +0100 |
parents | 82e931f88137 |
children | 890eaa0a5162 |
line wrap: on
line source
import ogr, osr import shpimporter class Importer: def __init__(self, config): self.config = config self.dbconn = 'OCI:%s/%s@%s' % (config.user, config.password, config.host) self.river_id = config.river_id self.dest_srs = osr.SpatialReference() self.dest_srs.ImportFromEPSG(config.target_srs) def getKind(self, path): raise NotImplementedError("Importer.getKind is abstract!") def getPath(self, base): raise NotImplementedError("Importer.getPath is abstract!") def getTablename(self): raise NotImplementedError("Importer.getTablename is abstract!") def getName(self): raise NotImplementedError("Importer.getTablename is abstract!") def IsFieldSet(self, feat, name): try: isset = feat.GetField(name) return isset is not None except: return False def IsDoubleFieldSet(self, feat, name): try: isset = feat.GetFieldAsDouble(name) return isset is not None except: return False def isShapeRelevant(self, name, path): return True def walkOverShapes(self, shape): (name, path) = shape if not self.isShapeRelevant(name, path): shpimporter.INFO("Skip shapefile '%s'" % path) return shp = ogr.Open(shape[1]) if shp is None: shpimporter.ERROR("Shapefile '%s' could not be opened!" % path) return shpimporter.INFO("Processing shapefile '%s'" % path) srcLayer = shp.GetLayerByName(name) if srcLayer is None: shpimporter.ERROR("Layer '%s' was not found!" % name) return return self.shape2Database(srcLayer, name, path) def transform(self, feat): geometry = feat.GetGeometryRef() src_srs = geometry.GetSpatialReference() if src_srs is None: shpimporter.ERROR("No source SRS given! No transformation possible!") return feat transformer = osr.CoordinateTransformation(src_srs, self.dest_srs) geometry.Transform(transformer) return feat def shape2Database(self, srcLayer, name, path): table = ogr.Open(self.dbconn) destLayer = table.GetLayerByName(self.getTablename()) if srcLayer is None: shpimporter.ERROR("Shapefile is None!") return -1 if destLayer is None: shpimporter.ERROR("No destination layer given!") return -1 count = srcLayer.GetFeatureCount() shpimporter.DEBUG("Try to add %i features to database." % count) srcLayer.ResetReading() geomType = -1 success = 0 unsupported = 0 creationFailed = 0 featureDef = destLayer.GetLayerDefn() for feat in srcLayer: geom = feat.GetGeometryRef() if geom is None: continue geomType = geom.GetGeometryType() if self.isGeometryValid(geomType): newFeat = self.createNewFeature(featureDef, feat, name=name, path=path) if newFeat is not None: newFeat.SetField("path", path) newFeat = self.transform(newFeat) res = destLayer.CreateFeature(newFeat) if res is None or res > 0: shpimporter.ERROR("Unable to insert feature: %r" % res) else: success = success + 1 else: creationFailed = creationFailed + 1 else: unsupported = unsupported + 1 shpimporter.INFO("Inserted %i features" % success) shpimporter.INFO("Failed to create %i features" % creationFailed) shpimporter.INFO("Found %i unsupported features" % unsupported) try: if self.config.dry_run > 0: return geomType destLayer.CommitTransaction() except e: shpimporter.ERROR("Exception while committing transaction.") return geomType