Implement admin utility for migrating week

src/data_storage.py

@@ -20,7 +20,7 @@ def initialize_data_storage():
         cur.execute("CREATE TABLE IF NOT EXISTS tokens( user_id INTEGER PRIMARY KEY, token TEXT)")
 
 
-def update_selection(user_id: str, position_id: str, player_id: str):
+def update_selection(user_id: str | int, position_id: str, player_id: str):
     with get_db_connection() as con:
         cur = con.cursor()
         cur.execute(
@@ -136,3 +136,20 @@ def get_all_rosters() -> list[tuple[int, str, str]]:
         cur = con.cursor()
         all_rosters = cur.execute("select * from user_rosters").fetchall()
     return all_rosters
+
+
+def get_all_rosters_week(week: int) -> list[tuple[int, str, str]]:
+    with get_db_connection() as con:
+        cur = con.cursor()
+        all_rosters = cur.execute(f"select * from user_rosters where position_id like '{week}%'").fetchall()
+    return all_rosters
+
+
+def migrate_players_from_week(migrate_from_week: int):
+    """
+    Migrate players from the week = migrate_from_week to the week = migrate_from_week + 1
+    """
+    rosters = get_all_rosters_week(migrate_from_week)
+    for user_id, position_id, player_id in rosters:
+        new_position_id = f"""{migrate_from_week + 1}-{position_id.split("-", 1)[1]}"""
+        update_selection(user_id, new_position_id, player_id)

src/pages/99_Admin.py

@@ -4,7 +4,7 @@ import streamlit as st
 from config import DEFAULT_ICON
 from shared_page import common_page_config
 
-from data_storage import add_new_user, is_admin, DB_PATH, drop_tables
+from data_storage import add_new_user, is_admin, DB_PATH, drop_tables, migrate_players_from_week
 
 
 def admin_add_new_user():
@@ -47,6 +47,19 @@ def drop_db_form():
             st.form_submit_button("Submit", on_click=drop_db_if_confirmed)
 
 
+def migrate_players_week():
+    with st.container():
+        st.header("Utility for migrating players to the next week")
+        week_migrate_from = st.selectbox("week to migrate from", options=[1, 2, 3])
+        st.text_input("Enter week to confirm", key="week_migrate_from")
+        if st.button("Migrate Week"):
+            if st.session_state.get("week_migrate_from") == str(week_migrate_from):
+                migrate_players_from_week(week_migrate_from)
+                st.warning("Week migrated")
+            else:
+                st.warning("Must confirm migration by entering matching week")
+
+
 def get_page():
     page_title = "Admin"
     st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
@@ -60,6 +73,7 @@ def get_page():
     database_backup_form()
     admin_add_new_user_form()
     drop_db_form()
+    migrate_players_week()
 
 
 if __name__ == "__main__":