YAML in Python
Internal
PyYAML
PyYAML provides YAML serialization/deserialization in Python.
Installation
pip install pyyaml
requirements.txt
:
pyyaml == 5.3.1
To install, see the "Installation" section from https://pyyaml.org/wiki/PyYAMLDocumentation.
Overview
Concepts
PyYAML core model is centered on constructors, representers and tags.
Constructor
A constructor allows you to take a serialized YAML node and return a class instance.
Representer
A representer is user-defined function that intercepts, as part of the YAML serialization process, data object instances to be serialized, optionally processes them, and then messages the Dumper
instance received as argument to create the proper serialized representation for the given data object instance, as a Node
instance. The instance thus created is returned as result of the representer function, contributing to the serialization result. The representer gets the Dumper
instance as a first argument, and the data object as the second.
This is an implementation of a representer that uses the literal block scalar representation for multi-line strings, and the default plain flow scalar representation for everything else.
def custom_scalar_representer(dumper: Dumper, data: str):
# for multi-line strings, use the literal block scalar representation, use the default otherwise
if data and '\n' in data:
return dumper.represent_scalar('tag:yaml.org,2002:str', data, '|')
else:
return dumper.represent_scalar('tag:yaml.org,2002:str', data)
In this case, tag:yaml.org,2002:str
is the tag name for string nodes.
The representers are registered with add_representer()
. Representers can be added for specific types (such as str
or int
):
yaml.add_representer(str, custom_scalar_representer)
Restoring the Default Representer
There may be situations when we want to restore the representer that was replaced by a add_representer()
method. This can be achieved by accessing directly the representer storage yaml.representer.Representer.yaml_representers
:
default_representer = yaml.representer.Representer.yaml_representers[str]
# replace
yaml.add_representer(str, custom_scalar_representer)
yaml_string = yaml.dump(data)
# restore
yaml.add_representer(str, default_representer)
yaml_string = yaml.dump(data)
Tag
The tag uses the special character !
preceding the tag name to label a YAML node.
A tag helps PyYAML to know which constructor or representer to call.
Deserialize YAML
import yaml
with open('some-file.yaml', 'rt') as f:
content = f.read()
data = yaml.safe_load(content)
# alternative:
data = yaml.load(content, Loader=yaml.Loader)
Serialize YAML
Suggestions on how to safely recursively navigate a complex data structure: