James McCool commited on
Commit
86d9bdc
·
1 Parent(s): c14bbec

Consolidate position filtering logic in hitter tab of Streamlit app

Browse files

- Streamlined position filtering by removing redundant condition checks for DraftKings and FanDuel, enhancing code clarity and maintainability.
- Ensured that the filtered data is consistently stored in session state for improved functionality across different table views.

Files changed (1) hide show
  1. src/streamlit_app.py +5 -18
src/streamlit_app.py CHANGED
@@ -202,14 +202,9 @@ with hitter_tab:
202
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
203
 
204
  if position_var_hitter:
205
- if site_var_hitter == 'Draftkings':
206
- position_mask = disp_raw['Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
207
- disp_raw = disp_raw[position_mask]
208
- elif site_var_hitter == 'Fanduel':
209
- position_mask = disp_raw['FD_Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
210
- disp_raw = disp_raw[position_mask]
211
 
212
- st.session_state['hitter_disp_frame'] = disp_raw
213
  elif table_var_hitter == 'Active Baselines':
214
  disp_raw = hitter_info
215
 
@@ -217,36 +212,28 @@ with hitter_tab:
217
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
218
 
219
  if position_var_hitter:
220
- if site_var_hitter == 'Draftkings':
221
- position_mask = disp_raw['Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
222
- disp_raw = disp_raw[position_mask]
223
- elif site_var_hitter == 'Fanduel':
224
- position_mask = disp_raw['FD_Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
225
- disp_raw = disp_raw[position_mask]
226
 
227
- st.session_state['hitter_disp_frame'] = disp_raw
228
  elif table_var_hitter == 'League Aggregate Baselines':
229
  disp_raw = hitter_agg
230
 
231
  if team_var_hitter is not None:
232
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
233
 
234
- st.session_state['hitter_disp_frame'] = disp_raw
235
  elif table_var_hitter == 'League Short Term Baselines':
236
  disp_raw = hitter_short
237
 
238
  if team_var_hitter is not None:
239
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
240
 
241
- st.session_state['hitter_disp_frame'] = disp_raw
242
  elif table_var_hitter == 'League Long Term Baselines':
243
  disp_raw = hitter_long
244
 
245
  if team_var_hitter is not None:
246
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
247
 
248
- st.session_state['hitter_disp_frame'] = disp_raw
249
-
250
  hitter_disp_container = st.container(border = True)
251
  hitter_disp_container = hitter_disp_container.empty()
252
 
 
202
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
203
 
204
  if position_var_hitter:
205
+ position_mask = disp_raw['Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
206
+ disp_raw = disp_raw[position_mask]
 
 
 
 
207
 
 
208
  elif table_var_hitter == 'Active Baselines':
209
  disp_raw = hitter_info
210
 
 
212
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
213
 
214
  if position_var_hitter:
215
+ position_mask = disp_raw['Position'].apply(lambda x: any(pos in x for pos in position_var_hitter))
216
+ disp_raw = disp_raw[position_mask]
 
 
 
 
217
 
 
218
  elif table_var_hitter == 'League Aggregate Baselines':
219
  disp_raw = hitter_agg
220
 
221
  if team_var_hitter is not None:
222
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
223
 
 
224
  elif table_var_hitter == 'League Short Term Baselines':
225
  disp_raw = hitter_short
226
 
227
  if team_var_hitter is not None:
228
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
229
 
 
230
  elif table_var_hitter == 'League Long Term Baselines':
231
  disp_raw = hitter_long
232
 
233
  if team_var_hitter is not None:
234
  disp_raw = disp_raw[disp_raw['Team'].isin(team_var_hitter)]
235
 
236
+ st.session_state['hitter_disp_frame'] = disp_raw
 
237
  hitter_disp_container = st.container(border = True)
238
  hitter_disp_container = hitter_disp_container.empty()
239