Refactor seed frame initialization functions to handle multiple slate types for DraftKings and FanDuel, improving data retrieval and organization based on slate selection.

src/sim_func_hold/regular_functions.py

@@ -7,11 +7,26 @@ from database import db
 @st.cache_data(ttl = 600)
 def init_DK_seed_frames(slate_var, sharp_split):
 
-    collection = db['DK_NFL_name_map']
-    cursor = collection.find()
-    raw_data = pd.DataFrame(list(cursor))
-    names_dict = dict(zip(raw_data['key'], raw_data['value']))
-    collection = db[f"DK_NFL_{slate_var}_seed_frame"] 
+    
+    if slate_var == 'Main Slate':
+        collection = db['DK_NFL_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"DK_NFL_seed_frame"] 
+    elif slate_var == 'Secondary Slate':
+        collection = db['DK_NFL_Secondary_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"DK_NFL_Secondary_seed_frame"] 
+    elif slate_var == 'Auxiliary Slate':
+        collection = db['DK_NFL_Auxiliary_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"DK_NFL_Auxiliary_seed_frame"] 
+
     cursor = collection.find().limit(sharp_split)
 
     raw_display = pd.DataFrame(list(cursor))
@@ -26,13 +41,27 @@ def init_DK_seed_frames(slate_var, sharp_split):
 @st.cache_data(ttl = 599)
 def init_FD_seed_frames(slate_var, sharp_split): 
 
-    collection = db['FD_NFL_name_map']
-    cursor = collection.find()
-    raw_data = pd.DataFrame(list(cursor))
-    names_dict = dict(zip(raw_data['key'], raw_data['value']))
-    
-    collection = db[f"FD_NFL_{slate_var}_seed_frame"] 
-    cursor = collection.find().limit(sharp_split)
+    if slate_var == 'Main Slate':
+        collection = db['FD_NFL_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"FD_NFL_seed_frame"] 
+        cursor = collection.find().limit(sharp_split)
+    elif slate_var == 'Secondary Slate':
+        collection = db['FD_NFL_Secondary_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"FD_NFL_Secondary_seed_frame"] 
+        cursor = collection.find().limit(sharp_split)
+    elif slate_var == 'Auxiliary Slate':
+        collection = db['FD_NFL_Auxiliary_name_map']
+        cursor = collection.find()
+        raw_data = pd.DataFrame(list(cursor))
+        names_dict = dict(zip(raw_data['key'], raw_data['value']))
+        collection = db[f"FD_NFL_Auxiliary_seed_frame"] 
+        cursor = collection.find().limit(sharp_split)
 
     raw_display = pd.DataFrame(list(cursor))
     raw_display = raw_display[['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX', 'DST', 'salary', 'proj', 'Team', 'Team_count', 'Secondary', 'Secondary_count', 'Own']]