Skip to content

Commit

Permalink
Make add navigation tool print progress
Browse files Browse the repository at this point in the history
  • Loading branch information
jakevdp committed Nov 18, 2016
1 parent 9f42681 commit 7b5f7a4
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions tools/add_navigation.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,11 +36,14 @@ def iter_navbars():
def write_navbars():
for nb_name, navbar in iter_navbars():
nb = nbformat.read(nb_name, as_version=4)
nb_file = os.path.basename(nb_name)
is_comment = lambda cell: cell.source.startswith(NAV_COMMENT)

if is_comment(nb.cells[1]):
print("- amending navbar for {0}".format(nb_file))
nb.cells[1].source = navbar
else:
print("- inserting navbar for {0}".format(nb_file))
nb.cells.insert(1, new_markdown_cell(source=navbar))

if is_comment(nb.cells[-1]):
Expand Down

0 comments on commit 7b5f7a4

Please sign in to comment.