Refactor stack column dictionary in 'app.py' to support multiple sites, enhancing flexibility in player stacking calculations for Draftkings and Fanduel.

app.py

@@ -31,11 +31,20 @@ from global_func.reassess_edge import reassess_edge
 freq_format = {'Finish_percentile': '{:.2%}', 'Lineup Edge': '{:.2%}', 'Win%': '{:.2%}'}
 stacking_sports = ['MLB', 'NHL', 'NFL', 'LOL', 'NCAAF']
 stack_column_dict = {
-    'MLB': ['C', 'C/1B', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3'],
-    'NHL': ['C', 'W', 'D'],
-    'NFL': ['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX'],
-    'LOL': ['TOP', 'JNG', 'MID', 'ADC', 'SUP', 'TEAM'],
-    'NCAAF': ['QB', 'WR1', 'WR2', 'WR3', 'FLEX', 'SFLEX'],
+    'Draftkings': {
+        'MLB': ['C', '1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3'],
+        'NHL': ['C', 'W', 'D'],
+        'NFL': ['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX'],
+        'LOL': ['TOP', 'JNG', 'MID', 'ADC', 'SUP', 'TEAM'],
+        'NCAAF': ['QB', 'WR1', 'WR2', 'WR3', 'FLEX', 'SFLEX'],
+    },
+    'Fanduel': {
+        'MLB': ['C/1B', '2B', '3B', 'SS', 'OF1', 'OF2', 'OF3', 'UTIL'],
+        'NHL': ['C', 'W', 'D'],
+        'NFL': ['QB', 'RB1', 'RB2', 'WR1', 'WR2', 'WR3', 'TE', 'FLEX'],
+        'LOL': ['TOP', 'JNG', 'MID', 'ADC', 'SUP', 'TEAM'],
+        'NCAAF': ['QB', 'WR1', 'WR2', 'WR3', 'FLEX', 'SFLEX'],
+    },
 }
 player_wrong_names_mlb = ['Enrique Hernandez', 'Joseph Cantillo', 'Mike Soroka', 'Jakob Bauers', 'Temi Fágbénlé']
 player_right_names_mlb = ['Kike Hernandez', 'Joey Cantillo', 'Michael Soroka', 'Jake Bauers', 'Temi Fagbenle']
@@ -452,16 +461,16 @@ if selected_tab == 'Data Load':
                 team_dict = dict(zip(st.session_state['projections_df']['player_names'], st.session_state['projections_df']['team']))
                 st.session_state['portfolio']['Stack'] = st.session_state['portfolio'].apply(
                     lambda row: Counter(
-                        team_dict.get(player, '') for player in row[stack_column_dict[sport_var]]
+                        team_dict.get(player, '') for player in row[stack_column_dict[site_var][sport_var]]
                         if team_dict.get(player, '') != ''
-                    ).most_common(1)[0][0] if any(team_dict.get(player, '') for player in row[stack_column_dict[sport_var]]) else '',
+                    ).most_common(1)[0][0] if any(team_dict.get(player, '') for player in row[stack_column_dict[site_var][sport_var]]) else '',
                     axis=1
                 )
                 st.session_state['portfolio']['Size'] = st.session_state['portfolio'].apply(
                     lambda row: Counter(
-                        team_dict.get(player, '') for player in row[stack_column_dict[sport_var]]
+                        team_dict.get(player, '') for player in row[stack_column_dict[site_var][sport_var]]
                         if team_dict.get(player, '') != ''
-                    ).most_common(1)[0][1] if any(team_dict.get(player, '') for player in row[stack_column_dict[sport_var]]) else 0,
+                    ).most_common(1)[0][1] if any(team_dict.get(player, '') for player in row[stack_column_dict[site_var][sport_var]]) else 0,
                     axis=1
                 )
                 st.session_state['stack_dict'] = dict(zip(st.session_state['portfolio'].index, st.session_state['portfolio']['Stack']))