Jon Solow commited on
Commit
c24fc80
·
1 Parent(s): 95bf27b

Add retry wrapper for cold start of database

Browse files
Files changed (1) hide show
  1. src/pages/2_My_Keepers.py +7 -1
src/pages/2_My_Keepers.py CHANGED
@@ -1,6 +1,7 @@
1
  import os
2
  import pandas as pd
3
  import streamlit as st
 
4
 
5
  from config import DEFAULT_ICON, LEAGUE_NAME, KEEPER_DATA_URL
6
  from shared_page import common_page_config
@@ -27,6 +28,11 @@ def load_data():
27
  return data, teams_list
28
 
29
 
 
 
 
 
 
30
  def get_page():
31
  page_title = f"{LEAGUE_NAME} - My Keepers"
32
  st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
@@ -44,7 +50,7 @@ def get_page():
44
  data, _ = load_data()
45
  data_user = data[(data["manager_guid"] == st.session_state.get("logged_in_guid")) & (data["eligible"])]
46
  st.header("Selected Keepers")
47
- selected_keeper_cost_map = get_selected_keepers(st.session_state.get("logged_in_guid"))
48
  if selected_keeper_cost_map:
49
  selected_data = data_user[data_user.player_id.isin(selected_keeper_cost_map.keys())]
50
  selected_data["keeper_cost"] = selected_data.player_id.map(selected_keeper_cost_map)
 
1
  import os
2
  import pandas as pd
3
  import streamlit as st
4
+ from tenacity import retry
5
 
6
  from config import DEFAULT_ICON, LEAGUE_NAME, KEEPER_DATA_URL
7
  from shared_page import common_page_config
 
28
  return data, teams_list
29
 
30
 
31
+ @retry()
32
+ def get_selected_keepers_with_retry():
33
+ return get_selected_keepers(st.session_state.get("logged_in_guid"))
34
+
35
+
36
  def get_page():
37
  page_title = f"{LEAGUE_NAME} - My Keepers"
38
  st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
 
50
  data, _ = load_data()
51
  data_user = data[(data["manager_guid"] == st.session_state.get("logged_in_guid")) & (data["eligible"])]
52
  st.header("Selected Keepers")
53
+ selected_keeper_cost_map = get_selected_keepers_with_retry()
54
  if selected_keeper_cost_map:
55
  selected_data = data_user[data_user.player_id.isin(selected_keeper_cost_map.keys())]
56
  selected_data["keeper_cost"] = selected_data.player_id.map(selected_keeper_cost_map)