def extract_ipynb(file_, exclude_output: bool): """ parse and extract ipynb files Args: file_ - file-like object opened in binary mode (+b) Returns: html - html version of notebook info - unmodified (is also passed in) """ # local import reduces amortized latency, saves memory from nbconvert import HTMLExporter import nbformat # get the file size file_.seek(0, os.SEEK_END) size = file_.tell() if size > LAMBDA_MAX_OUT: exclude_output = True # rewind file_.seek(0, os.SEEK_SET) html_exporter = HTMLExporter() html_exporter.template_file = 'basic' html_exporter.exclude_output = exclude_output notebook = nbformat.read(file_, 4) html, _ = html_exporter.from_notebook_node(notebook) return html, {}
def extract_ipynb(file_, exclude_output): """ parse and extract ipynb files Args: file_ - file-like object opened in binary mode (+b) Returns: html - html version of notebook info - unmodified (is also passed in) """ # local import reduces amortized latency, saves memory from nbconvert import HTMLExporter import nbformat html_exporter = HTMLExporter() html_exporter.template_file = 'basic' html_exporter.exclude_output = exclude_output notebook = nbformat.read(file_, 4) html, _ = html_exporter.from_notebook_node(notebook) return html, {}