Adding NHL database query support

app.py

@@ -45,11 +45,19 @@ dk_db_nba_showdown_selections = ['DK_NBA_SD_seed_frame_Showdown #1', 'DK_NBA_SD_
 fd_db_nba_showdown_selections = ['FD_NBA_SD_seed_frame_Showdown #1', 'FD_NBA_SD_seed_frame_Showdown #2', 'FD_NBA_SD_seed_frame_Showdown #3', 'FD_NBA_SD_seed_frame_Showdown #4', 'FD_NBA_SD_seed_frame_Showdown #5', 'FD_NBA_SD_seed_frame_Showdown #6',
 'FD_NBA_SD_seed_frame_Showdown #7', 'FD_NBA_SD_seed_frame_Showdown #8', 'FD_NBA_SD_seed_frame_Showdown #9', 'FD_NBA_SD_seed_frame_Showdown #10', 'FD_NBA_SD_seed_frame_Showdown #11', 'FD_NBA_SD_seed_frame_Showdown #12', 'FD_NBA_SD_seed_frame_Showdown #13',
 'FD_NBA_SD_seed_frame_Showdown #14', 'FD_NBA_SD_seed_frame_Showdown #15']
+dk_db_nhl_showdown_selections = ['DK_NHL_SD_seed_frame_Showdown #1', 'DK_NHL_SD_seed_frame_Showdown #2', 'DK_NHL_SD_seed_frame_Showdown #3', 'DK_NHL_SD_seed_frame_Showdown #4', 'DK_NHL_SD_seed_frame_Showdown #5', 'DK_NHL_SD_seed_frame_Showdown #6',
+'DK_NHL_SD_seed_frame_Showdown #7', 'DK_NHL_SD_seed_frame_Showdown #8', 'DK_NHL_SD_seed_frame_Showdown #9', 'DK_NHL_SD_seed_frame_Showdown #10', 'DK_NHL_SD_seed_frame_Showdown #11', 'DK_NHL_SD_seed_frame_Showdown #12', 'DK_NHL_SD_seed_frame_Showdown #13',
+'DK_NHL_SD_seed_frame_Showdown #14', 'DK_NHL_SD_seed_frame_Showdown #15']
+fd_db_nhl_showdown_selections = ['FD_NHL_SD_seed_frame_Showdown #1', 'FD_NHL_SD_seed_frame_Showdown #2', 'FD_NHL_SD_seed_frame_Showdown #3', 'FD_NHL_SD_seed_frame_Showdown #4', 'FD_NHL_SD_seed_frame_Showdown #5', 'FD_NHL_SD_seed_frame_Showdown #6',
+'FD_NHL_SD_seed_frame_Showdown #7', 'FD_NHL_SD_seed_frame_Showdown #8', 'FD_NHL_SD_seed_frame_Showdown #9', 'FD_NHL_SD_seed_frame_Showdown #10', 'FD_NHL_SD_seed_frame_Showdown #11', 'FD_NHL_SD_seed_frame_Showdown #12', 'FD_NHL_SD_seed_frame_Showdown #13',
+'FD_NHL_SD_seed_frame_Showdown #14', 'FD_NHL_SD_seed_frame_Showdown #15']
 
 dk_nfl_showdown_db_translation = dict(zip(showdown_selections, dk_db_nfl_showdown_selections))
 fd_nfl_showdown_db_translation = dict(zip(showdown_selections, fd_db_nfl_showdown_selections))
 dk_nba_showdown_db_translation = dict(zip(showdown_selections, dk_db_nba_showdown_selections))
 fd_nba_showdown_db_translation = dict(zip(showdown_selections, fd_db_nba_showdown_selections))
+dk_nhl_showdown_db_translation = dict(zip(showdown_selections, dk_db_nhl_showdown_selections))
+fd_nhl_showdown_db_translation = dict(zip(showdown_selections, fd_db_nhl_showdown_selections))
 
 freq_format = {'Finish_percentile': '{:.2%}', 'Lineup Edge': '{:.2%}', 'Lineup Edge_Raw': '{:.2%}', 'Win%': '{:.2%}'}
 stacking_sports = ['MLB', 'NHL', 'NFL', 'LOL', 'NCAAF']
@@ -393,6 +401,52 @@ except:
     nba_slate_names_fd = []
     nba_slate_name_lookup_fd = {}
 
+def define_dk_nhl_showdown_slates():
+    collection = nhl_db["Player_Level_SD_ROO"]
+    cursor = collection.find()
+    raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[raw_display['Site'] == 'Draftkings']
+    unique_slates = raw_display['Slate'].unique()
+
+    slate_names = []
+
+    for slate in unique_slates:
+        slate_data = raw_display[raw_display['Slate'] == slate]
+        slate_name = slate_data.iloc[0]['Team'] + ' vs. ' + slate_data.iloc[0]['Opp']
+        slate_names.append(slate_name)
+    
+    slate_name_lookup = dict(zip(slate_names, unique_slates))
+    return slate_names, slate_name_lookup
+
+def define_fd_nhl_showdown_slates():
+    collection = nhl_db["Player_Level_SD_ROO"]
+    cursor = collection.find()
+    raw_display = pd.DataFrame(list(cursor))
+    raw_display = raw_display[raw_display['Site'] == 'Fanduel']
+    unique_slates = raw_display['Slate'].unique()
+
+    slate_names = []
+
+    for slate in unique_slates:
+        slate_data = raw_display[raw_display['Slate'] == slate]
+        slate_name = slate_data.iloc[0]['Team'] + ' vs. ' + slate_data.iloc[0]['Opp']
+        slate_names.append(slate_name)
+    
+    slate_name_lookup = dict(zip(slate_names, unique_slates))
+    return slate_names, slate_name_lookup
+
+try:
+    nhl_slate_names_dk, nhl_slate_name_lookup_dk = define_dk_nhl_showdown_slates()
+except:
+    nhl_slate_names_dk = []
+    nhl_slate_name_lookup_dk = {}
+    
+try:
+    nhl_slate_names_fd, nhl_slate_name_lookup_fd = define_fd_nhl_showdown_slates()
+except:
+    nhl_slate_names_fd = []
+    nhl_slate_name_lookup_fd = {}
+
 # Memory optimization helper functions
 def chunk_name_matching(portfolio_names, csv_names, chunk_size=1000):
     """Process name matching in chunks to reduce memory usage"""
@@ -770,6 +824,8 @@ with st.container():
                     slate_var3 = st.radio("Which slate data are you loading?", (nba_slate_names_dk if type_var == 'Showdown' else ['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
                 elif sport_var == 'NFL':
                     slate_var3 = st.radio("Which slate data are you loading?", (nfl_slate_names_dk if type_var == 'Showdown' else ['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
+                elif sport_var == 'NHL':
+                    slate_var3 = st.radio("Which slate data are you loading?", (nhl_slate_names_dk if type_var == 'Showdown' else ['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
                 else:
                     slate_var3 = st.radio("Which slate data are you loading?", (['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
             elif site_var == 'Fanduel':
@@ -777,6 +833,8 @@ with st.container():
                     slate_var3 = st.radio("Which slate data are you loading?", (nba_slate_names_fd if type_var == 'Showdown' else ['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
                 elif sport_var == 'NFL':
                     slate_var3 = st.radio("Which slate data are you loading?", (nfl_slate_names_fd if type_var == 'Showdown' else ['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
+                elif sport_var == 'NHL':
+                    slate_var3 = st.radio("Which slate data are you loading?", (nhl_slate_names_fd if type_var == 'Showdown' else ['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
                 else:
                     slate_var3 = st.radio("Which slate data are you loading?", (['Main', 'Secondary', 'Auxiliary']), key='slate_var3_radio')
 
@@ -785,8 +843,12 @@ with st.container():
                 salary_min_var = st.number_input("Minimum salary used", min_value = 0, max_value = 50000, value = 49000, step = 100, key = 'salary_min_var_dk')
                 salary_max_var = st.number_input("Maximum salary used", min_value = 0, max_value = 50000, value = 50000, step = 100, key = 'salary_max_var_dk')
             elif site_var == 'Fanduel':
-                salary_min_var = st.number_input("Minimum salary used", min_value = 0, max_value = 60000, value = 59000, step = 100, key = 'salary_min_var_fd')
-                salary_max_var = st.number_input("Maximum salary used", min_value = 0, max_value = 60000, value = 60000, step = 100, key = 'salary_max_var_fd')
+                if sport_var == 'NHL':
+                    salary_min_var = st.number_input("Minimum salary used", min_value = 0, max_value = 55000, value = 54000, step = 100, key = 'salary_min_var_fd')
+                    salary_max_var = st.number_input("Maximum salary used", min_value = 0, max_value = 55000, value = 55000, step = 100, key = 'salary_max_var_fd')
+                else:
+                    salary_min_var = st.number_input("Minimum salary used", min_value = 0, max_value = 60000, value = 59000, step = 100, key = 'salary_min_var_fd')
+                    salary_max_var = st.number_input("Maximum salary used", min_value = 0, max_value = 60000, value = 60000, step = 100, key = 'salary_max_var_fd')
         with optimals_stacks_col:
             if site_var == 'Draftkings':
                 min_stacks_var = st.number_input("Minimum stacks used", min_value = 0, max_value = 5, value = 1, step = 1, key = 'min_stacks_var_dk')
@@ -811,6 +873,14 @@ with st.container():
         nba_showdown_salaries = grab_nba_showdown_salaries()
     except:
         nba_showdown_salaries = None
+    try:
+        nhl_reg_salaries = grab_nhl_reg_salaries(slate_var3)
+    except:
+        nhl_reg_salaries = None
+    try:
+        nhl_showdown_salaries = grab_nhl_showdown_salaries()
+    except:
+        nhl_showdown_salaries = None
 
 try:
     selected_tab = st.segmented_control(
@@ -860,6 +930,11 @@ if selected_tab == 'Data Load':
                         st.session_state['csv_file'] = load_csv(nfl_reg_salaries)
                     elif type_var == 'Showdown':
                         st.session_state['csv_file'] = load_csv(nfl_showdown_salaries)
+                elif sport_var == 'NHL':
+                    if type_var == 'Classic':
+                        st.session_state['csv_file'] = load_csv(nhl_reg_salaries)
+                    elif type_var == 'Showdown':
+                        st.session_state['csv_file'] = load_csv(nhl_showdown_salaries)
                 st.session_state['pricing_loaded'] = True
                 
             try:
@@ -932,11 +1007,15 @@ if selected_tab == 'Data Load':
                             portfolio_load = init_DK_NBA_lineups(type_var, slate_var3, prio_var, 50, dk_nba_showdown_db_translation, lineup_num_var, [])
                         elif sport_var == 'NFL':
                             portfolio_load = init_DK_NFL_lineups(type_var, slate_var3, prio_var, 50, dk_nfl_showdown_db_translation, lineup_num_var, [])
+                        elif sport_var == 'NHL':
+                            portfolio_load = init_DK_NHL_lineups(type_var, slate_var3, prio_var, 50, dk_nhl_showdown_db_translation, lineup_num_var, [])
                     else:
                         if sport_var == 'NBA':
                             portfolio_load = init_DK_NBA_lineups(type_var, nba_slate_name_lookup_dk[slate_var3], prio_var, 50, dk_nba_showdown_db_translation, lineup_num_var, [])
                         elif sport_var == 'NFL':
                             portfolio_load = init_DK_NFL_lineups(type_var, nfl_slate_name_lookup_dk[slate_var3], prio_var, 50, dk_nfl_showdown_db_translation, lineup_num_var, [])
+                        elif sport_var == 'NHL':
+                            portfolio_load = init_DK_NHL_lineups(type_var, nhl_slate_name_lookup_dk[slate_var3], prio_var, 50, dk_nhl_showdown_db_translation, lineup_num_var, [])
                     st.session_state['db_portfolio_file'] = pd.DataFrame(portfolio_load)
                     st.session_state['portfolio_loaded'] = True
                     if 'portfolio' in st.session_state:
@@ -949,11 +1028,15 @@ if selected_tab == 'Data Load':
                             portfolio_load = init_FD_NBA_lineups(type_var, slate_var3, prio_var, 50, fd_nba_showdown_db_translation, lineup_num_var, [])
                         elif sport_var == 'NFL':
                             portfolio_load = init_FD_NFL_lineups(type_var, slate_var3, prio_var, 50, fd_nfl_showdown_db_translation, lineup_num_var, [])
+                        elif sport_var == 'NHL':
+                            portfolio_load = init_FD_NHL_lineups(type_var, slate_var3, prio_var, 50, fd_nhl_showdown_db_translation, lineup_num_var, [])
                     else:
                         if sport_var == 'NBA':
                             portfolio_load = init_FD_NBA_lineups(type_var, nba_slate_name_lookup_fd[slate_var3], prio_var, 50, fd_nba_showdown_db_translation, lineup_num_var, [])
                         elif sport_var == 'NFL':
                             portfolio_load = init_FD_NFL_lineups(type_var, nfl_slate_name_lookup_fd[slate_var3], prio_var, 50, fd_nfl_showdown_db_translation, lineup_num_var, [])
+                        elif sport_var == 'NHL':
+                            portfolio_load = init_FD_NHL_lineups(type_var, nhl_slate_name_lookup_fd[slate_var3], prio_var, 50, fd_nhl_showdown_db_translation, lineup_num_var, [])
                     st.session_state['db_portfolio_file'] = pd.DataFrame(portfolio_load)
                     st.session_state['portfolio_loaded'] = True
                     if 'portfolio' in st.session_state:
@@ -1077,6 +1160,17 @@ if selected_tab == 'Data Load':
                                 projections_file = init_nfl_baselines(type_var, site_var, slate_var3)[1]
                             elif type_var == 'Showdown':
                                 projections_file = init_nfl_baselines(type_var, site_var, slate_var3)[3]
+                    elif sport_var == 'NHL':
+                        if site_var == 'Draftkings':
+                            if type_var == 'Classic':
+                                projections_file = init_nhl_baselines(type_var, site_var, slate_var3)[0]
+                            elif type_var == 'Showdown':
+                                projections_file = init_nhl_baselines(type_var, site_var, slate_var3)[2]
+                        elif site_var == 'Fanduel':
+                            if type_var == 'Classic':
+                                projections_file = init_nhl_baselines(type_var, site_var, slate_var3)[1]
+                            elif type_var == 'Showdown':
+                                projections_file = init_nhl_baselines(type_var, site_var, slate_var3)[3]
                     st.session_state['db_projections_file'] = projections_file
                     st.session_state['projections_loaded'] = True
             if 'projections_df' in st.session_state: