correcting database pull

database_queries.py

@@ -452,7 +452,7 @@ def init_nba_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', 'Team', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id', 'slate', 'version']]
+        raw_display = raw_display[['Player', 'Position', 'Team', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id', 'slate', 'site', 'version', 'timestamp']]
         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]
@@ -465,9 +465,12 @@ def init_nba_baselines(type_var: str, site_var: str, slate_var: str):
         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)
 
-        timestamp = sd_raw['timestamp'].values[0]
+        dk_roo_raw = dk_roo_raw.drop(columns=['player_ID', 'slate', 'version', 'timestamp', 'site'])
+        fd_roo_raw = fd_roo_raw.drop(columns=['player_ID', 'slate', 'version', 'timestamp', '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'})
 
-        roo_raw = None
         dk_roo_raw = None
         fd_roo_raw = None
         dk_id_map = None
@@ -478,7 +481,7 @@ def init_nba_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', 'Team', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id', 'slate', 'version']]
+        raw_display = raw_display[['Player', 'Position', 'Team', 'Salary', 'Median', 'Own', 'CPT_Own', 'player_id', 'slate', 'site', 'version', 'timestamp']]
         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]
@@ -487,17 +490,19 @@ def init_nba_baselines(type_var: str, site_var: str, slate_var: str):
         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')
-        roo_raw = raw_display.sort_values(by='Median', ascending=False)
 
-        timestamp = roo_raw['timestamp'].values[0]
+        dk_roo_raw = dk_roo_raw.drop(columns=['player_ID', 'slate', 'version', 'timestamp', 'site'])
+        fd_roo_raw = fd_roo_raw.drop(columns=['player_ID', 'slate', 'version', 'timestamp', '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'})
 
-        sd_raw = None
         dk_sd_roo_raw = None
         fd_sd_roo_raw = None
         dk_sd_id_map = None
         fd_sd_id_map = None
     
-    return roo_raw, dk_roo_raw, fd_roo_raw, sd_raw, dk_sd_roo_raw, fd_sd_roo_raw, dk_id_map, fd_id_map, dk_sd_id_map, fd_sd_id_map, timestamp
+    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_NBA_lineups(type_var, slate_var, prio_var, prio_mix, nba_db_translation, lineup_num, player_var2):