Added stocks.py for settings

data.py

@@ -1,14 +1,20 @@
 import database
+import stocks
 
 
+# Retrieve last and second to last timestamp from the server
 db = database.db
 cursor = db.cursor()
 query = """SELECT DISTINCT timestamp FROM stocks ORDER BY stocks.timestamp DESC LIMIT 1 OFFSET 1;"""
 cursor.execute(query)
-one_before_latest_entry_datetime = cursor.fetchone()
+one_before_latest_entry_timestamp = cursor.fetchone()[0]
 query = """SELECT DISTINCT timestamp FROM stocks ORDER BY stocks.timestamp DESC LIMIT 1;"""
 cursor.execute(query)
-latest_entry_datetime = cursor.fetchone(query)
+latest_entry_timestamp = cursor.fetchone()[0]
+
+print(one_before_latest_entry_timestamp)
+print(latest_entry_timestamp)
+
 
 cursor.close()
 db.close()

database.py

@@ -1,8 +1,12 @@
+from dotenv import load_dotenv
 import os
 import psycopg2
 
-
-database = os.environ.STOCK_DB_DATABASE
-user = os.environ.STOCK_DB_USER
-password = os.environ.STOCK_DB_PASSWD
-db = psycopg2.connect(database=database, user=user, password=password)
+# Develop only
+load_dotenv()
+host = os.getenv("STOCK_DB_HOST")
+port = os.getenv("STOCK_DB_PORT")
+database = os.getenv("STOCK_DB_DATABASE")
+user = os.getenv("STOCK_DB_USER")
+password = os.getenv("STOCK_DB_PASSWD")
+db = psycopg2.connect(host=host, port=port, database=database, user=user, password=password)

stocks.py

@@ -0,0 +1,3 @@
+# stock number, max price, min(price * drop) in B
+# So this means stock # 15 (FHC), min price $335  and a $10B drop
+fhc = (15, 335, 10)