Commit db6e9e8

committed

copy figures as well

1 parent 12a9c39 commit db6e9e8Copy full SHA for db6e9e8

File tree

1 file changed

+32

-11

lines changed

website
- copy_notebooks.py

1 file changed

+32

-11

lines changed

`‎website/copy_notebooks.py`

Lines changed: 32 additions & 11 deletions

Original file line number	Diff line number	Diff line change
`@@ -4,12 +4,13 @@`
`4`	`4`	`"""`
`5`	`5`	`import os`
`6`	`6`	`import nbformat`
	`7`	`+import shutil`
`7`	`8`
`8`	`9`	`PAGEFILE = """title: {title}`
`9`	`10`	`slug: {slug}`
`10`	`11`	`Template: page`
`11`	`12`
`12`		`-{{% notebook notebooks/{notebook_file} cells[2:] %}}`
	`13`	`+{{% notebook notebooks/{notebook_file} cells[{cells}] %}}`
`13`	`14`	`"""`
`14`	`15`
`15`	`16`
`@@ -24,38 +25,58 @@ def abspath_from_here(*args):`
`24`	`25`
`25`	`26`
`26`	`27`	`def copy_notebooks():`
`27`		`- nblist = sorted(os.listdir(NB_SOURCE_DIR))`
	`28`	`+ nblist = sorted(nb for nb in os.listdir(NB_SOURCE_DIR)`
	`29`	`+ if nb.endswith('.ipynb'))`
`28`	`30`	`name_map = {nb: nb.rsplit('.', 1)[0] + '.html'`
`29`	`31`	`for nb in nblist}`
`30`	`32`
	`33`	`+ figsource = abspath_from_here('..', 'notebooks', 'figures')`
	`34`	`+ figdest = abspath_from_here('content', 'figures')`
	`35`	`+`
	`36`	`+ if os.path.exists(figdest):`
	`37`	`+ shutil.rmtree(figdest)`
	`38`	`+ shutil.copytree(figsource, figdest)`
	`39`	`+`
	`40`	`+ figurelist = os.listdir(abspath_from_here('content', 'figures'))`
	`41`	`+ figure_map = {os.path.join('figures', fig) : os.path.join('/figures', fig)`
	`42`	`+ for fig in figurelist}`
	`43`	`+`
`31`	`44`	`for nb in nblist:`
`32`	`45`	`base, ext = os.path.splitext(nb)`
`33`		`- if ext != '.ipynb':`
`34`		`- continue`
`35`	`46`	`print('-', nb)`
`36`	`47`
`37`	`48`	`content = nbformat.read(os.path.join(NB_SOURCE_DIR, nb),`
`38`	`49`	`as_version=4)`
`39`		`- title = content.cells[2].source`
`40`		`- if not title.startswith('#'):`
`41`		`- raise ValueError('title not found in third cell')`
`42`		`- title = title.lstrip('#').strip()`
`43`	`50`
`44`		`- # put nav below title`
`45`		`- content.cells[1], content.cells[2] = content.cells[2], content.cells[1]`
	`51`	`+ if nb == 'Index.ipynb':`
	`52`	`+ cells = '1:'`
	`53`	`+ title = 'Python Data Science Handbook'`
	`54`	`+ else:`
	`55`	`+ cells = '2:'`
	`56`	`+ # put nav below title`
	`57`	`+ title = content.cells[2].source`
	`58`	`+ if not title.startswith('#') or len(title.splitlines()) > 1:`
	`59`	`+ raise ValueError('title not found in third cell')`
	`60`	`+ title = title.lstrip('#').strip()`
	`61`	`+ content.cells[1], content.cells[2] = content.cells[2], content.cells[1]`
`46`	`62`
`47`	`63`	`for cell in content.cells:`
`48`	`64`	`if cell.cell_type == 'markdown':`
`49`	`65`	`for nbname, htmlname in name_map.items():`
`50`	`66`	`if nbname in cell.source:`
`51`	`67`	`cell.source = cell.source.replace(nbname, htmlname)`
	`68`	`+ for figname, newfigname in figure_map.items():`
	`69`	`+ if figname in cell.source:`
	`70`	`+ cell.source = cell.source.replace(figname, newfigname)`
	`71`	`+`
`52`	`72`	`nbformat.write(content, os.path.join(NB_DEST_DIR, nb))`
`53`	`73`
`54`	`74`	`pagefile = os.path.join(PAGE_DEST_DIR, base + '.md')`
`55`	`75`	`with open(pagefile, 'w') as f:`
`56`	`76`	`f.write(PAGEFILE.format(title=title,`
`57`	`77`	`slug=base.lower(),`
`58`		`- notebook_file=nb))`
	`78`	`+ notebook_file=nb,`
	`79`	`+ cells=cells))`
`59`	`80`
`60`	`81`	`if __name__ == '__main__':`
`61`	`82`	`copy_notebooks()`

0 commit comments

Comments

(0)

Navigation Menu

Search code, repositories, users, issues, pull requests...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Commit db6e9e8

File tree

1 file changed

1 file changed

`‎website/copy_notebooks.py`

0 commit comments