from util import Reporter, read_markdown
+# Things an image file's name can end with.
+PATH_SUFFICES = {
+ '.gif',
+ '.jpg',
+ '.png',
+ '.svg'
+}
+
+
def main():
"""Main driver."""
content = read_markdown(parser, filename)
result = []
find_image_nodes(content['doc'], result)
+ find_image_links(content['doc'], result)
return result
find_image_nodes(child, result)
+def find_image_links(doc, result):
+ """Find all links to files in the 'fig' directory."""
+
+ if (doc['type'] == 'a') and ('attr' in doc) and ('href' in doc['attr']):
+ path = doc['attr']['href']
+ if os.path.splitext(path)[1].lower() in PATH_SUFFICES:
+ result.append({'alt':'', 'src': doc['attr']['href']})
+ else:
+ for child in doc.get('children', []):
+ find_image_links(child, result)
+
+
def save(stream, images):
"""Save results as Markdown."""