|
|
import os.path, re, nbformat, jupyter_contrib_nbextensions |
|
|
from nbconvert.preprocessors import Preprocessor |
|
|
from nbconvert import HTMLExporter |
|
|
from traitlets.config import Config |
|
|
from pathlib import Path |
|
|
|
|
|
__all__ = ['read_nb', 'convert_nb', 'convert_all'] |
|
|
|
|
|
exporter = HTMLExporter(Config()) |
|
|
exporter.exclude_input_prompt=True |
|
|
exporter.exclude_output_prompt=True |
|
|
|
|
|
exporter.template_file = 'jekyll.tpl' |
|
|
path = Path(__file__).parent |
|
|
exporter.template_path.append(str(path)) |
|
|
|
|
|
def read_nb(fname): |
|
|
"Read the notebook in `fname`." |
|
|
with open(fname,'r') as f: return nbformat.reads(f.read(), as_version=4) |
|
|
|
|
|
def convert_nb(fname, dest_path='.'): |
|
|
"Convert a notebook `fname` to html file in `dest_path`." |
|
|
from .gen_notebooks import remove_undoc_cells, remove_code_cell_jupyter_widget_state_elem |
|
|
nb = read_nb(fname) |
|
|
nb['cells'] = remove_undoc_cells(nb['cells']) |
|
|
nb['cells'] = remove_code_cell_jupyter_widget_state_elem(nb['cells']) |
|
|
fname = Path(fname).absolute() |
|
|
dest_name = fname.with_suffix('.html').name |
|
|
meta = nb['metadata'] |
|
|
meta_jekyll = meta['jekyll'] if 'jekyll' in meta else {'title': fname.with_suffix('').name} |
|
|
meta_jekyll['nb_path'] = f'{fname.parent.name}/{fname.name}' |
|
|
with open(f'{dest_path}/{dest_name}','w') as f: |
|
|
f.write(exporter.from_notebook_node(nb, resources=meta_jekyll)[0]) |
|
|
|
|
|
def convert_all(folder, dest_path='.', force_all=False): |
|
|
"Convert modified notebooks in `folder` to html pages in `dest_path`." |
|
|
path = Path(folder) |
|
|
|
|
|
changed_cnt = 0 |
|
|
for fname in path.glob("*.ipynb"): |
|
|
|
|
|
fname_out = Path(dest_path)/fname.with_suffix('.html').name |
|
|
if not force_all and fname_out.exists(): |
|
|
in_mod = os.path.getmtime(fname) |
|
|
out_mod = os.path.getmtime(fname_out) |
|
|
if in_mod < out_mod: continue |
|
|
|
|
|
print(f"converting: {fname} => {fname_out}") |
|
|
changed_cnt += 1 |
|
|
convert_nb(fname, dest_path=dest_path) |
|
|
if not changed_cnt: print("No notebooks were modified") |
|
|
|
