Implement db reset button in admin page and login via db

src/data_storage.py

@@ -105,3 +105,11 @@ def create_new_token_for_user(user_id: int, existing_user: bool = False):
                     """
         )
     return token
+
+
+def drop_tables():
+    with get_db_connection() as con:
+        cur = con.cursor()
+        cur.execute("DROP TABLE user_rosters")
+        cur.execute("DROP TABLE users")
+        cur.execute("DROP TABLE tokens")

src/login.py

@@ -1,4 +1,3 @@
-import hmac
 import json
 import pandas as pd
 import re
@@ -140,22 +139,6 @@ def check_password():
     return False
 
 
-def login_by_token(token: str):
-    # returns true if logged in successfully
-    df = conn.read(
-        worksheet="user-tokens",
-        usecols=[0, 1],
-        ttl=3600,
-    )
-    df = df[df.token.notna()]
-    for row in df.itertuples():
-        if hmac.compare_digest(row.token, token):
-            user_id = int(row.id)
-            st.session_state["logged_in_user"] = user_id
-            return True
-    return False
-
-
 def get_logged_in_user_name_email() -> tuple[str | None, str | None]:
     if not (user_id := st.session_state.get("logged_in_user")):
         # if not logged

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
+from data_storage import add_new_user, is_admin, DB_PATH, drop_tables
 
 
 def admin_add_new_user():
@@ -28,6 +28,22 @@ def database_backup_form():
             st.download_button("Download db", f, file_name=file_name)
 
 
+def drop_db_if_confirmed():
+    confirmed = st.session_state.get("delete_confirmation") == "delete"
+    if confirmed:
+        drop_tables()
+        st.warning("Database tables have been dropped")
+    else:
+        st.warning("Must confirm deletion by typing 'delete'")
+
+
+def drop_db_form():
+    with st.form("Reset DB"):
+        st.header("Reset DB")
+        st.text_input("Enter 'delete' to confirm deletion", key="delete_confirmation")
+        st.form_submit_button("Submit", on_click=drop_db_if_confirmed)
+
+
 def get_page():
     page_title = "Admin"
     st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
@@ -40,6 +56,7 @@ def get_page():
 
     database_backup_form()
     admin_add_new_user_form()
+    drop_db_form()
 
 
 if __name__ == "__main__":

src/shared_page.py

@@ -2,8 +2,7 @@ import os
 import streamlit as st
 
 from queries.nflverse.github_data import load_assets_if_no_tables
-from login import login_by_token
-from data_storage import initialize_data_storage
+from data_storage import initialize_data_storage, login_by_token
 
 
 def get_local_style():