nbjson.py
55 lines
| 1.4 KiB
| text/x-python
|
PythonLexer
MinRK
|
r18573 | """Read and write notebooks in JSON format.""" | ||
MinRK
|
r18568 | |||
MinRK
|
r18573 | # Copyright (c) IPython Development Team. | ||
# Distributed under the terms of the Modified BSD License. | ||||
MinRK
|
r18568 | |||
import copy | ||||
import json | ||||
MinRK
|
r18573 | from IPython.utils import py3compat | ||
MinRK
|
r18568 | from .nbbase import from_dict | ||
from .rwbase import ( | ||||
MinRK
|
r18573 | NotebookReader, NotebookWriter, rejoin_lines, split_lines | ||
MinRK
|
r18568 | ) | ||
class BytesEncoder(json.JSONEncoder): | ||||
"""A JSON encoder that accepts b64 (and other *ascii*) bytestrings.""" | ||||
def default(self, obj): | ||||
if isinstance(obj, bytes): | ||||
return obj.decode('ascii') | ||||
return json.JSONEncoder.default(self, obj) | ||||
class JSONReader(NotebookReader): | ||||
def reads(self, s, **kwargs): | ||||
nb = json.loads(s, **kwargs) | ||||
nb = self.to_notebook(nb, **kwargs) | ||||
return nb | ||||
def to_notebook(self, d, **kwargs): | ||||
return rejoin_lines(from_dict(d)) | ||||
class JSONWriter(NotebookWriter): | ||||
def writes(self, nb, **kwargs): | ||||
kwargs['cls'] = BytesEncoder | ||||
kwargs['indent'] = 1 | ||||
kwargs['sort_keys'] = True | ||||
kwargs['separators'] = (',',': ') | ||||
if kwargs.pop('split_lines', True): | ||||
nb = split_lines(copy.deepcopy(nb)) | ||||
return py3compat.str_to_unicode(json.dumps(nb, **kwargs), 'utf-8') | ||||
_reader = JSONReader() | ||||
_writer = JSONWriter() | ||||
reads = _reader.reads | ||||
read = _reader.read | ||||
to_notebook = _reader.to_notebook | ||||
write = _writer.write | ||||
writes = _writer.writes | ||||