Mercurial > dive4elements > river
annotate backend/contrib/shpimporter/jetties.py @ 8709:71ae18553377
(issue1754) Also fixup master artifact for already loaded recommendations
As the load many call creates (and needs to create) all artifacts
in the reccomendations this would break "old" recommendations after
a reload through the "Pencil" mode.
A better fix would probably be to just set the master artifiact
in the recommendations but due to the extremly weird interaction
with the server at this point it is not trivial to identify how
this can be done.
| author | Andre Heinecke <andre.heinecke@intevation.de> |
|---|---|
| date | Fri, 24 Apr 2015 15:57:23 +0200 |
| parents | 5aa05a7a34b7 |
| children |
| rev | line source |
|---|---|
| 5353 | 1 # -*- coding: utf-8 -*- |
| 2 import os | |
| 3 | |
| 4 try: | |
| 5 from osgeo import ogr | |
| 6 except ImportError: | |
| 7 import ogr | |
| 8 | |
| 9 from importer import Importer | |
| 10 import utils | |
| 11 | |
| 12 import logging | |
| 13 logger = logging.getLogger("Jetties") | |
| 14 | |
| 15 PATH="Geodaesie/Bauwerke" | |
| 16 NAME="Jetties" | |
| 17 | |
| 18 # strings need to be lowercase | |
|
5357
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
19 # buhnenkopf 0 |
|
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
20 # buhnenfuß 1 |
|
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
21 # buhnenwurzel 2 |
| 5353 | 22 JETTY_KIND = { |
|
5357
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
23 "bkl" : 0, |
|
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
24 "bkr" : 0, |
|
5384
5a42ca06b93e
Importer act according to specs when importing jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5357
diff
changeset
|
25 "bk" : 0, |
|
5357
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
26 "bfl" : 1, |
|
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
27 "bfr" : 1, |
|
5384
5a42ca06b93e
Importer act according to specs when importing jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5357
diff
changeset
|
28 "bf" : 1, |
|
5357
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
29 "bwl" : 2, |
|
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
30 "bwr" : 2, |
|
5384
5a42ca06b93e
Importer act according to specs when importing jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5357
diff
changeset
|
31 "bw" : 2, |
| 5353 | 32 } |
| 33 | |
| 34 class Jetties(Importer): | |
| 35 fieldmap = { | |
|
5384
5a42ca06b93e
Importer act according to specs when importing jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5357
diff
changeset
|
36 "^station$" : "km", |
|
5a42ca06b93e
Importer act according to specs when importing jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5357
diff
changeset
|
37 "^km$" : "km", |
|
5a42ca06b93e
Importer act according to specs when importing jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5357
diff
changeset
|
38 "^wsv-km$" : "km", |
|
5a42ca06b93e
Importer act according to specs when importing jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5357
diff
changeset
|
39 "^z$" : "z", |
|
5a42ca06b93e
Importer act according to specs when importing jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5357
diff
changeset
|
40 "^H[oeö]{0,2}he$" : "z", |
|
5a42ca06b93e
Importer act according to specs when importing jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5357
diff
changeset
|
41 "^m+NHN$" : "z", |
| 5353 | 42 } |
| 43 | |
| 44 def getPath(self, base): | |
| 45 return "%s/%s" % (base, PATH) | |
| 46 | |
| 47 def getTablename(self): | |
| 48 return "jetties" | |
| 49 | |
| 50 def getName(self): | |
| 51 return "JETTIES" | |
| 52 | |
| 53 def isGeometryValid(self, geomType): | |
| 54 return geomType == ogr.wkbPoint or geomType == ogr.wkbPoint25D | |
| 55 | |
| 56 def isShapeRelevant(self, name, path): | |
| 57 if not path.endswith("Buhnen.shp"): | |
| 58 return False | |
| 59 shp = ogr.Open(path) | |
| 60 return self.isGeometryValid(shp.GetLayerByName(name).GetGeomType()) | |
| 61 | |
| 62 def createNewFeature(self, featureDef, feat, **args): | |
| 63 newFeat = ogr.Feature(featureDef) | |
| 64 geometry = feat.GetGeometryRef() | |
| 65 geometry.SetCoordinateDimension(2) | |
| 66 | |
| 67 self.copyFields(feat, newFeat, self.fieldmap) | |
| 68 | |
| 69 newFeat.SetGeometry(geometry) | |
| 70 | |
|
5438
91b14ade2149
Set river_id for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5384
diff
changeset
|
71 newFeat.SetField("river_id", self.river_id) |
|
91b14ade2149
Set river_id for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5384
diff
changeset
|
72 |
| 5353 | 73 artname = self.searchField("^type$") |
| 74 if self.IsFieldSet(feat, artname): | |
| 75 self.handled(artname) | |
| 76 kind_id = JETTY_KIND.get(feat.GetField(artname).lower()) | |
|
5357
776427b5aa76
Add updated type mapping for jetties
Andre Heinecke <aheinecke@intevation.de>
parents:
5353
diff
changeset
|
77 if kind_id == None: |
| 5353 | 78 logger.warn("Unknown Type: %s" % \ |
| 79 feat.GetField(artname)) | |
| 80 else: | |
| 81 newFeat.SetField("kind_id", kind_id) | |
| 82 | |
| 83 return newFeat | |
| 84 |