Replace rosters stored in google with sqlite3 local file

src/data_storage.py

@@ -0,0 +1,37 @@
+import os
+import sqlite3
+
+DATA_DIR = "/data"
+
+DB_PATH = os.path.join(DATA_DIR, "data.db")
+
+
+def get_db_connection():
+    return sqlite3.connect(DB_PATH)
+
+
+def initialize_data_storage():
+    with get_db_connection() as con:
+        cur = con.cursor()
+        cur.execute("CREATE TABLE IF NOT EXISTS user_rosters( user_id INTEGER, position_id TEXT, player_id TEXT)")
+
+
+def update_selection(user_id: str, position_id: str, player_id: str):
+    with get_db_connection() as con:
+        cur = con.cursor()
+        cur.execute(
+            f"""REPLACE INTO user_rosters (user_id, position_id, player_id )
+                    VALUES({user_id}, '{position_id}', '{player_id}')
+                    """
+        )
+
+
+def get_user_team(user_id):
+    with get_db_connection() as con:
+        cur = con.cursor()
+        cur = get_db_connection().cursor()
+        team = cur.execute(f"select * from user_rosters where user_id = {user_id}").fetchall()
+    if team:
+        return {x[1]: x[2] for x in team}
+    else:
+        return {}

src/login.py

@@ -135,25 +135,6 @@ def check_password():
     return False
 
 
-def get_user_team():
-    user_id = st.session_state.get("logged_in_user")
-    if not user_id:
-        return {}
-    df_team = conn.read(
-        worksheet=f"user-{user_id}-roster",
-        ttl=1,
-    )
-    return df_team.loc[0].to_dict()
-
-
-def save_user_team(team_selections):
-    user_id = st.session_state.get("logged_in_user")
-    conn.update(
-        worksheet=f"user-{user_id}-roster",
-        data=pd.DataFrame(team_selections, index=[0]),
-    )
-
-
 def login_by_token(token: str):
     # returns true if logged in successfully
     df = conn.read(

src/pages/10_Set_Your_Lineup.py

@@ -10,7 +10,8 @@ from domain.playoffs import PLAYOFF_WEEK_TO_NAME, CURRENT_PLAYOFF_WEEK, ROSTER_W
 from domain.teams import SCHEDULE_NAME_TO_PFR_NAME_MAP, PLAYOFFS_TEAMS
 from queries.nflverse.github_data import get_weekly_rosters
 from queries.pfr.league_schedule import get_season_time_map
-from login import check_password, get_user_team, save_user_team
+from login import check_password
+from data_storage import update_selection, get_user_team
 
 
 @dataclass
@@ -159,15 +160,14 @@ def position_cell(
                 display_player(selected_player)
                 return
             else:
-                update_and_save_selection(pos_label, selected_player_from_box.gsis_id, existing_selection_map),
+                update_and_save_selection(pos_label, selected_player_from_box.gsis_id),
         display_player(selected_player_from_box)
     else:
         display_player(selected_player)
 
 
-def update_and_save_selection(pos_label: str, selection_id: str, existing_selection_map):
-    existing_selection_map[pos_label] = selection_id
-    save_user_team(existing_selection_map)
+def update_and_save_selection(pos_label: str, selection_id: str):
+    update_selection(st.session_state["logged_in_user"], pos_label, selection_id)
 
 
 def get_page():
@@ -182,7 +182,7 @@ def get_page():
 
     if st.button("Refresh Data"):
         st.rerun()
-    existing_selections = get_user_team()
+    existing_selections = get_user_team(st.session_state["logged_in_user"])
 
     player_options = load_options()