Refactor initialization functions in Streamlit app to remove database argument, streamlining data loading for DraftKings and FanDuel simulations.

src/sim_func_hold/regular_functions.py

@@ -3,8 +3,10 @@ import pandas as pd
 import numpy as np
 from pymongo import MongoClient
 
+from streamlit_app import db
+
 @st.cache_data(ttl = 600)
-def init_DK_seed_frames(slate_var, sharp_split, db):
+def init_DK_seed_frames(slate_var, sharp_split):
 
     collection = db['DK_NFL_name_map']
     cursor = collection.find()
@@ -23,7 +25,7 @@ def init_DK_seed_frames(slate_var, sharp_split, db):
     return DK_seed
 
 @st.cache_data(ttl = 599)
-def init_FD_seed_frames(slate_var, sharp_split, db): 
+def init_FD_seed_frames(slate_var, sharp_split): 
 
     collection = db['FD_NFL_name_map']
     cursor = collection.find()
@@ -43,7 +45,7 @@ def init_FD_seed_frames(slate_var, sharp_split, db):
     return FD_seed
 
 @st.cache_data(ttl = 599)
-def init_baselines(slate_var, db):
+def init_baselines(slate_var):
     collection = db["DK_NFL_ROO"] 
     cursor = collection.find()
 

src/streamlit_app.py

@@ -69,12 +69,12 @@ if st.button("Load/Reset Data", key='reset2'):
     st.cache_data.clear()
     for key in st.session_state.keys():
         del st.session_state[key]
-    DK_seed = init_DK_seed_frames('Main Slate', 10000, db)
-    DK_sd_seed = init_DK_SD_seed_frames('Main Slate', 10000, db)
-    FD_seed = init_FD_seed_frames('Main Slate', 10000, db)
-    FD_sd_seed = init_FD_SD_seed_frames('Main Slate', 10000, db)
-    dk_raw, fd_raw = init_baselines('Main Slate', db)
-    dk_sd_raw, fd_sd_raw = init_SD_baselines('Main Slate', db)
+    DK_seed = init_DK_seed_frames('Main Slate', 10000)
+    DK_sd_seed = init_DK_SD_seed_frames('Main Slate', 10000)
+    FD_seed = init_FD_seed_frames('Main Slate', 10000)
+    FD_sd_seed = init_FD_SD_seed_frames('Main Slate', 10000)
+    dk_raw, fd_raw = init_baselines('Main Slate')
+    dk_sd_raw, fd_sd_raw = init_SD_baselines('Main Slate')
     dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_ID))
     dk_sd_id_dict = dict(zip(dk_sd_raw.Player, dk_sd_raw.player_ID))
     fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_ID))
@@ -91,7 +91,7 @@ selected_tab = st.segmented_control(
 )
             
 if selected_tab == "Regular Slate Contest Sims":
-    dk_raw, fd_raw = init_baselines('Main Slate', db)
+    dk_raw, fd_raw = init_baselines('Main Slate')
     raw_baselines = dk_raw
     column_names = dk_columns
     with st.expander("Info and Filters"):
@@ -129,15 +129,15 @@ if selected_tab == "Regular Slate Contest Sims":
             
         if 'working_seed' not in st.session_state:
             if sim_site_var1 == 'Draftkings':
-                st.session_state.working_seed = init_DK_seed_frames(sim_slate_var1, sharp_split, db)
-                dk_raw, fd_raw = init_baselines(sim_slate_var1, db)
+                st.session_state.working_seed = init_DK_seed_frames(sim_slate_var1, sharp_split)
+                dk_raw, fd_raw = init_baselines(sim_slate_var1)
                 dk_id_dict = dict(zip(dk_raw.Player, dk_raw.player_ID)) 
                     
                 raw_baselines = dk_raw
                 column_names = dk_columns
             elif sim_site_var1 == 'Fanduel':
-                st.session_state.working_seed = init_FD_seed_frames(sim_slate_var1, sharp_split, db)
-                dk_raw, fd_raw = init_baselines(sim_slate_var1, db)
+                st.session_state.working_seed = init_FD_seed_frames(sim_slate_var1, sharp_split)
+                dk_raw, fd_raw = init_baselines(sim_slate_var1)
                 fd_id_dict = dict(zip(fd_raw.Player, fd_raw.player_ID))
                     
                 raw_baselines = fd_raw
@@ -556,7 +556,7 @@ if selected_tab == "Regular Slate Contest Sims":
                 )
 
 if selected_tab == "Showdown Contest Sims":
-    dk_raw, fd_raw = init_SD_baselines('Main Slate', db)
+    dk_raw, fd_raw = init_SD_baselines('Main Slate')
     raw_baselines = dk_raw
     column_names = dk_columns
     with st.expander("Info and Filters"):
@@ -594,12 +594,12 @@ if selected_tab == "Showdown Contest Sims":
             
         if 'sd_working_seed' not in st.session_state:
             if sim_site_var2 == 'Draftkings':
-                st.session_state.sd_working_seed = init_DK_SD_seed_frames(sim_slate_var2, sharp_split, db)
+                st.session_state.sd_working_seed = init_DK_SD_seed_frames(sim_slate_var2, sharp_split)
                 export_id_dict = dict(zip(dk_raw.Player, dk_raw.player_ID)) 
                 raw_baselines = dk_raw
                 column_names = dk_columns
             elif sim_site_var2 == 'Fanduel':
-                st.session_state.sd_working_seed = init_FD_SD_seed_frames(sim_slate_var2, sharp_split, db)
+                st.session_state.sd_working_seed = init_FD_SD_seed_frames(sim_slate_var2, sharp_split)
                 export_id_dict = dict(zip(fd_raw.Player, fd_raw.player_ID))
                 raw_baselines = fd_raw
                 column_names = fd_columns