James McCool commited on
Commit
6a84d90
·
1 Parent(s): 2d5437b

Enhance position filtering in create_position_export_dict to include 'QB' for NCAAF, and streamline sorting logic for Draftkings and other sites.

Browse files
Files changed (1) hide show
  1. app.py +6 -9
app.py CHANGED
@@ -54,6 +54,8 @@ def create_position_export_dict(column_name, csv_file, site_var, type_var, sport
54
  elif position_filter == 'FLEX' or position_filter == 'UTIL':
55
  if sport_var == 'NFL':
56
  filtered_df = csv_file['Position'].isin(['RB', 'WR', 'TE'])
 
 
57
  elif sport_var == 'SOC':
58
  filtered_df = csv_file['Position'].str.contains(['D', 'M', 'F'], na=False, regex=False)
59
  elif sport_var == 'NCAAF':
@@ -77,17 +79,12 @@ def create_position_export_dict(column_name, csv_file, site_var, type_var, sport
77
  filtered_df = csv_file
78
 
79
  # Create the export dictionary for this position
80
-
81
- try:
82
  filtered_df = filtered_df.sort_values(by='Salary', ascending=False).drop_duplicates(subset=['Name'])
83
  return dict(zip(filtered_df['Name'], filtered_df['Name + ID']))
84
- except:
85
- try:
86
- filtered_df = filtered_df.sort_values(by='Salary', ascending=False).drop_duplicates(subset=['Nickname'])
87
- return dict(zip(filtered_df['Nickname'], filtered_df['Id']))
88
- except:
89
- # Final fallback
90
- return {}
91
 
92
  except Exception as e:
93
  st.error(f"Error creating position export dict for {column_name}: {str(e)}")
 
54
  elif position_filter == 'FLEX' or position_filter == 'UTIL':
55
  if sport_var == 'NFL':
56
  filtered_df = csv_file['Position'].isin(['RB', 'WR', 'TE'])
57
+ elif sport_var == 'NCAAF':
58
+ filtered_df = csv_file['Position'].isin(['RB', 'WR', 'TE', 'QB'])
59
  elif sport_var == 'SOC':
60
  filtered_df = csv_file['Position'].str.contains(['D', 'M', 'F'], na=False, regex=False)
61
  elif sport_var == 'NCAAF':
 
79
  filtered_df = csv_file
80
 
81
  # Create the export dictionary for this position
82
+ if site_var == 'Draftkings':
 
83
  filtered_df = filtered_df.sort_values(by='Salary', ascending=False).drop_duplicates(subset=['Name'])
84
  return dict(zip(filtered_df['Name'], filtered_df['Name + ID']))
85
+ else:
86
+ filtered_df = filtered_df.sort_values(by='Salary', ascending=False).drop_duplicates(subset=['Nickname'])
87
+ return dict(zip(filtered_df['Nickname'], filtered_df['Id']))
 
 
 
 
88
 
89
  except Exception as e:
90
  st.error(f"Error creating position export dict for {column_name}: {str(e)}")