summaryrefslogtreecommitdiff
path: root/server/continuedev/core/env.py
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')