Преглед изворни кода

Adjust to_padded_yaml transformation to use the AnsibleDumper

* Previously we used yaml.safe_dump
* Now we use yamp.dump with the `Dumper` parameter set to the
  AnsibleDumper class.
* AnsibleDumper subclasses yaml.SafeDumper, so we aren't losing any
  safety nets
Tim Bielawa пре 8 година
родитељ
комит
1c796356d3
1 измењених фајлова са 4 додато и 1 уклоњено
  1. 4 1
      filter_plugins/oo_filters.py

+ 4 - 1
filter_plugins/oo_filters.py

@@ -16,6 +16,7 @@ import pkg_resources
 import re
 import json
 import yaml
+from ansible.parsing.yaml.dumper import AnsibleDumper
 from ansible.utils.unicode import to_unicode
 from urlparse import urlparse
 
@@ -621,7 +622,9 @@ class FilterModule(object):
             return ""
 
         try:
-            transformed = yaml.safe_dump(data, indent=indent, allow_unicode=True, default_flow_style=False, **kw)
+            transformed = yaml.dump(data, indent=indent, allow_unicode=True,
+                                    default_flow_style=False,
+                                    Dumper=AnsibleDumper, **kw)
             padded = "\n".join([" " * level * indent + line for line in transformed.splitlines()])
             return to_unicode("\n{0}".format(padded))
         except Exception as my_e: