try catch prevent Uncaught TypeError (#36)

blueswen · May 3, 2024 · dc38a6c · dc38a6c
1 parent 7d2c0ab
commit dc38a6c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/mkdocs_glightbox/plugin.py b/mkdocs_glightbox/plugin.py
@@ -96,8 +96,16 @@ def on_post_page(self, output, page, config, **kwargs):
         js_code = ""
         if self.using_material_privacy:
             js_code += """document.querySelectorAll('.glightbox').forEach(function(element) {
-    var imgSrc = element.querySelector('img').src;
-    element.setAttribute('href', imgSrc);
+    try {
+        var img = element.querySelector('img');
+        if (img && img.src) {
+            element.setAttribute('href', img.src);
+        } else {
+            console.log('No img element with src attribute found');
+        }
+    } catch (error) {
+        console.log('Error:', error);
+    }
 });
 """
         js_code += f"const lightbox = GLightbox({json.dumps(lb_config)});\n"

diff --git a/tests/fixtures/docs/edge_cases.md b/tests/fixtures/docs/edge_cases.md
@@ -0,0 +1 @@
+<p class="glightbox">Not an image</p>