Jon Solow commited on
Commit
3b2782b
·
1 Parent(s): a0c0946

Add a reset button

Browse files
src/pages/2_My_Keepers.py CHANGED
@@ -4,7 +4,7 @@ import streamlit as st
4
 
5
  from config import DEFAULT_ICON, LEAGUE_NAME, KEEPER_DATA_URL
6
  from shared_page import common_page_config
7
- from queries.supabase_db.client import get_selected_keepers, select_keeper
8
 
9
  from login_component import is_token_in_session
10
 
@@ -113,6 +113,11 @@ def get_page():
113
  },
114
  )
115
 
 
 
 
 
 
116
 
117
  if __name__ == "__main__":
118
  get_page()
 
4
 
5
  from config import DEFAULT_ICON, LEAGUE_NAME, KEEPER_DATA_URL
6
  from shared_page import common_page_config
7
+ from queries.supabase_db.client import get_selected_keepers, select_keeper, reset_keepers
8
 
9
  from login_component import is_token_in_session
10
 
 
113
  },
114
  )
115
 
116
+ st.header("Reset All Keepers")
117
+ if st.button("Reset all my selections"):
118
+ reset_keepers(st.session_state.get("logged_in_guid"))
119
+ st.rerun()
120
+
121
 
122
  if __name__ == "__main__":
123
  get_page()
src/queries/supabase_db/client.py CHANGED
@@ -2,8 +2,8 @@ import os
2
 
3
  from supabase import create_client, Client
4
 
5
- url: str = os.environ.get("SUPABASE_URL")
6
- key: str = os.environ.get("SUPABASE_KEY")
7
 
8
  supabase_client: Client = create_client(url, key)
9
 
@@ -22,3 +22,7 @@ def select_keeper(player_id: int, manager_guid: str, keeper_cost: int):
22
  .insert({"player_id": player_id, "manager_guid": manager_guid, "keeper_cost": keeper_cost})
23
  .execute()
24
  )
 
 
 
 
 
2
 
3
  from supabase import create_client, Client
4
 
5
+ url: str = str(os.environ.get("SUPABASE_URL"))
6
+ key: str = str(os.environ.get("SUPABASE_KEY"))
7
 
8
  supabase_client: Client = create_client(url, key)
9
 
 
22
  .insert({"player_id": player_id, "manager_guid": manager_guid, "keeper_cost": keeper_cost})
23
  .execute()
24
  )
25
+
26
+
27
+ def reset_keepers(manager_guid: str):
28
+ return supabase_client.table("selected_keepers").delete().eq("manager_guid", manager_guid).execute().data