Add supabase client and ability to select keepers

src/pages/2_My_Keepers.py

@@ -4,7 +4,7 @@ import streamlit as st
 
 from config import DEFAULT_ICON, LEAGUE_NAME, KEEPER_DATA_URL
 from shared_page import common_page_config
-
+from queries.supabase_db.client import get_selected_keepers, select_keeper
 
 from login_component import is_token_in_session
 
@@ -43,33 +43,75 @@ def get_page():
         st.warning("Work in progress.  Ability to save selections is yet to be completed.")
         data, _ = load_data()
         data_user = data[(data["manager_guid"] == st.session_state.get("logged_in_guid")) & (data["eligible"])]
-        st.header("Options")
-        st.dataframe(
-            data_user,
-            hide_index=True,
-            height=35 * (len(data_user) + 1) + 5,
-            column_order=[
-                "headshot_url",
-                "name",
-                "team",
-                "position",
-                "keeper_cost",
-                "years_eligible",
-            ],
-            column_config={
-                "headshot_url": st.column_config.ImageColumn(label="", help="Player image"),
-                "name": st.column_config.TextColumn(label="Name", help="Player's name"),
-                "team": st.column_config.TextColumn(label="NFL Team"),
-                "position": st.column_config.TextColumn(label="Position", help="Player's position"),
-                "keeper_cost": st.column_config.NumberColumn(
-                    label="Keeper Cost", help="Draft Round Cost to keep player.  See Rules for details."
-                ),
-                "years_eligible": st.column_config.NumberColumn(
-                    label="Years Eligible",
-                    help="Number of further consecutive seasons player can be kept (subject to maximum of 2)",
-                ),
-            },
+        st.header("Selected Keepers")
+        selected_keeper_cost_map = get_selected_keepers(st.session_state.get("logged_in_guid"))
+        if selected_keeper_cost_map:
+            selected_data = data_user[data_user.player_id.isin(selected_keeper_cost_map.keys())]
+            selected_data["keeper_cost"] = selected_data.player_id.map(selected_keeper_cost_map)
+            st.dataframe(
+                selected_data,
+                hide_index=True,
+                height=35 * (len(selected_data) + 1) + 5,
+                column_order=[
+                    "headshot_url",
+                    "name",
+                    "position",
+                    "keeper_cost",
+                    "years_eligible",
+                ],
+                column_config={
+                    "headshot_url": st.column_config.ImageColumn(label="", help="Player image"),
+                    "name": st.column_config.TextColumn(label="Name", help="Player's name"),
+                    "position": st.column_config.TextColumn(label="Position", help="Player's position"),
+                    "keeper_cost": st.column_config.NumberColumn(
+                        label="Keeper Cost", help="Draft Round Cost to keep player.  See Rules for details."
+                    ),
+                    "years_eligible": st.column_config.NumberColumn(
+                        label="Years Eligible",
+                        help="Number of further consecutive seasons player can be kept (subject to maximum of 2)",
+                    ),
+                },
+            )
+        st.header("Add Keeper")
+
+        selection = st.selectbox(
+            "Add Keeper",
+            data_user[~data_user.player_id.isin(selected_keeper_cost_map.keys())].to_dict(orient="records"),
+            format_func=lambda x: f"{x['name']} - {x['keeper_cost']}",
+            index=None,
         )
+        if selection:
+            if st.button("Add Selection"):
+                select_keeper(selection["player_id"], selection["manager_guid"], selection["keeper_cost"])
+                st.rerun()
+        st.header("Options")
+        if st.checkbox("Show all my options"):
+            st.dataframe(
+                data_user,
+                hide_index=True,
+                height=35 * (len(data_user) + 1) + 5,
+                column_order=[
+                    "headshot_url",
+                    "name",
+                    "team",
+                    "position",
+                    "keeper_cost",
+                    "years_eligible",
+                ],
+                column_config={
+                    "headshot_url": st.column_config.ImageColumn(label="", help="Player image"),
+                    "name": st.column_config.TextColumn(label="Name", help="Player's name"),
+                    "team": st.column_config.TextColumn(label="NFL Team"),
+                    "position": st.column_config.TextColumn(label="Position", help="Player's position"),
+                    "keeper_cost": st.column_config.NumberColumn(
+                        label="Keeper Cost", help="Draft Round Cost to keep player.  See Rules for details."
+                    ),
+                    "years_eligible": st.column_config.NumberColumn(
+                        label="Years Eligible",
+                        help="Number of further consecutive seasons player can be kept (subject to maximum of 2)",
+                    ),
+                },
+            )
 
 
 if __name__ == "__main__":

src/queries/supabase_db/client.py

@@ -0,0 +1,24 @@
+import os
+
+from supabase import create_client, Client
+
+url: str = os.environ.get("SUPABASE_URL")
+key: str = os.environ.get("SUPABASE_KEY")
+
+supabase_client: Client = create_client(url, key)
+
+
+def get_selected_keepers(manager_guid: str):
+    """Returns a map of selected player id and actual keeper cost applied for logged in guid"""
+    response_data = (
+        supabase_client.table("selected_keepers").select("*").eq("manager_guid", manager_guid).execute().data
+    )
+    return {x["player_id"]: x["keeper_cost"] for x in response_data}
+
+
+def select_keeper(player_id: int, manager_guid: str, keeper_cost: int):
+    return (
+        supabase_client.table("selected_keepers")
+        .insert({"player_id": player_id, "manager_guid": manager_guid, "keeper_cost": keeper_cost})
+        .execute()
+    )