Allow to update vars

This commit is contained in:
Luke Murphy 2020-04-16 12:25:50 +02:00
parent 7089caee97
commit 32ceaca084
No known key found for this signature in database
GPG Key ID: 5E2EF5A63E3718CC
1 changed files with 14 additions and 1 deletions

View File

@ -8,7 +8,10 @@ class FilterModule(object):
def filters(self):
"""Filters list."""
return {"update_env": self.filter_update_env}
return {
"update_env": self.filter_update_env,
"update_vars": self.filter_update_vars,
}
def filter_update_env(self, config, new_env):
"""Update the config.env dictionary with new keys."""
@ -19,3 +22,13 @@ class FilterModule(object):
config["env"][key] = val
return config
def filter_update_vars(self, config, new_vars):
"""Update the config.vars dictionary with new keys."""
if "vars" not in config:
return config
for key, val in new_vars.items():
config["vars"][key] = val
return config