Refactored db info to the .env file

pilot/const/db.py

@@ -1,5 +0,0 @@
-DB_NAME = 'pilot'
-DB_HOST = 'localhost'
-DB_PORT = '5432'
-DB_USER = 'admin'
-DB_PASSWORD = 'admin'

pilot/database/models/components/base_models.py

@@ -1,17 +1,22 @@
+import os
 from peewee import *
 from datetime import datetime
 from uuid import uuid4
 
-from const import db
+DB_NAME = os.getenv("DB_NAME")
+DB_HOST = os.getenv("DB_HOST")
+DB_PORT = os.getenv("DB_PORT")
+DB_USER = os.getenv("DB_USER")
+DB_PASSWORD = os.getenv("DB_PASSWORD")
 
 
 # Establish connection to the database
 database = PostgresqlDatabase(
-    db.DB_NAME,
-    user=db.DB_USER,
-    password=db.DB_PASSWORD,
-    host=db.DB_HOST,
-    port=db.DB_PORT
+    DB_NAME,
+    user=DB_USER,
+    password=DB_PASSWORD,
+    host=DB_HOST,
+    port=DB_PORT
 )
 
 

pilot/main.py

@@ -2,6 +2,8 @@
 from __future__ import print_function, unicode_literals
 
 from dotenv import load_dotenv
+load_dotenv()
+
 from helpers.Project import Project
 
 from utils.arguments import get_arguments
@@ -9,8 +11,6 @@ from logger.logger import logger
 
 
 def init():
-    load_dotenv()
-
     arguments = get_arguments()
 
     logger.info(f"Starting with args: {arguments}")