From eff941dbf8d022d380f11ea4efc44375bb198719 Mon Sep 17 00:00:00 2001 From: Justin Riley Date: Wed, 3 Aug 2016 00:13:16 -0400 Subject: [PATCH] utils: add get_kwargs_from_env function --- onedns/utils.py | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) 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={}):