Refactor database connection to use environment variables for credentials

views/workouts_form.py

@@ -3,12 +3,21 @@ import mysql.connector
 from mysql.connector import errorcode
 import pandas as pd
 from datetime import datetime, date
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+user = os.getenv('user')
+password = os.getenv('password')
+host = os.getenv('host')
+database = os.getenv('database')
+port = os.getenv('port')
 
 workout_table = ['chest','back','legs']
 selected_workout = st.selectbox('Select a Workout',options=workout_table)
 
 try:
-    conn = mysql.connector.connect(user='root',password='Throwaway', host='localhost',port=3306,database='workouts')
+    conn = mysql.connector.connect(user=user,password=password, host=host,port=port,database=database)
 except mysql.connector.Error as err:
     if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
         print("Something is wrong with your user name or password")

views/workouts_report.py

@@ -2,9 +2,18 @@ import streamlit as st
 import mysql.connector
 from mysql.connector import errorcode
 import pandas as pd
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+user = os.getenv('user')
+password = os.getenv('password')
+host = os.getenv('host')
+database = os.getenv('database')
+port = os.getenv('port')
 
 try:
-    conn = mysql.connector.connect(user='root',password='Throwaway', host='localhost',port=3306,database='workouts')
+    conn = mysql.connector.connect(user=user,password=password, host=host,port=port,database=database)
 except mysql.connector.Error as err:
     if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
         print("Something is wrong with your user name or password")

views/workouts_templates.py

@@ -3,13 +3,22 @@ import mysql.connector
 from mysql.connector import errorcode
 import pandas as pd
 from datetime import datetime, date
+from dotenv import load_dotenv
+import os
 
+load_dotenv()
+user = os.getenv('user')
+password = os.getenv('password')
+host = os.getenv('host')
+database = os.getenv('database')
+port = os.getenv('port')
 workout_table = ['chest','back','legs']
+
 selected_workout = st.selectbox('Select a Workout',options=workout_table)
 edited_df=[]
 
 try:
-    conn = mysql.connector.connect(user='root',password='Throwaway', host='localhost',port=3306,database='workouts')
+    conn = mysql.connector.connect(user=user,password=password, host=host,port=port,database=database)
 except mysql.connector.Error as err:
     if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
         print("Something is wrong with your user name or password")
@@ -41,7 +50,7 @@ edited_df = st.data_editor(df)
 # Button to save all changes back to the dataframe
 if st.button('Save All Changes'):
     st.write("All changes saved!")
-    conn = mysql.connector.connect(user='root',password='Throwaway', host='localhost',port=3306,database='workouts')
+    conn = mysql.connector.connect(user=user,password=password, host=host,port=port,database=database)
     cursor = conn.cursor()
     query = (f"DELETE FROM {working_table}")
     cursor.execute(query)