def preprocess(self, nb, resources):
"""Concatenates the cells from the header and footer notebooks to the
given cells.
"""
new_cells = []
# header
if self.header:
with open(self.header, 'r') as fh:
header_nb = read_nb(fh, as_version=current_nbformat)
new_cells.extend(header_nb.cells)
# body
new_cells.extend(nb.cells)
# footer
if self.footer:
with open(self.footer, 'r') as fh:
footer_nb = read_nb(fh, as_version=current_nbformat)
new_cells.extend(footer_nb.cells)
nb.cells = new_cells
super(IncludeHeaderFooter, self).preprocess(nb, resources)
return nb, resources
def setup(self):
self.nbs = {}
for basename, filename in self.files.items():
with open(filename, "r") as fh:
self.nbs[basename] = read_nb(fh, as_version=current_nbformat)
def _read_nb(self, filename):
fullpath = os.path.join(os.path.dirname(__file__), filename)
with open(fullpath, "r") as fh:
nb = read_nb(fh, as_version=current_nbformat)
return nb
