jupyter_cache_quickstart_init_add_execute_notebook.py

1import nbformat
2from jupyter_cache import get_cache
3
4# Initialize the cache (creates a folder named .jupyter_cache)
5cache = get_cache(".jupyter_cache")
6
7# Create a sample notebook
8nb = nbformat.v4.new_notebook()
9nb.cells.append(nbformat.v4.new_code_cell("print('hello world')"))
10
11# Add the notebook to the cache
12# This returns a record of the notebook, including its hash
13record = cache.add_notebook(nb)
14
15# Execute the notebook and cache the result
16# The cache will only execute the notebook if a valid result is not already present
17cache.execute_notebook(record.pk)
18
19# Retrieve the executed notebook from the cache
20cached_nb_bundle = cache.get_notebook(record.pk)
21print(cached_nb_bundle.nb.cells[0].outputs[0].text)