Sha256: d9d12ab4d64372b01c1f5799bf569a671241b4b2f9c5db316c9f944b6228dafc
Contents?: true
Size: 1.67 KB
Versions: 56
Compression:
Stored size: 1.67 KB
Contents
require 'yaml'
module Puppet::Util::Yaml
YamlLoadExceptions = [::StandardError, ::Psych::Exception]
class YamlLoadError < Puppet::Error; end
# Safely load the content as YAML. By default only the following
# classes can be deserialized:
#
# * TrueClass
# * FalseClass
# * NilClass
# * Numeric
# * String
# * Array
# * Hash
#
# Attempting to deserialize other classes will raise an YamlLoadError
# exception unless they are specified in the array of *allowed_classes*.
# @param [String] yaml The yaml content to parse.
# @param [Array] allowed_classes Additional list of classes that can be deserialized.
# @param [String] filename The filename to load from, used if an exception is raised.
# @raise [YamlLoadException] If deserialization fails.
# @return The parsed YAML, which can be Hash, Array or scalar types.
def self.safe_load(yaml, allowed_classes = [], filename = nil)
data = YAML.safe_load(yaml, allowed_classes, [], true, filename)
data = false if data.nil?
data
rescue ::Psych::DisallowedClass => detail
path = filename ? "(#{filename})" : "(<unknown>)"
raise YamlLoadError.new("#{path}: #{detail.message}", detail)
rescue *YamlLoadExceptions => detail
raise YamlLoadError.new(detail.message, detail)
end
# Safely load the content from a file as YAML.
#
# @see Puppet::Util::Yaml.safe_load
def self.safe_load_file(filename, allowed_classes = [])
yaml = Puppet::FileSystem.read(filename, :encoding => 'bom|utf-8')
safe_load(yaml, allowed_classes, filename)
end
def self.dump(structure, filename)
Puppet::FileSystem.replace_file(filename, 0660) do |fh|
YAML.dump(structure, fh)
end
end
end
Version data entries
56 entries across 56 versions & 1 rubygems