pbdl-book/json-ipynb-removeImgs.py
2021-06-30 19:35:47 +02:00

40 lines
1.0 KiB
Python

import json
# for spell checking single files
fnOut="temp_reduced.txt"
fn="temp.txt"
with open(fn) as file:
d = json.load(file)
print(d.keys())
#print(d["cells"][0].keys())
t="cells"
for i in range(len(d[t])):
#for i in range(len(d[t])):
#print(d[t][0]["cell_type"])
#print(d[t][i]["cell_type"])
# remove images after code
if d[t][i]["cell_type"]=="code":
#print(d[t][i].keys())
#d[t][i]["outputs"] = ""
#print(d[t][i]["outputs"])
#print(len( d[t][i]["outputs"] ))
for j in range(len( d[t][i]["outputs"] )):
#print(type( d[t][i]["outputs"][j] ))
print( d[t][i]["outputs"][j].keys() )
if d[t][i]["outputs"][j]["output_type"]=="display_data":
#if d[t][i]["outputs"][j]["data"]=="display_data":
print( d[t][i]["outputs"][j]["data"].keys() )
#if( d[t][i]["outputs"][j]["data"].contains('image/png') ):
if( 'image/png' in d[t][i]["outputs"][j]["data"].keys() ):
d[t][i]["outputs"][j]["data"]['image/png'] = ""
with open(fnOut,'w') as fileOut:
json.dump(d,fileOut, indent=1, sort_keys=True)