Added check to see if requested layer is in the list.

contributes to #193
This commit is contained in:
Jaime van Kessel 2015-08-24 10:34:40 +02:00
parent 752a4b26fa
commit a85347134e

View file

@ -28,7 +28,8 @@ class LayerData(MeshData):
self._layers[layer].polygons.append(p) self._layers[layer].polygons.append(p)
def getLayer(self, layer): def getLayer(self, layer):
return self._layers[layer] if layer in self._layers:
return self._layers[layer]
def getLayers(self): def getLayers(self):
return self._layers return self._layers