From cbde083ad644fb44808500d5ae59fbb0562c1e28 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Thu, 21 Dec 2023 20:24:26 +0100
Subject: [PATCH] Make sure that secrets would work on aws

---
 .../database/postgres/database.py             | 31 ++++++++++++++-----
 1 file changed, 24 insertions(+), 7 deletions(-)

diff --git a/cognitive_architecture/database/postgres/database.py b/cognitive_architecture/database/postgres/database.py
index 6b14553f3..592a025f0 100644
--- a/cognitive_architecture/database/postgres/database.py
+++ b/cognitive_architecture/database/postgres/database.py
@@ -15,21 +15,38 @@ load_dotenv()
 # # Get the parent directory of your script and add it to sys.path
 # parent_dir = os.path.dirname(script_dir)
 # sys.path.append(parent_dir)
-from ...config import Config
-config = Config()
-config.load()
+# from ...config import Config
+# config = Config()
+# config.load()
 
 
 # in seconds
 MAX_RETRIES = 3
 RETRY_DELAY = 5
 
+import os
 
 
-host = config.postgres_host
-username = config.postgres_user
-password = config.postgres_password
-database_name = config.postgres_db
+if os.environ.get('AWS_ENV') == 'prd' or os.environ.get('AWS_ENV') == 'dev':
+    host = os.environ.get('POSTGRES_PROD_HOST')
+    username = os.environ.get('POSTGRES_USER')
+    password = os.environ.get('POSTGRES_PASSWORD')
+    database_name = os.environ.get('POSTGRES_DB')
+elif os.environ.get('AWS_ENV') == 'local':
+    host = os.environ.get('POSTGRES_HOST')
+    username = os.environ.get('POSTGRES_USER')
+    password = os.environ.get('POSTGRES_PASSWORD')
+    database_name = os.environ.get('POSTGRES_DB')
+else:
+    host = os.environ.get('POSTGRES_HOST')
+    username = os.environ.get('POSTGRES_USER')
+    password = os.environ.get('POSTGRES_PASSWORD')
+    database_name = os.environ.get('POSTGRES_DB')
+
+# host = config.postgres_host
+# username = config.postgres_user
+# password = config.postgres_password
+# database_name = config.postgres_db