From 213df75a16726389ddec5f8847f26e3d92cf7e31 Mon Sep 17 00:00:00 2001
From: Bhargav <bhargavkuvadiya117@gmail.com>
Date: Thu, 5 Oct 2023 02:05:00 +0530
Subject: [PATCH] feat: add dotenv (#8)

Closes #4
---
 codeqai/app.py | 37 +++++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)

diff --git a/codeqai/app.py b/codeqai/app.py
index 6daf3fd..df367aa 100644
--- a/codeqai/app.py
+++ b/codeqai/app.py
@@ -15,6 +15,43 @@
 from codeqai.llm import LLM
 from codeqai.vector_store import VectorStore
 
+from dotenv import dotenv_values
+
+def env_loader(env_path, required_keys=None):
+    """
+        Args : 
+        env_path = source path of .env file.
+        required_keys = ["OPENAI_KEY"] #change this according to need
+
+        #running/calling the function.
+        configs = env_loader('.env', required_keys)
+    """
+    
+    #create env file if does not exists
+    #parse required keys in the file if it's not None
+    if not os.path.exists(env_path):
+        with open(env_path, 'w') as env_f:
+            if required_keys:
+                for key in required_keys:
+                    env_f.write(f'{key}=""\n')
+            else:
+                pass
+    
+    configs = dotenv_values(env_path)
+    changed = False
+    for key, value in configs.items():
+        if not value:
+            value = input(f'[+] Key {key} is required enter it\'s value :: > ')
+            configs[key] = value
+            changed = True
+
+    #update the .env file if config is changed/taken from user
+    if changed:
+        with open(env_path, 'w') as env_f:
+            for key, value in configs.items():
+                env_f.write(f'{key}="{value}"\n')
+
+    return configs
 
 def run():
     parser = argparse.ArgumentParser()