srsly
Modern high-performance serialization utilities for Python
Description
<a href="https://explosion.ai"><img src="https://explosion.ai/assets/img/logo.svg" width="125" height="125" align="right" /></a>
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 had 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:
ujsonmsgpackmsgpack-numpycloudpickleruamel.yaml(without unsafe implementations!)
Installation
⚠️ Note that
v2.xis 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.
python -m pip install -U pip setuptools wheel
python -m 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 with a Python distribution including header files, a compiler (XCode command-line tools on macOS / OS X or Visual C++ build tools on Windows), pip and git installed.
Install from source:
# clone the repo
git clone https://github.com/explosion/srsly
cd srsly
# create a virtual environment
python -m venv .env
source .env/bin/activate
# update pip
python -m pip install -U pip setuptools wheel
# compile and install from source
python -m pip install .
For developers, install requirements separately and then install in editable mode without build isolation:
# install in editable mode
python -m pip install -r requirements.txt
python -m pip install --no-build-isolation --editable .
# run test suite
python -m pytest --pyargs srsly
API
JSON
📦 The underlying module is exposed via
srsly.ujson. However, we normally interact with it via the utility functions only.
<kbd>function</kbd> srsly.json_dumps
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. |
<kbd>function</kbd> srsly.json_loads
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. |
<kbd>function</kbd> srsly.write_json
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. |
<kbd>function</kbd> srsly.read_json
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. |
<kbd>function</kbd> srsly.write_gzip_json
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. |
<kbd>function</kbd> srsly.write_gzip_jsonl
Create a gzipped JSONL file and dump contents.
data = [{"foo": "bar"}, {"baz": 123}]
srsly.write_gzip_json("/path/to/file.jsonl.gz", data)
| Argument | Type | Description |
|---|---|---|
path | str / Path | The file path. |
lines | - | The JSON-serializable contents of each line. |
append | bool | Whether or not to append to the location. Appending to .gz files is generally not recommended, as it doesn't allow the algorithm to take advantage of all data when compressing - files may hence be poorly compressed. |
append_new_line | bool | Whether or not to write a new line before appending to the file. |
<kbd>function</kbd> srsly.read_gzip_json
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. |
<kbd>function</kbd> srsly.read_gzip_jsonl
Load gzipped JSONL from a file.
data = srsly.read_gzip_jsonl("/path/to/file.jsonl.gz")
| Argument | Type | Description |
|---|---|---|
path | str / Path | The file path. |
| RETURNS | dict / list | The loaded JSONL content. |
<kbd>function</kbd> srsly.write_jsonl
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 |