Mercurial > dive4elements > river
view backend/contrib/shpimporter/floodmarks.py @ 9713:aef987124822 3.2.x
Do not rely on database default for setting hws_lines.official to 0
It's not under the control of the importer whether the field is rendered
into an INSERT statement, whether it's actually set or not. If it is
unset but rendered into the statement, the field would be set to NULL.
Thus, always set it explicitly.
While here, remove an unused import.
author | Tom Gottfried <tom@intevation.de> |
---|---|
date | Fri, 12 Mar 2021 15:11:44 +0100 |
parents | 90ba3ae2ced1 |
children |
line wrap: on
line source
# -*- coding: utf-8 -*- try: from osgeo import ogr except ImportError: import ogr from importer import Importer import logging import os import re TABLE_NAME="flood_marks" PATH="Hydrologie/HW-Marken" NAME="Floodmarks" logger = logging.getLogger(NAME) class Floodmark(Importer): fieldmap = { "^station$" : "km", "^km$" : "km", "^wsv-km$" : "km", "^FlussKm$" : "km", "^z$" : "z", "^z\d*" : "z", # z02, z1890, usw. "^m+NHN$" : "z", "^Ort$" : "location", "^Pegel$" : "location", } def getPath(self, base): return "%s/%s" % (base, PATH) def getTablename(self): return TABLE_NAME def getName(self): return NAME def isGeometryValid(self, geomType): return geomType == ogr.wkbPoint def isShapeRelevant(self, name, path): return "hw-marken" in name.lower() def createNewFeature(self, featureDef, feat, **args): newFeat = ogr.Feature(featureDef) geometry = feat.GetGeometryRef() geometry.SetCoordinateDimension(2) newFeat.SetGeometry(geometry) self.copyFields(feat, newFeat, self.fieldmap) newFeat.SetField("river_id", self.river_id) filename = os.path.basename(args['path']) # Try to extract the year from the filename match = re.search(r"([_\-])(\d\d\d\d)([_\-])", filename) if match: year = match.groups()[1] year = int(year) newFeat.SetField("year", year) else: logger.warn(u"Could not extract year from filename: %s " % filename) return newFeat