Updating PGA for DK/DF availbaility

app.py

@@ -6,8 +6,6 @@ from rapidfuzz import process
 from collections import Counter
 import io
 
-## import global functions
-## Need to make sure that I take all functions from here and give them a home
 from global_func.clean_player_name import clean_player_name
 from global_func.load_file import load_file
 from global_func.load_ss_file import load_ss_file
@@ -1172,7 +1170,7 @@ with st.container():
         site_var = st.selectbox("Select Site", ['Draftkings', 'Fanduel'])
 
     with col3:
-        sport_var = st.selectbox("Select Sport", ['NFL', 'MLB', 'NBA', 'NHL', 'NCAAF', 'MMA', 'CS2', 'LOL', 'COD', 'TENNIS', 'NASCAR', 'GOLF', 'WNBA', 'F1'], key='sport_var')
+        sport_var = st.selectbox("Select Sport", ['NFL', 'MLB', 'NBA', 'NHL', 'MMA', 'NASCAR', 'GOLF'], key='sport_var')
         
     with col4:
         type_var = st.selectbox("Select Game Type", ['Classic', 'Showdown'])
@@ -1347,7 +1345,6 @@ except:
     )
 
 # Data Load Tab Functions
-# I am losing my mind
 if selected_tab == 'Data Load':
     col1, col2, col3 = st.columns(3)
 
@@ -1593,13 +1590,11 @@ if selected_tab == 'Data Load':
 
                 if st.session_state['portfolio'] is not None:
 
-                    # Optimize data types early for memory efficiency
                     st.session_state['portfolio'] = optimize_dataframe_dtypes(st.session_state['portfolio'])
                     
                     st.success('Portfolio file loaded successfully!')
                     for col in st.session_state['portfolio'].select_dtypes(include=['object', 'category']).columns:
                         if st.session_state['portfolio'][col].dtype == 'category':
-                            # Handle categorical columns
                             st.session_state['portfolio'][col] = st.session_state['portfolio'][col].cat.rename_categories(
                                 lambda x: player_right_names_mlb.get(x, x) if x in player_wrong_names_mlb else x
                             )

database.py

@@ -16,7 +16,8 @@ def init_conn():
         nhl_db = client["NHL_Database"]
         mma_db = client["MMA_Database"]
         pga_db = client["PGA_Database"]
+        nascar_db = client['NASCAR_Database']
 
-        return nfl_db, salaries_db, nba_db, mlb_db, nhl_db, mma_db, pga_db
+        return nfl_db, salaries_db, nba_db, mlb_db, nhl_db, mma_db, pga_db, nascar_db
     
-nfl_db, salaries_db, nba_db, mlb_db, nhl_db, mma_db, pga_db = init_conn()
+nfl_db, salaries_db, nba_db, mlb_db, nhl_db, mma_db, pga_db, nascar_db = init_conn()

database_queries.py

@@ -1672,20 +1672,19 @@ def init_pga_baselines(type_var: str, site_var: str, slate_var: str):
         cursor = collection.find()
 
         raw_display = pd.DataFrame(list(cursor))
-        raw_display = raw_display[['Player', 'Position', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id']]
+        raw_display = raw_display[['Player', 'Position', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id', 'site']]
         raw_display['Team'] = 'PGA'
-        raw_display['Site'] = site_var
         raw_display = raw_display.rename(columns={"player_id": "player_ID"})
         raw_display['Median'] = raw_display['Median'].replace('', 0).astype(float)
         raw_display = raw_display.loc[raw_display['Median'] > 0]
-        dk_roo_raw = raw_display[raw_display['Site'] == 'Draftkings']
-        fd_roo_raw = raw_display[raw_display['Site'] == 'Draftkings']
+        dk_roo_raw = raw_display[raw_display['site'] == 'Draftkings']
+        fd_roo_raw = raw_display[raw_display['site'] == 'Fanduel']
         dk_id_map = dict(zip(dk_roo_raw['Player'], dk_roo_raw['player_ID']))
         fd_id_map = dict(zip(fd_roo_raw['Player'], fd_roo_raw['player_ID']))
         raw_display = raw_display.apply(pd.to_numeric, errors='ignore')
 
-        dk_roo_raw = dk_roo_raw.drop(columns=['player_ID'])
-        fd_roo_raw = fd_roo_raw.drop(columns=['player_ID'])
+        dk_roo_raw = dk_roo_raw.drop(columns=['player_ID', 'site'])
+        fd_roo_raw = fd_roo_raw.drop(columns=['player_ID', 'site'])
 
         dk_roo_raw = dk_roo_raw.rename(columns={'Player': 'player_names', 'Position': 'position', 'Team': 'team', 'Salary': 'salary', 'Median': 'median', 'Own': 'ownership', 'CPT_Own': 'captain ownership'})
         fd_roo_raw = fd_roo_raw.rename(columns={'Player': 'player_names', 'Position': 'position', 'Team': 'team', 'Salary': 'salary', 'Median': 'median', 'Own': 'ownership', 'CPT_Own': 'captain ownership'})
@@ -2039,6 +2038,75 @@ def init_FD_PGA_lineups(type_var, slate_var, prio_var, prio_mix, lineup_num, sal
 
     return FD_seed
 
+def init_nascar_baselines(type_var: str, site_var: str, slate_var: str):
+
+    if slate_var == 'Main':
+        slate_var = 'Main Slate'
+    elif slate_var == 'Secondary':
+        slate_var = 'Secondary Slate'
+    elif slate_var == 'Auxiliary':
+        slate_var = 'Late Slate'
+
+    if type_var == 'Showdown':
+        collection = nascar_db["Player_Level_Showdown_ROO"]
+        cursor = collection.find()
+
+        raw_display = pd.DataFrame(list(cursor))
+        raw_display = raw_display[['Player', 'Position', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id', 'slate', 'site', 'version']]
+        raw_display['Team'] = 'NASCAR'
+        raw_display['Median'] = raw_display['Median'].replace('', 0).astype(float)
+        raw_display = raw_display.rename(columns={"player_id": "player_ID"})
+        raw_display = raw_display.loc[raw_display['Median'] > 0]
+        raw_display = raw_display.apply(pd.to_numeric, errors='ignore')
+        sd_raw = raw_display.sort_values(by='Median', ascending=False)
+        dk_sd_roo_raw = sd_raw[sd_raw['site'] == 'Draftkings']
+        dk_sd_id_map = dict(zip(dk_sd_roo_raw['Player'], dk_sd_roo_raw['player_ID']))
+        fd_sd_roo_raw = sd_raw[sd_raw['site'] == 'Fanduel']
+        fd_sd_id_map = dict(zip(fd_sd_roo_raw['Player'], fd_sd_roo_raw['player_ID']))
+        fd_sd_roo_raw['player_ID'] = fd_sd_roo_raw['player_ID'].astype(str)
+        fd_sd_roo_raw['player_ID'] = fd_sd_roo_raw['player_ID'].str.rsplit('-', n=1).str[0].astype(str)
+
+        dk_sd_roo_raw = dk_sd_roo_raw.drop(columns=['player_ID', 'slate', 'version', 'site'])
+        fd_sd_roo_raw = fd_sd_roo_raw.drop(columns=['player_ID', 'slate', 'version', 'site'])
+
+        dk_sd_roo_raw = dk_sd_roo_raw.rename(columns={'Player': 'player_names', 'Position': 'position', 'Team': 'team', 'Salary': 'salary', 'Median': 'median', 'Own': 'ownership', 'CPT_Own': 'captain ownership'})
+        fd_sd_roo_raw = fd_sd_roo_raw.rename(columns={'Player': 'player_names', 'Position': 'position', 'Team': 'team', 'Salary': 'salary', 'Median': 'median', 'Own': 'ownership', 'CPT_Own': 'captain ownership'})
+
+        dk_roo_raw = None
+        fd_roo_raw = None
+        dk_id_map = None
+        fd_id_map = None
+
+    else:
+        collection = nascar_db["Player_Level_ROO"]
+        cursor = collection.find()
+
+        raw_display = pd.DataFrame(list(cursor))
+        raw_display = raw_display[['Player', 'Position', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id']]
+        raw_display['Team'] = 'NASCAR'
+        raw_display['Site'] = site_var
+        raw_display = raw_display.rename(columns={"player_id": "player_ID"})
+        raw_display['Median'] = raw_display['Median'].replace('', 0).astype(float)
+        raw_display = raw_display.loc[raw_display['Median'] > 0]
+        dk_roo_raw = raw_display[raw_display['Site'] == 'Draftkings']
+        fd_roo_raw = raw_display[raw_display['Site'] == 'Draftkings']
+        dk_id_map = dict(zip(dk_roo_raw['Player'], dk_roo_raw['player_ID']))
+        fd_id_map = dict(zip(fd_roo_raw['Player'], fd_roo_raw['player_ID']))
+        raw_display = raw_display.apply(pd.to_numeric, errors='ignore')
+
+        dk_roo_raw = dk_roo_raw.drop(columns=['player_ID'])
+        fd_roo_raw = fd_roo_raw.drop(columns=['player_ID'])
+
+        dk_roo_raw = dk_roo_raw.rename(columns={'Player': 'player_names', 'Position': 'position', 'Team': 'team', 'Salary': 'salary', 'Median': 'median', 'Own': 'ownership', 'CPT_Own': 'captain ownership'})
+        fd_roo_raw = fd_roo_raw.rename(columns={'Player': 'player_names', 'Position': 'position', 'Team': 'team', 'Salary': 'salary', 'Median': 'median', 'Own': 'ownership', 'CPT_Own': 'captain ownership'})
+
+        dk_sd_roo_raw = None
+        fd_sd_roo_raw = None
+        dk_sd_id_map = None
+        fd_sd_id_map = None
+    
+    return dk_roo_raw, fd_roo_raw, dk_sd_roo_raw, fd_sd_roo_raw, dk_id_map, fd_id_map, dk_sd_id_map, fd_sd_id_map
+
 def init_DK_NASCAR_lineups(type_var, slate_var, prio_var, prio_mix, lineup_num, salary_min, salary_max, player_var2):
 
     if prio_var == 'Mix':