srsly: Modern high-performance serialization utilities for Python
This package bundles some of the best Python serialization libraries into one standalone package, with a high-level API that makes it easy to write code that's correct across platforms and Pythons. This allows us to provide all the serialization utilities we need in a single binary wheel. Currently supports JSON, JSONL, MessagePack, Pickle and YAML.
Motivation
Serialization is hard, especially across Python versions and multiple platforms. After dealing with many subtle bugs over the years (encodings, locales, large files) our libraries like spaCy and Prodigy have steadily grown a number of utility functions to wrap the multiple serialization formats we need to support (especially json
, msgpack
and pickle
). These wrapping functions ended up duplicated across our codebases, so we wanted to put them in one place.
At the same time, we noticed that having a lot of small dependencies was making maintenance harder, and making installation slower. To solve this, we've made srsly
standalone, by including the component packages directly within it. This way we can provide all the serialization utilities we need in a single binary wheel.
srsly
currently includes forks of the following packages:
ujson
msgpack
msgpack-numpy
cloudpickle
ruamel.yaml
(without unsafe implementations!)
Installation
โ ๏ธ Note thatv2.x
is only compatible with Python 3.6+. For 2.7+ compatibility, usev1.x
.
srsly
can be installed from pip. Before installing, make sure that your pip
, setuptools
and wheel
are up to date.
pip install -U pip setuptools wheel
pip install srsly
Or from conda via conda-forge:
conda install -c conda-forge srsly
Alternatively, you can also compile the library from source. You'll need to make sure that you have a development environment consisting of a Python distribution including header files, a compiler (XCode command-line tools on macOS / OS X or Visual C++ build tools on Windows), pip, virtualenv and git installed.
pip install -r requirements.txt # install development dependencies
python setup.py build_ext --inplace # compile the library
API
JSON
๐ฆ The underlying module is exposed viasrsly.ujson
. However, we normally interact with it via the utility functions only.
srsly.json_dumps
function Serialize an object to a JSON string. Falls back to json
if sort_keys=True
is used (until it's fixed in ujson
).
data = {"foo": "bar", "baz": 123}
json_string = srsly.json_dumps(data)
Argument | Type | Description |
---|---|---|
data |
- | The JSON-serializable data to output. |
indent |
int | Number of spaces used to indent JSON. Defaults to 0 . |
sort_keys |
bool | Sort dictionary keys. Defaults to False . |
RETURNS | str | The serialized string. |
srsly.json_loads
function Deserialize unicode or bytes to a Python object.
data = '{"foo": "bar", "baz": 123}'
obj = srsly.json_loads(data)
Argument | Type | Description |
---|---|---|
data |
str / bytes | The data to deserialize. |
RETURNS | - | The deserialized Python object. |
srsly.write_json
function Create a JSON file and dump contents or write to standard output.
data = {"foo": "bar", "baz": 123}
srsly.write_json("/path/to/file.json", data)
Argument | Type | Description |
---|---|---|
path |
str / Path |
The file path or "-" to write to stdout. |
data |
- | The JSON-serializable data to output. |
indent |
int | Number of spaces used to indent JSON. Defaults to 2 . |
srsly.read_json
function Load JSON from a file or standard input.
data = srsly.read_json("/path/to/file.json")
Argument | Type | Description |
---|---|---|
path |
str / Path |
The file path or "-" to read from stdin. |
RETURNS | dict / list | The loaded JSON content. |
srsly.write_gzip_json
function Create a gzipped JSON file and dump contents.
data = {"foo": "bar", "baz": 123}
srsly.write_gzip_json("/path/to/file.json.gz", data)
Argument | Type | Description |
---|---|---|
path |
str / Path |
The file path. |
data |
- | The JSON-serializable data to output. |
indent |
int | Number of spaces used to indent JSON. Defaults to 2 . |
srsly.read_gzip_json
function Load gzipped JSON from a file.
data = srsly.read_gzip_json("/path/to/file.json.gz")
Argument | Type | Description |
---|---|---|
path |
str / Path |
The file path. |
RETURNS | dict / list | The loaded JSON content. |
srsly.write_jsonl
function Create a JSONL file (newline-delimited JSON) and dump contents line by line, or write to standard output.
data = [{"foo": "bar"}, {"baz": 123}]
srsly.write_jsonl("/path/to/file.jsonl", data)
Argument | Type | Description |
---|---|---|
path |
str / Path |
The file path or "-" to write to stdout. |
lines |
iterable | The JSON-serializable lines. |
append |
bool | Append to an existing file. Will open it in "a" mode and insert a newline before writing lines. Defaults to False . |
append_new_line |
bool | Defines whether a new line should first be written when appending to an existing file. Defaults to True . |
srsly.read_jsonl
function Read a JSONL file (newline-delimited JSON) or from JSONL data from standard input and yield contents line by line. Blank lines will always be skipped.
data = srsly.read_jsonl("/path/to/file.jsonl")
Argument | Type | Description |
---|---|---|
path |
str / Path | The file path or "-" to read from stdin. |
skip |
bool | Skip broken lines and don't raise ValueError . Defaults to False . |
YIELDS | - | The loaded JSON contents of each line. |
srsly.is_json_serializable
function Check if a Python object is JSON-serializable.
assert srsly.is_json_serializable({"hello": "world"}) is True
assert srsly.is_json_serializable(lambda x: x) is False
Argument | Type | Description |
---|---|---|
obj |
- | The object to check. |
RETURNS | bool | Whether the object is JSON-serializable. |
msgpack
๐ฆ The underlying module is exposed viasrsly.msgpack
. However, we normally interact with it via the utility functions only.
srsly.msgpack_dumps
function Serialize an object to a msgpack byte string.
data = {"foo": "bar", "baz": 123}
msg = srsly.msgpack_dumps(data)
Argument | Type | Description |
---|---|---|
data |
- | The data to serialize. |
RETURNS | bytes | The serialized bytes. |
srsly.msgpack_loads
function Deserialize msgpack bytes to a Python object.
msg = b"\x82\xa3foo\xa3bar\xa3baz{"
data = srsly.msgpack_loads(msg)
Argument | Type | Description |
---|---|---|
data |
bytes | The data to deserialize. |
use_list |
bool | Don't use tuples instead of lists. Can make deserialization slower. Defaults to True . |
RETURNS | - | The deserialized Python object. |
srsly.write_msgpack
function Create a msgpack file and dump contents.
data = {"foo": "bar", "baz": 123}
srsly.write_msgpack("/path/to/file.msg", data)
Argument | Type | Description |
---|---|---|
path |
str / Path |
The file path. |
data |
- | The data to serialize. |
srsly.read_msgpack
function Load a msgpack file.
data = srsly.read_msgpack("/path/to/file.msg")
Argument | Type | Description |
---|---|---|
path |
str / Path |
The file path. |
use_list |
bool | Don't use tuples instead of lists. Can make deserialization slower. Defaults to True . |
RETURNS | - | The loaded and deserialized content. |
pickle
๐ฆ The underlying module is exposed viasrsly.cloudpickle
. However, we normally interact with it via the utility functions only.
srsly.pickle_dumps
function Serialize a Python object with pickle.
data = {"foo": "bar", "baz": 123}
pickled_data = srsly.pickle_dumps(data)
Argument | Type | Description |
---|---|---|
data |
- | The object to serialize. |
protocol |
int | Protocol to use. -1 for highest. Defaults to None . |
RETURNS | bytes | The serialized object. |
srsly.pickle_loads
function Deserialize bytes with pickle.
pickled_data = b"\x80\x04\x95\x19\x00\x00\x00\x00\x00\x00\x00}\x94(\x8c\x03foo\x94\x8c\x03bar\x94\x8c\x03baz\x94K{u."
data = srsly.pickle_loads(pickled_data)
Argument | Type | Description |
---|---|---|
data |
bytes | The data to deserialize. |
RETURNS | - | The deserialized Python object. |
YAML
๐ฆ The underlying module is exposed viasrsly.ruamel_yaml
. However, we normally interact with it via the utility functions only.
srsly.yaml_dumps
function Serialize an object to a YAML string. See the ruamel.yaml
docs for details on the indentation format.
data = {"foo": "bar", "baz": 123}
yaml_string = srsly.yaml_dumps(data)
Argument | Type | Description |
---|---|---|
data |
- | The JSON-serializable data to output. |
indent_mapping |
int | Mapping indentation. Defaults to 2 . |
indent_sequence |
int | Sequence indentation. Defaults to 4 . |
indent_offset |
int | Indentation offset. Defaults to 2 . |
sort_keys |
bool | Sort dictionary keys. Defaults to False . |
RETURNS | str | The serialized string. |
srsly.yaml_loads
function Deserialize unicode or a file object to a Python object.
data = 'foo: bar\nbaz: 123'
obj = srsly.yaml_loads(data)
Argument | Type | Description |
---|---|---|
data |
str / file | The data to deserialize. |
RETURNS | - | The deserialized Python object. |
srsly.write_yaml
function Create a YAML file and dump contents or write to standard output.
data = {"foo": "bar", "baz": 123}
srsly.write_yaml("/path/to/file.yml", data)
Argument | Type | Description |
---|---|---|
path |
str / Path |
The file path or "-" to write to stdout. |
data |
- | The JSON-serializable data to output. |
indent_mapping |
int | Mapping indentation. Defaults to 2 . |
indent_sequence |
int | Sequence indentation. Defaults to 4 . |
indent_offset |
int | Indentation offset. Defaults to 2 . |
sort_keys |
bool | Sort dictionary keys. Defaults to False . |
srsly.read_yaml
function Load YAML from a file or standard input.
data = srsly.read_yaml("/path/to/file.yml")
Argument | Type | Description |
---|---|---|
path |
str / Path |
The file path or "-" to read from stdin. |
RETURNS | dict / list | The loaded YAML content. |
srsly.is_yaml_serializable
function Check if a Python object is YAML-serializable.
assert srsly.is_yaml_serializable({"hello": "world"}) is True
assert srsly.is_yaml_serializable(lambda x: x) is False
Argument | Type | Description |
---|---|---|
obj |
- | The object to check. |
RETURNS | bool | Whether the object is YAML-serializable. |