Current File : //proc/227033/root/lib/python2.7/site-packages/salt/utils/yamlencoding.py
# -*- coding: utf-8 -*-
'''
Functions for adding yaml encoding to the jinja context
'''
# Import Python libs
from __future__ import absolute_import, print_function, unicode_literals
import io
# Import 3rd-party libs
import yaml # pylint: disable=blacklisted-import
from salt.ext import six
# Import salt libs
from salt.utils.decorators.jinja import jinja_filter
@jinja_filter()
def yaml_dquote(text):
'''
Make text into a double-quoted YAML string with correct escaping
for special characters. Includes the opening and closing double
quote characters.
'''
with io.StringIO() as ostream:
yemitter = yaml.emitter.Emitter(ostream, width=six.MAXSIZE)
yemitter.write_double_quoted(six.text_type(text))
return ostream.getvalue()
@jinja_filter()
def yaml_squote(text):
'''
Make text into a single-quoted YAML string with correct escaping
for special characters. Includes the opening and closing single
quote characters.
'''
with io.StringIO() as ostream:
yemitter = yaml.emitter.Emitter(ostream, width=six.MAXSIZE)
yemitter.write_single_quoted(six.text_type(text))
return ostream.getvalue()
@jinja_filter()
def yaml_encode(data):
'''
A simple YAML encode that can take a single-element datatype and return
a string representation.
'''
yrepr = yaml.representer.SafeRepresenter()
ynode = yrepr.represent_data(data)
if not isinstance(ynode, yaml.ScalarNode):
raise TypeError(
"yaml_encode() only works with YAML scalar data;"
" failed for {0}".format(type(data))
)
tag = ynode.tag.rsplit(':', 1)[-1]
ret = ynode.value
if tag == "str":
ret = yaml_dquote(ynode.value)
return ret
Mini Shell