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

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

diff --git a/cognitive_architecture/database/create_database.py b/cognitive_architecture/database/create_database.py
index ac9c3a852..ad8a43fed 100644
--- a/cognitive_architecture/database/create_database.py
+++ b/cognitive_architecture/database/create_database.py
@@ -9,9 +9,9 @@ from postgres.models import operation
 from postgres.models import sessions
 from postgres.models import user
 from postgres.models import docs
-from cognitive_architecture.config import Config
-config = Config()
-config.load()
+# from cognitive_architecture.config import Config
+# config = Config()
+# config.load()
 
 
 
@@ -24,8 +24,25 @@ load_dotenv()
 import os
 
 
+import os
 
 
+if os.environ.get('AWS_ENV') == 'prd' or os.environ.get('AWS_ENV') == 'dev':
+    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')
+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')
+
 
 
 def create_admin_engine(username, password, host, database_name):
@@ -63,10 +80,10 @@ if __name__ == "__main__":
     # password = os.getenv('POSTGRES_PASSWORD')
     # database_name = os.getenv('POSTGRES_DB')
     # environment = os.environ.get("ENV")
-    host  = config.postgres_host
-    username = config.postgres_user
-    password = config.postgres_password
-    database_name = config.postgres_db
+    # host  = config.postgres_host
+    # username = config.postgres_user
+    # password = config.postgres_password
+    # database_name = config.postgres_db
 
     engine = create_admin_engine(username, password, host, database_name)