bugfix - handle raster layer when getting features

openscope · Jan 13, 2020 · f37b6e6 · f37b6e6
1 parent 710b65d
commit f37b6e6
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/OpenScope/utilities/drawing.py b/OpenScope/utilities/drawing.py
@@ -5,6 +5,7 @@
     QgsDistanceArea,
     QgsFeature,
     QgsGeometry,
+    QgsMapLayer,
     QgsProject,
     QgsWkbTypes
 )
@@ -177,11 +178,16 @@ def _getSelectedFeatures():
     """Gets all the selected features"""
     features = []
 
-    for layer in QgsProject.instance().layerTreeRoot().findLayers():
-        if not layer.layer().selectedFeatureCount():
+    for item in QgsProject.instance().layerTreeRoot().findLayers():
+        layer = item.layer()
+
+        if layer.type() != QgsMapLayer.VectorLayer:
+            continue
+
+        if not layer.selectedFeatureCount():
             continue
 
-        features.extend(layer.layer().selectedFeatures())
+        features.extend(layer.selectedFeatures())
 
     return features