File size: 5,574 Bytes
d1ceb73 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 |
"""The basic dict based notebook format.
The Python representation of a notebook is a nested structure of
dictionary subclasses that support attribute access.
The functions in this module are merely
helpers to build the structs in the right form.
Authors:
* Brian Granger
"""
# -----------------------------------------------------------------------------
# Copyright (C) 2008-2011 The IPython Development Team
#
# Distributed under the terms of the BSD License. The full license is in
# the file LICENSE, distributed as part of this software.
# -----------------------------------------------------------------------------
# -----------------------------------------------------------------------------
# Imports
# -----------------------------------------------------------------------------
from __future__ import annotations
from nbformat._struct import Struct
# -----------------------------------------------------------------------------
# Code
# -----------------------------------------------------------------------------
class NotebookNode(Struct):
"""A notebook node object."""
def from_dict(d):
"""Create notebook node(s) from a value."""
if isinstance(d, dict):
newd = NotebookNode()
for k, v in d.items():
newd[k] = from_dict(v)
return newd
if isinstance(d, (tuple, list)):
return [from_dict(i) for i in d]
return d
def new_output(
output_type=None,
output_text=None,
output_png=None,
output_html=None,
output_svg=None,
output_latex=None,
output_json=None,
output_javascript=None,
output_jpeg=None,
prompt_number=None,
etype=None,
evalue=None,
traceback=None,
):
"""Create a new code cell with input and output"""
output = NotebookNode()
if output_type is not None:
output.output_type = str(output_type)
if output_type != "pyerr":
if output_text is not None:
output.text = str(output_text)
if output_png is not None:
output.png = bytes(output_png)
if output_jpeg is not None:
output.jpeg = bytes(output_jpeg)
if output_html is not None:
output.html = str(output_html)
if output_svg is not None:
output.svg = str(output_svg)
if output_latex is not None:
output.latex = str(output_latex)
if output_json is not None:
output.json = str(output_json)
if output_javascript is not None:
output.javascript = str(output_javascript)
if output_type == "pyout" and prompt_number is not None:
output.prompt_number = int(prompt_number)
if output_type == "pyerr":
if etype is not None:
output.etype = str(etype)
if evalue is not None:
output.evalue = str(evalue)
if traceback is not None:
output.traceback = [str(frame) for frame in list(traceback)]
return output
def new_code_cell(
input=None,
prompt_number=None,
outputs=None,
language="python",
collapsed=False,
):
"""Create a new code cell with input and output"""
cell = NotebookNode()
cell.cell_type = "code"
if language is not None:
cell.language = str(language)
if input is not None:
cell.input = str(input)
if prompt_number is not None:
cell.prompt_number = int(prompt_number)
if outputs is None:
cell.outputs = []
else:
cell.outputs = outputs
if collapsed is not None:
cell.collapsed = bool(collapsed)
return cell
def new_text_cell(cell_type, source=None, rendered=None):
"""Create a new text cell."""
cell = NotebookNode()
if source is not None:
cell.source = str(source)
if rendered is not None:
cell.rendered = str(rendered)
cell.cell_type = cell_type
return cell
def new_worksheet(name=None, cells=None):
"""Create a worksheet by name with with a list of cells."""
ws = NotebookNode()
if name is not None:
ws.name = str(name)
if cells is None:
ws.cells = []
else:
ws.cells = list(cells)
return ws
def new_notebook(metadata=None, worksheets=None):
"""Create a notebook by name, id and a list of worksheets."""
nb = NotebookNode()
nb.nbformat = 2
if worksheets is None:
nb.worksheets = []
else:
nb.worksheets = list(worksheets)
if metadata is None:
nb.metadata = new_metadata()
else:
nb.metadata = NotebookNode(metadata)
return nb
def new_metadata(
name=None,
authors=None,
license=None,
created=None,
modified=None,
gistid=None,
):
"""Create a new metadata node."""
metadata = NotebookNode()
if name is not None:
metadata.name = str(name)
if authors is not None:
metadata.authors = list(authors)
if created is not None:
metadata.created = str(created)
if modified is not None:
metadata.modified = str(modified)
if license is not None:
metadata.license = str(license)
if gistid is not None:
metadata.gistid = str(gistid)
return metadata
def new_author(name=None, email=None, affiliation=None, url=None):
"""Create a new author."""
author = NotebookNode()
if name is not None:
author.name = str(name)
if email is not None:
author.email = str(email)
if affiliation is not None:
author.affiliation = str(affiliation)
if url is not None:
author.url = str(url)
return author
|