Allow database download

src/data_storage.py

@@ -1,4 +1,5 @@
 import os
+from secrets import token_urlsafe
 import sqlite3
 
 DATA_DIR = "/data"
@@ -15,6 +16,7 @@ def initialize_data_storage():
         cur = con.cursor()
         cur.execute("CREATE TABLE IF NOT EXISTS user_rosters( user_id INTEGER, position_id TEXT, player_id TEXT)")
         cur.execute("CREATE TABLE IF NOT EXISTS users( user_id INTEGER PRIMARY KEY ASC, email TEXT, name TEXT)")
+        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):
@@ -74,3 +76,31 @@ def get_user_id_if_email_exists(email: str) -> int | None:
 def is_admin(user_id: int):
     # Replace with db data field later
     return user_id == 1
+
+
+def login_by_token(token: str):
+    # returns true if logged in successfully
+    with get_db_connection() as con:
+        cur = con.cursor()
+        query_result = cur.execute(f"select user_id from users where token = '{token}'").fetchone()
+        if query_result:
+            user_id = query_result[0]
+        else:
+            user_id = None
+    return user_id
+
+
+def create_new_token_for_user(user_id: int, existing_user: bool = False):
+    # returns true if logged in successfully
+    token = token_urlsafe(32)
+    if existing_user:
+        sql_cmd = "REPLACE"
+    else:
+        sql_cmd = "INSERT"
+    with get_db_connection() as con:
+        cur = con.cursor()
+        cur.execute(
+            f"""{sql_cmd} INTO tokens (user_id, token )
+                    VALUES({user_id}, '{token}')
+                    """
+        )

src/pages/99_Admin.py

@@ -1,9 +1,10 @@
+import datetime
 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
+from data_storage import add_new_user, is_admin, DB_PATH
 
 
 def admin_add_new_user():
@@ -19,6 +20,14 @@ def admin_add_new_user_form():
         st.form_submit_button("Submit", on_click=admin_add_new_user)
 
 
+def database_backup_form():
+    with st.container():
+        st.header("Utility for backing up and restoring db file")
+        with open(DB_PATH, "rb") as f:
+            file_name = f"backup-{datetime.datetime.now().strftime('%y%m%d%H%M')}.db"
+            st.download_button("Download db", f, file_name=file_name)
+
+
 def get_page():
     page_title = "Admin"
     st.set_page_config(page_title=page_title, page_icon=DEFAULT_ICON, layout="wide")
@@ -29,6 +38,7 @@ def get_page():
 
     st.title(page_title)
 
+    database_backup_form()
     admin_add_new_user_form()