Add functionality to grab and update player SR_IDs in app.py, integrating API calls for team rosters and enhancing data management in the NFL_SR_IDs worksheet.

app.py

@@ -31,6 +31,7 @@ gc = gspread.service_account_from_dict(credentials)
 
 NCAAF_model_url = st.secrets['NCAAF_model_url']
 pff_url = st.secrets['pff_url']
+FL_api_hub_url = st.secrets['FL_master_api_hub']
 
 headers = {
     'Authorization': 'Bearer 6984da1f-2c81-4140-8206-d018af38533f',
@@ -41,6 +42,63 @@ proj_source_options = ['Dwain', 'Aggregate', 'Freedman', 'Kickers', 'Defenses']
 proj_replace_options = ['dwain', 'aggregate', 'freedman', 'kickers', 'defenses']
 api_replace_options = ['dwain', 'freedman', 'ian', 'waz', 'charch', 'kendall', 'sam', 'berry', 'james', 'james_ncaaf']
 
+def grab_sr_ids(headers: dict) -> pd.DataFrame:
+    sr_injuries = "https://api.sportradar.com/nfl/official/trial/v7/en/league/hierarchy.json"
+    response = requests.get(sr_injuries, headers=headers)
+    data = response.json()
+
+    print(data)
+
+    team_ids = []
+    player_fa_ids = []
+    player_sr_ids = []
+    player_first_names = []
+    player_prefer_first_name = []
+    player_last_name = []
+    player_name_suffix = []
+    player_position = []
+    player_team = []
+
+    row_data = []
+
+    for conference in data['conferences']:
+        for division in conference['divisions']:
+            for team in division['teams']:
+                team_ids.append(team['id'])
+    
+    print(team_ids)
+
+    for team_id in team_ids:
+        
+        sr_injuries = f"https://api.sportradar.com/nfl/official/trial/v7/en/teams/{team_id}/full_roster.json"
+        response = requests.get(sr_injuries, headers=headers)
+        data = response.json()
+
+        team_name = data['alias']
+
+        st.write(team_name + ' - ' + team_id + ' being processed')
+        try:
+            for player in data['players']:
+                player_fa_ids.append(999999)
+                player_sr_ids.append(player['id'])
+                player_first_names.append(player['first_name'])
+                player_prefer_first_name.append('-')
+                player_last_name.append(player['last_name'])
+                player_name_suffix.append('-')
+                player_position.append(player['position'])
+                player_team.append(team_name)
+
+                row_data.append([
+                    999999, player['id'], player['first_name'], '-', player['last_name'], '-', player['position'], team_name
+                ])
+        except:
+            st.write(f"Error processing team: {team_name} - {team_id}")
+            continue
+    
+    st.write(pd.DataFrame(row_data, columns=['FA_ID', 'SR_ID', 'First_Name', 'Prefer_First_Name', 'Last_Name', 'Name_Suffix', 'Position', 'Team']).head(100))
+    
+    return pd.DataFrame(row_data, columns=['FA_ID', 'SR_ID', 'First_Name', 'Prefer_First_Name', 'Last_Name', 'Name_Suffix', 'Position', 'Team'])
+
 tab1, tab2, tab3, tab4, tab5 = st.tabs(['Rankings Update', 'Projections Update', 'Game Model Update', 'Rest of Season Update', 'NCAAF Script'])
 
 with tab1:
@@ -110,6 +168,17 @@ with tab2:
                   if response.status_code == 200:
                       st.write("Uploading!")
                       st.write(response.json())
+        
+        with col4:
+            st.info("Update SR_ID list")
+            if st.button("Update SR_ID list", key='sr_id_list'):
+                player_df = grab_sr_ids(headers)
+
+                sh = gc.open_by_url(FL_api_hub_url)
+                worksheet = sh.worksheet('NFL_SR_IDs')
+                worksheet.batch_clear(['A:H'])
+                worksheet.update([player_df.columns.values.tolist()] + player_df.values.tolist())
+                st.write("Uploaded SR_ID list")
 
 with tab3:
     st.info("We will add this functionality as soon as we have the endpoints to make it work!")