James McCool commited on
Commit
9e80538
·
1 Parent(s): 5db8a23

Enhance file loading functionality in `app.py` and `load_file.py`

Browse files

- Updated `app.py` to include `entry_list` in session state after loading contest files, improving data accessibility.
- Modified `load_file.py` to generate and return a sorted list of unique entry names, enhancing data organization and usability.

Files changed (2) hide show
  1. app.py +1 -1
  2. global_func/load_file.py +4 -2
app.py CHANGED
@@ -61,7 +61,7 @@ with tab1:
61
  del st.session_state['Contest']
62
 
63
  if Contest_file:
64
- st.session_state['Contest'], st.session_state['position_dict'], st.session_state['ownership_dict'] = load_file(Contest_file)
65
  st.session_state['Contest'] = st.session_state['Contest'].dropna(how='all')
66
  st.session_state['Contest'] = st.session_state['Contest'].reset_index(drop=True)
67
  if st.session_state['Contest'] is not None:
 
61
  del st.session_state['Contest']
62
 
63
  if Contest_file:
64
+ st.session_state['Contest'], st.session_state['position_dict'], st.session_state['ownership_dict'], st.session_state['entry_list'] = load_file(Contest_file)
65
  st.session_state['Contest'] = st.session_state['Contest'].dropna(how='all')
66
  st.session_state['Contest'] = st.session_state['Contest'].reset_index(drop=True)
67
  if st.session_state['Contest'] is not None:
global_func/load_file.py CHANGED
@@ -21,12 +21,14 @@ def load_file(upload):
21
 
22
  df = raw_df[['EntryId', 'EntryName', 'TimeRemaining', 'Points', 'Lineup', 'Player', 'Roster Position', '%Drafted', 'FPTS']]
23
  df = df.rename(columns={'Roster Position': 'Pos', '%Drafted': 'Own'})
24
- df['Lineup'] = df['Lineup'].replace(pos_values, ',')
25
  df['Lineup'] = df['Lineup'].str.split(',')
26
  position_dict = dict(zip(df['Player'], df['Pos']))
27
  ownership_dict = dict(zip(df['Player'], df['Own']))
 
 
28
 
29
- return df, position_dict, ownership_dict
30
  except Exception as e:
31
  st.error(f'Error loading file: {str(e)}')
32
  return None
 
21
 
22
  df = raw_df[['EntryId', 'EntryName', 'TimeRemaining', 'Points', 'Lineup', 'Player', 'Roster Position', '%Drafted', 'FPTS']]
23
  df = df.rename(columns={'Roster Position': 'Pos', '%Drafted': 'Own'})
24
+ df['Lineup'] = df['Lineup'].replace(pos_values, [',']*len(pos_values), regex=True)
25
  df['Lineup'] = df['Lineup'].str.split(',')
26
  position_dict = dict(zip(df['Player'], df['Pos']))
27
  ownership_dict = dict(zip(df['Player'], df['Own']))
28
+ entry_list = list(set(df['EntryName']))
29
+ entry_list.sort()
30
 
31
+ return df, position_dict, ownership_dict, entry_list
32
  except Exception as e:
33
  st.error(f'Error loading file: {str(e)}')
34
  return None