import os
import json
notebooks = os.listdir()
for notebook in notebooks:
    if 'Challenge' in notebook or 'Graph' in notebook:
        print(notebook) 
        with open(notebook) as json_file:
            data = json.load(json_file)
            finalCellData = []
            for cell in data['cells']:               
                if cell['cell_type'] == "code":
                    cell['source'] = cell['source'][0:2]
                finalCellData.append(cell)
                if cell['cell_type'] == 'markdown':
                    if "# Competition Unstructured Extension\n" in cell['source']:
                        break
            data['cells'] = finalCellData 

        with open(notebook, 'w') as outfile:
            json.dump(data, outfile)
2021-06-06-Party_Resolution_Graph_Example.ipynb
2021-06-06-Challenge_3_Proactive_Analysis_into_Transaction_Parties.ipynb
2021-06-06-Challenge_2_Reactive_Analysis_into_Reporting_Entities.ipynb
2021-06-06-Challenge_1_Proactive_Analysis_into_Reporting_Patterns.ipynb
2021-06-06-Challenge_4_Reactive_Analysis_into_Transaction_Parties.ipynb