Riccardo Pittau
bb1c2ed11f
In ruamel there's no option to avoid a trailing whitespace after wrapping lines, so we need to post process the output to remove that if present. Change-Id: Iced01614c7dda0d02195a806733c9857306f07fc Signed-off-by: Riccardo Pittau <elfosardo@gmail.com>
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import ruamel.yaml
|
|
import ruamel.yaml.compat
|
|
|
|
|
|
def dumps(obj):
|
|
"""Dumps yaml content into a string."""
|
|
yaml = ruamel.yaml.YAML()
|
|
yaml.width = 66
|
|
|
|
stream = ruamel.yaml.compat.StringIO()
|
|
yaml.explicit_start = True
|
|
yaml.indent(mapping=2, sequence=4, offset=2)
|
|
yaml.dump(obj, stream)
|
|
|
|
# Remove trailing whitespace.
|
|
# NOTE(rpittau): This can be removed once the bug introduced
|
|
# in ruamel 0.18.13 is fixed.
|
|
yaml_output = stream.getvalue()
|
|
lines = yaml_output.split('\n')
|
|
cleaned_yaml_lines = [line.rstrip() for line in lines]
|
|
return '\n'.join(cleaned_yaml_lines)
|
|
|
|
|
|
def loads(blob):
|
|
"""Load a yaml blob and retain key ordering."""
|
|
yaml = ruamel.yaml.YAML()
|
|
return yaml.load(blob)
|