fix: viewpoint has no snapshot

bimdata · Oct 17, 2023 · 4da3cea · 4da3cea
1 parent d82b074
commit 4da3cea
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/webhooks/handlers.py b/webhooks/handlers.py
@@ -41,10 +41,16 @@ def _refresh_bcf(self, save: bool) -> None:
             )
         except Exception:
             viewpoints = []
-        if len(viewpoints) > 0:
-            self.notification.payload["topic"]["snapshot_url"] = viewpoints[0][
-                "snapshot"
-            ]["snapshot_data"]
+        snapshot_data = next(
+            (
+                viewpoint["snapshot"]["snapshot_data"]
+                for viewpoint in viewpoints
+                if viewpoint["snapshot"]
+            ),
+            None,
+        )
+        if snapshot_data:
+            self.notification.payload["topic"]["snapshot_url"] = snapshot_data
             if save:
                 self.notification.save()