show_results.py

"""
Hide or reveal the solutions to the exercies by adding or removing the remove_cell tag for each cell labeled loesung.

Demo:
to reveal solutions start the program:
	python show_results.py 

and specify a path when prompted:
	01_computer/01_zahlendarstellung
	
press return again when the path is correct or cancel with strg + c


Demo 2:
to hide solutions just run:
	python show_results.py hide
	
then same steps like above

"""


import sys
import nbformat as nbf
from glob import glob

# Collect a list of all notebooks in the content folder

hide = False
if len(sys.argv) >1 and sys.argv[1] == 'hide':
    hide = True
path = "./book/content/Exercises/"
print("The path can further be specified. Finish with Return.")
while True:
    folder = input("Change all notebooks in " + path)
    if folder is '':
        break
    path += folder + "/"
notebooks = glob(path + "**/*.ipynb", recursive=True)

# Search through each notebook and look for the loesung tag
for ipath in notebooks:
    ntbk = nbf.read(ipath, nbf.NO_CONVERT)

    for cell in ntbk.cells:
        cell_tags = cell.get('metadata', {}).get('tags', [])
        if 'loesung' in cell_tags:
            if hide and 'remove_cell' not in cell_tags:
                cell_tags.append('remove_cell')
                print('Hid cell in ' + ipath)
            elif not hide:
                if 'remove_cell' in cell_tags:
                    cell_tags.remove('remove_cell')
                    print('Revealed cell in' + ipath)
# Adds toggle to code cells. does not work for markdown                   
#                if 'hide-cell' not in cell_tags:
#                	cell_tags.append('hide-cell')
        if len(cell_tags) > 0:
            cell['metadata']['tags'] = cell_tags

    nbf.write(ntbk, ipath)