def notebook_to_md()

in tools/notebook_to_md.py [0:0]

• 28 lines of code
• 7 McCabe index (conditional complexity)

def notebook_to_md(nbfile, output_dir):
    import nbformat, nbconvert, os, shutil
    from collections import OrderedDict
    from base64 import b64encode
    
    nb = nbformat.read(nbfile, as_version=nbformat.NO_CONVERT)

    dir_name, nb_name = os.path.split(nbfile)
    base_nb_name      = os.path.splitext(nb_name)[0]    

    exporter = nbconvert.get_exporter("markdown")
    resources = {'output_files_dir': base_nb_name + "_files", "build_dir": output_dir}
    
    mdtext, resources = nbconvert.export(exporter, nb=nb, resources=resources)

    # Handle metadata specific to docs.microsoft.com
    meta = nb["metadata"].get("ms_docs_meta", {})
    docs_meta = OrderedDict()
    for key in ['title', 'description', 'services', 'author', 'manager',  'ms.service', 'ms.technology', 'ms.topic', 'ms.date', 'ms.author']:
        if key in meta:
            docs_meta[key] = meta[key]

    if len(docs_meta) > 0:
        docs_meta_md = '\n'.join(['---'] + [': '.join(kv) for kv in docs_meta.items()] + ['---'])
        mdtext = docs_meta_md + '\n' + mdtext

    # Replace images containing paths to file with base64-encoded inline images
    for k,v in resources["outputs"].items():
        ext = os.path.splitext(k)[-1][1:]
        if ext in ("png", "jpg", "svg"):
            b64 = b64encode(v)
            v2  = "data:{0};base64,{1}".format(ext, b64.decode("ascii"))
            mdtext = mdtext.replace(k, v2)
        
    writer = nbconvert.writers.FilesWriter()
    writer.write(mdtext, resources, base_nb_name)
    shutil.rmtree(resources["output_files_dir"], ignore_errors=True)
    
    return mdtext, resources