diff --git a/onedns/utils.py b/onedns/utils.py index b49ca00..e457430 100644 --- a/onedns/utils.py +++ b/onedns/utils.py @@ -1,10 +1,21 @@ +import os + from onedns.logger import log -def get_kwargs_from_dict(d, prefix): - kwargs = dict((i.replace(prefix, ''), d[i]) - for i in d.keys() if i.startswith(prefix)) - return kwargs +def get_kwargs_from_dict(d, prefix, lower=False): + tups_list = [] + for i in d: + if i.startswith(prefix): + arg = i.replace(prefix, '') + if lower: + arg = arg.lower() + tups_list.append((arg, d[i])) + return dict(tups_list) + + +def get_kwargs_from_env(prefix, lower=False): + return get_kwargs_from_dict(os.environ, prefix, lower=lower) def shell(local_ns={}):