Skip to content

Commit eb1c135

Browse files
committed
for fetching of environment variable
1 parent 2a67120 commit eb1c135

1 file changed

Lines changed: 28 additions & 0 deletions

File tree

src/utils/shared/fetch_env.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
from os import getenv
2+
3+
from rich.console import Console
4+
5+
from src.utils.misc.uinput import uinput
6+
from src.utils.log.logger import Logger
7+
8+
9+
def fetch_env(log: Logger, console: Console, env_var: str) -> str:
10+
"""Fetch the value of given env variable.
11+
12+
Args:
13+
log -- instance of Logger
14+
env_var -- variable to fetch
15+
16+
Returns:
17+
The value of env variable or None
18+
"""
19+
20+
try:
21+
env_value: str = getenv(env_var.upper())
22+
except OSError as Err:
23+
log.logger("e", f"{Err}. No value found for {env_var}.")
24+
env_value = uinput(
25+
console, f"Kindly input the value for {env_var}"
26+
)
27+
28+
return env_value

0 commit comments

Comments
 (0)