import json, fire, re

from pathlib import Path

def is_export(cell):

    if cell['cell_type'] != 'code': return False

    src = cell['source']

    if len(src) == 0 or len(src[0]) < 7: return False

    return re.match(r'^\s*#\s*export\s*$', src[0], re.IGNORECASE) is not None

def notebook2script(fname):

    fname = Path(fname)

    fname_out = f'nb_{fname.stem.split("_")[0]}.py'

    main_dic = json.load(open(fname, 'r'))

    code_cells = [c for c in main_dic['cells'] if is_export(c)]

    module = f'''

#################################################

### THIS FILE WAS AUTOGENERATED! DO NOT EDIT! ###

#################################################

# file to edit: dev_nb/{fname.name}

'''

    for cell in code_cells: module += ''.join(cell['source'][1:]) + '\n\n'

    # remove trailing spaces

    module = re.sub(r' +$', '', module, flags=re.MULTILINE)

    open(fname.parent/'exp'/fname_out,'w').write(module[:-2])

    print(f"Converted {fname} to {fname_out}")

if __name__ == '__main__': fire.Fire(notebook2script)