view flys-backend/contrib/shpimporter/axis.py @ 5004:f1c01fecf194 dami

Fix imports of ogr, this removes a warning in gdal 1.9 but keeps compatibility with older versions
author Andre Heinecke <aheinecke@intevation.de>
date Fri, 15 Feb 2013 16:18:34 +0100
parents 23a090b41543
children c5187ab9f571
line wrap: on
line source
try:
    from osgeo import ogr
except ImportErrror:
    import ogr

from importer import Importer

NAME="Axis"
TABLE_NAME="river_axes"
PATH="Geodaesie/Flussachse+km"


class Axis(Importer):

    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 == 2


    def isShapeRelevant(self, name, path):
        return "achse" in name.lower()


    def createNewFeature(self, featureDef, feat, **args):
        newFeat = ogr.Feature(featureDef)
        newFeat.SetGeometry(feat.GetGeometryRef())
        newFeat.SetField("name", args['name'])

        if self.IsFieldSet(feat, "river_id"):
            riverId = feat.GetField("river_id")
        else:
            riverId = self.river_id

        if self.IsFieldSet(feat, "kind"):
            kind = feat.GetField("kind")
        else:
            kind = 0

        newFeat.SetField("river_id", riverId)
        newFeat.SetField("kind", kind)

        return newFeat

http://dive4elements.wald.intevation.org