node_hierarchy/parser/ShapesParser.py

19 lines
717 B
Python
Raw Permalink Normal View History

2016-05-29 18:56:50 +02:00
from parser.JsonParser import JsonParser
from shapely.geometry import shape
class ShapesParser(JsonParser):
def __init__(self, filePath, targetName):
super().__init__(filePath.rstrip('/') + '/' + targetName + '.geojson')
self.shapes = self.__createShapes__()
def __createShapes__(self):
shapes = []
if 'features' in self.__jsonData__:
for feature in self.__jsonData__['features']:
if feature['geometry']:
shapes.append(shape(feature['geometry']))
elif 'geometries' in self.__jsonData__:
for geometry in self.__jsonData__['geometries']:
shapes.append(shape(geometry))
2016-05-29 18:56:50 +02:00
return shapes