bug with listing Transforms

specklesystems · Nov 25, 2024 · 58e959d · 58e959d
1 parent daa91f7
commit 58e959d
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/speckle/ui_widgets/widget_transforms.py b/speckle/ui_widgets/widget_transforms.py
@@ -17,6 +17,8 @@
 import webbrowser
 import specklepy_qt_ui.qt_ui
 
+from plugin_utils.helpers import UNSUPPORTED_PROVIDERS
+
 FORM_CLASS, _ = uic.loadUiType(
     os.path.join(
         os.path.join(
@@ -220,9 +222,11 @@ def populateLayersByTransform(self):
 
                 elif "elevation" in transform.lower():
                     if isinstance(layer, QgsRasterLayer):
-                        # avoiding tiling layers
-                        ds = gdal.Open(layer.source(), gdal.GA_ReadOnly)
-                        if ds is None:
+                        # avoiding Tile and WMS layers
+                        data_provider_type = (
+                            layer.providerType()
+                        )  # == ogr, memory, gdal, delimitedtext
+                        if data_provider_type in UNSUPPORTED_PROVIDERS:
                             continue
 
                         # for satellites
@@ -309,9 +313,11 @@ def populateSavedElevationLayer(
             countRaster = 1
             for i, layer in enumerate(self.dataStorage.all_layers):
                 if isinstance(layer, QgsRasterLayer):
-                    # avoiding tiling layers
-                    ds = gdal.Open(layer.source(), gdal.GA_ReadOnly)
-                    if ds is None:
+                    # avoiding Tile and WMS layers
+                    data_provider_type = (
+                        layer.providerType()
+                    )  # == ogr, memory, gdal, delimitedtext
+                    if data_provider_type in UNSUPPORTED_PROVIDERS:
                         continue
                     elif layer.bandCount() != 1:
                         continue
@@ -382,7 +388,7 @@ def saveElevationLayer(self) -> "DataStorage":
                         )
                         layer = None
                         break
-                    
+
         set_elevationLayer(self.dataStorage)
 
         try: