db init params

binance_db/db.py

@@ -3,8 +3,11 @@
 from sqlalchemy.orm import sessionmaker
 
 class BinanceDB():
-    def __init__(self, connstr, echo=False):
-        self.engine = create_engine(connstr, echo=echo)
+    def __init__(self, password, user='binancedb', url='postgres', port=5432, db='binancedb', echo=False):
+        conn = 'postgresql://{}:{}@{}:{}/{}'.format(
+            user, password, url, port, db)
+
+        self.engine = create_engine(conn, echo=echo)
         Candle.metadata.create_all(self.engine)
 
     def get_session(self):

main.py

@@ -31,15 +31,15 @@ def connect_db():
     pgport = os.environ['BDB_POSTGRES_PORT']
     pguser = os.environ['BDB_POSTGRES_USER']
     pgpass = os.environ['BDB_POSTGRES_PASS']
-    conn = f'postgresql://{pguser}:{pgpass}@{pgurl}:{pgport}/{pguser}'
+    pgdb = os.environ['BDB_POSTGRES_DB']
 
     # let pg start up. first run can take longer than others
     logger.info("Waiting for Postgres...")
     pg_try = 0
     while True:
         time.sleep(5)
         try:
-            bdb = BinanceDB(conn)
+            bdb = BinanceDB(user=pguser, password=pgpass, url=pgurl, port=pgport, db=pgdb)
             break
         except:
             pg_try += 1