James McCool commited on
Commit
9e77e56
·
1 Parent(s): 46bf4d1

added some database persistence

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -565,7 +565,7 @@ if selected_tab == 'Data Load':
565
  if st.button("Load from Database after inserting site CSV"):
566
  if site_var == 'Draftkings':
567
  portfolio_load = init_DK_NFL_lineups(type_var, 'Main', 'proj', 50, dk_nfl_showdown_db_translation, 1000, [])
568
- portfolio_file = pd.DataFrame(portfolio_load)
569
  st.session_state['portfolio_loaded'] = True
570
  if 'portfolio' in st.session_state:
571
  del st.session_state['portfolio']
@@ -573,12 +573,15 @@ if selected_tab == 'Data Load':
573
  del st.session_state['export_portfolio']
574
  else:
575
  portfolio_load = init_FD_NFL_lineups(type_var, 'Main', 'proj', 50, fd_nfl_showdown_db_translation, 1000, [])
576
- portfolio_file = pd.DataFrame(portfolio_load)
577
  st.session_state['portfolio_loaded'] = True
578
  if 'portfolio' in st.session_state:
579
  del st.session_state['portfolio']
580
  if 'export_portfolio' in st.session_state:
581
  del st.session_state['export_portfolio']
 
 
 
582
 
583
  elif upload_toggle == 'SaberSim (Just IDs)' or upload_toggle == 'Draftkings/Fanduel (Names + IDs)':
584
  portfolio_file = st.file_uploader("Upload Portfolio File (CSV or Excel)", type=['csv', 'xlsx', 'xls'])
@@ -599,6 +602,7 @@ if selected_tab == 'Data Load':
599
  if 'portfolio' not in st.session_state:
600
  if st.session_state['portfolio_loaded']:
601
  if upload_toggle == 'Paydirt DB':
 
602
  st.session_state['export_portfolio'] = portfolio_file
603
  st.session_state['portfolio'] = portfolio_file
604
 
 
565
  if st.button("Load from Database after inserting site CSV"):
566
  if site_var == 'Draftkings':
567
  portfolio_load = init_DK_NFL_lineups(type_var, 'Main', 'proj', 50, dk_nfl_showdown_db_translation, 1000, [])
568
+ st.session_state['db_portfolio_file'] = pd.DataFrame(portfolio_load)
569
  st.session_state['portfolio_loaded'] = True
570
  if 'portfolio' in st.session_state:
571
  del st.session_state['portfolio']
 
573
  del st.session_state['export_portfolio']
574
  else:
575
  portfolio_load = init_FD_NFL_lineups(type_var, 'Main', 'proj', 50, fd_nfl_showdown_db_translation, 1000, [])
576
+ st.session_state['db_portfolio_file'] = pd.DataFrame(portfolio_load)
577
  st.session_state['portfolio_loaded'] = True
578
  if 'portfolio' in st.session_state:
579
  del st.session_state['portfolio']
580
  if 'export_portfolio' in st.session_state:
581
  del st.session_state['export_portfolio']
582
+ # Use database portfolio if it exists
583
+ elif 'db_portfolio_file' in st.session_state:
584
+ st.session_state['portfolio_loaded'] = True
585
 
586
  elif upload_toggle == 'SaberSim (Just IDs)' or upload_toggle == 'Draftkings/Fanduel (Names + IDs)':
587
  portfolio_file = st.file_uploader("Upload Portfolio File (CSV or Excel)", type=['csv', 'xlsx', 'xls'])
 
602
  if 'portfolio' not in st.session_state:
603
  if st.session_state['portfolio_loaded']:
604
  if upload_toggle == 'Paydirt DB':
605
+ portfolio_file = st.session_state['db_portfolio_file']
606
  st.session_state['export_portfolio'] = portfolio_file
607
  st.session_state['portfolio'] = portfolio_file
608