From b616ca80ca61142a6e9f27941220be26f7bc6ad1 Mon Sep 17 00:00:00 2001 From: Cutty Date: Sun, 22 Feb 2026 22:51:15 -0700 Subject: [PATCH] Refactor database connection to use environment variables for credentials --- requirements.txt | Bin 84 -> 100 bytes views/workouts_form.py | 11 ++++++++++- views/workouts_report.py | 11 ++++++++++- views/workouts_templates.py | 13 +++++++++++-- 4 files changed, 31 insertions(+), 4 deletions(-) diff --git a/requirements.txt b/requirements.txt index 9f5edb3f430c9232e58d1c2c33c42c9088d861ce..5fcb09b708f422351069b4488217633d2fe5bb0e 100644 GIT binary patch delta 21 ccmWGZnGhnt%fQ8u!jR8U!jQ_4$56%q05f$2Q~&?~ delta 4 LcmYcZnGga11Ox${ diff --git a/views/workouts_form.py b/views/workouts_form.py index 2d4afb3..f410365 100644 --- a/views/workouts_form.py +++ b/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") diff --git a/views/workouts_report.py b/views/workouts_report.py index 45eaa94..782d2d2 100644 --- a/views/workouts_report.py +++ b/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") diff --git a/views/workouts_templates.py b/views/workouts_templates.py index 3093d5c..b0ef607 100644 --- a/views/workouts_templates.py +++ b/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)