blob: 60b8653879388c713083a1139624deb3c358b5bd (
plain)
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
|
import os
from dotenv import load_dotenv
def get_env_var(var_name: str):
load_dotenv()
return os.getenv(var_name)
def make_sure_env_exists():
if not os.path.exists(".env"):
with open(".env", "w") as f:
f.write("")
def save_env_var(var_name: str, var_value: str):
make_sure_env_exists()
with open(".env", "r") as f:
lines = f.readlines()
with open(".env", "w") as f:
values = {}
for line in lines:
key, value = line.split("=")
value = value.replace('"', "")
values[key] = value
values[var_name] = var_value
for key, value in values.items():
f.write(f'{key}="{value}"\n')
|