From 3557aac1d8cf5df140667c146005223cafdb1ddf Mon Sep 17 00:00:00 2001 From: Ross McFarland Date: Tue, 6 Sep 2022 15:17:58 -0700 Subject: [PATCH] Allow our custom yaml config to dump value types --- octodns/yaml.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/octodns/yaml.py b/octodns/yaml.py index 5b02431..bfbc3a3 100644 --- a/octodns/yaml.py +++ b/octodns/yaml.py @@ -11,6 +11,7 @@ from __future__ import ( from natsort import natsort_keygen from yaml import SafeDumper, SafeLoader, load, dump +from yaml.representer import SafeRepresenter from yaml.constructor import ConstructorError @@ -61,6 +62,10 @@ class SortingDumper(SafeDumper): SortingDumper.add_representer(dict, SortingDumper._representer) +# This should handle all the record value types which are ultimately either str +# or dict at some point in their inheritance hierarchy +SortingDumper.add_multi_representer(str, SafeRepresenter.represent_str) +SortingDumper.add_multi_representer(dict, SortingDumper._representer) def safe_dump(data, fh, **options):