WIP tutorial cell updater tool so that we can inject headers/footers

asumagic · asumagic · commit f705c8bff8f1 · 2024-09-13T16:22:27.000+02:00
diff --git a/tools/tutorial-cell-updater.py b/tools/tutorial-cell-updater.py
@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+
+import nbformat
+import glob
+import logging
+import json
+logging.basicConfig(level=logging.INFO)
+
+def update_notebook(fname):
+    logging.info(f"Updating {fname}")
+
+    with open(fname) as f:
+        nb = json.load(f)
+        
+        cells = nb["cells"]
+        
+        for cell in cells:
+            print(list(cell.keys()))
+
+    with open(fname, "w") as wf:
+        json.dump(nb, wf, indent=1, ensure_ascii=False)
+        print(file=wf) # print final newline that jupyter adds apparently
+
+if __name__ == "__main__":
+    for fname in glob.glob("./tutorials/**/*.ipynb", recursive=True):
+        update_notebook(fname)
+        break
