utils: add get_kwargs_from_env function

master
Justin Riley 2016-08-03 00:13:16 -04:00
parent f312c86df6
commit eff941dbf8
1 changed files with 15 additions and 4 deletions

View File

@ -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={}):