Making the code a little more robust

HPC-workshops · Jul 13, 2016 · cf51440 · cf51440
1 parent 466c28d
commit cf51440
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bin/extract_figures.py b/bin/extract_figures.py
@@ -58,7 +58,8 @@ def find_image_nodes(doc, result):
 
     if (doc['type'] == 'img') or \
        ((doc['type'] == 'html_element') and (doc['value'] == 'img')):
-        result.append({'alt': doc['attr']['alt'], 'src': doc['attr']['src']})
+        alt = doc['attr'].get('alt', '')
+        result.append({'alt': alt, 'src': doc['attr']['src']})
     else:
         for child in doc.get('children', []):
             find_image_nodes(child, result)