mirror of
https://github.com/kennethreitz/env.git
synced 2026-06-05 23:10:16 +00:00
env is written
This commit is contained in:
@@ -0,0 +1,52 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
from os import environ
|
||||
|
||||
def lower_dict(d):
|
||||
"""Lower cases string keys in given dict."""
|
||||
|
||||
_d = {}
|
||||
|
||||
for k, v in d.iteritems():
|
||||
try:
|
||||
_d[k.lower()] = v
|
||||
except AttributeError:
|
||||
_d[k] = v
|
||||
|
||||
return _d
|
||||
|
||||
|
||||
def prefix(prefix):
|
||||
"""Returns a dictionary of all environment variables starting with
|
||||
the given prefix, lower cased and stripped.
|
||||
"""
|
||||
|
||||
d = {}
|
||||
e = lower_dict(environ.copy())
|
||||
|
||||
prefix = prefix.lower()
|
||||
|
||||
for k, v in e.iteritems():
|
||||
try:
|
||||
if k.startswith(prefix):
|
||||
k = k[len(prefix):]
|
||||
d[k] = v
|
||||
except AttributeError:
|
||||
print k
|
||||
pass
|
||||
|
||||
return d
|
||||
|
||||
|
||||
def map(**kwargs):
|
||||
"""Returns a dictionary of the given keyword arguments mapped to their
|
||||
values from the environment.
|
||||
"""
|
||||
|
||||
d = {}
|
||||
e = lower_dict(environ.copy())
|
||||
|
||||
for k, v in kwargs.iteritems():
|
||||
d[k] = e.get(v)
|
||||
|
||||
return d
|
||||
Reference in New Issue
Block a user