-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathenv_utils.py
44 lines (32 loc) · 921 Bytes
/
env_utils.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import os
def cast_float(key, default=""):
item = os.getenv(key, default)
item = float(item)
assert(type(item) is float)
return item
def cast_int(key, default=""):
item = os.getenv(key, default)
item = int(item)
assert(type(item) is int)
return item
def cast_string(key, default="", require=False, strip=True, empty_to_none=False):
item = os.getenv(key, default)
if strip:
item.strip()
if require:
assert(len(item) > 0)
if len(item) == 0 and empty_to_none:
item = None
return item
def cast_list(key, default="", separator="|", strip=True):
base_item = os.getenv(key, default).split(separator)
item = []
for x in base_item:
if strip:
x = x.strip()
if len(x) == 0:
continue
item.append(x)
return item
def cast_bool(key, default="false"):
return os.getenv(key, default).lower() in ("true", "1", "yes")