diff options
-rw-r--r-- | load_vars.rb | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/load_vars.rb b/load_vars.rb index 87effad..02e99a0 100644 --- a/load_vars.rb +++ b/load_vars.rb @@ -40,11 +40,19 @@ # module Puppet::Parser::Functions - newfunction(:load_vars, :type => :statement) do |args| + newfunction(:load_vars, :type => :statement) do |arguments| + + number_of_arguments = arguments.size + + if number_of_arguments < 2 + raise(Puppet::ParseError, "Wrong number of arguments " + + "given (#{number_of_arguments} for 2)") + end + data = {} - file = args[0] - key = args[1] if args[1] + file = arguments[0] + key = arguments[1] if arguments[1] if File.exists?(file) |