jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 03:07:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix YAML dump (#7388)

Browse Source 
* Fix YAML dump

* Add test
This commit is contained in:
Paulus Schoutsen 2017-05-01 20:09:49 -07:00 committed by GitHub
parent fafd0d4e4c
commit da2521a299
2 changed files with 26 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
homeassistant/util/yaml.py
View File

@ -20,18 +20,20 @@ _SECRET_YAML = 'secrets.yaml'
__SECRET_CACHE = {} # type: Dict
def _add_reference(obj, loader, node):
"""Add file reference information to an object."""
class NodeListClass(list):
"""Wrapper class to be able to add attributes on a list."""
pass
class NodeStrClass(str):
"""Wrapper class to be able to add attributes on a string."""
pass
def _add_reference(obj, loader, node):
"""Add file reference information to an object."""
if isinstance(obj, list):
obj = NodeListClass(obj)
if isinstance(obj, str):
@ -305,4 +307,9 @@ def represent_odict(dump, tag, mapping, flow_style=None):
yaml.SafeDumper.add_representer(
OrderedDict,
lambda dumper, value:
represent_odict(dumper, u'tag:yaml.org,2002:map', value))
represent_odict(dumper, 'tag:yaml.org,2002:map', value))
yaml.SafeDumper.add_representer(
NodeListClass,
lambda dumper, value:
dumper.represent_sequence('tag:yaml.org,2002:seq', value))

8
tests/util/test_yaml.py
View File

@ -385,3 +385,11 @@ class TestSecrets(unittest.TestCase):
load_yaml(self._yaml_path, 'api_password: !secret pw')
assert mock_error.call_count == 1, \
"Expected an error about logger: value"
def test_representing_yaml_loaded_data():
"""Test we can represent YAML loaded data."""
files = {YAML_CONFIG_FILE: 'key: [1, "2", 3]'}
with patch_yaml_files(files):
data = load_yaml_config_file(YAML_CONFIG_FILE)
assert yaml.dump(data) == "key:\n- 1\n- '2'\n- 3\n"