Get admin page to load

src/data_storage.py

@@ -1,37 +1,22 @@
-import os
 from secrets import token_urlsafe
 import streamlit as st
-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)")
-        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)")
+from queries.supabase_db.client import supabase_client
 
 
 def update_selection(user_id: str | int, position_id: str, player_id: str):
-    with get_db_connection() as con:
-        cur = con.cursor()
-        cur.execute(
-            f"""DELETE FROM user_rosters where user_id = {user_id} and position_id = '{position_id}'
-                    """
-        )
-        cur.execute(
-            f"""INSERT INTO user_rosters (user_id, position_id, player_id )
-                    VALUES({user_id}, '{position_id}', '{player_id}')
-                    """
-        )
+    pass
+    # with get_db_connection() as con:
+    #     cur = con.cursor()
+    #     cur.execute(
+    #         f"""DELETE FROM user_rosters where user_id = {user_id} and position_id = '{position_id}'
+    #                 """
+    #     )
+    #     cur.execute(
+    #         f"""INSERT INTO user_rosters (user_id, position_id, player_id )
+    #                 VALUES({user_id}, '{position_id}', '{player_id}')
+    #                 """
+    #     )
 
 
 def get_user_team(user_id):
@@ -55,16 +40,16 @@ def add_new_user(email: str, name: str):
 
 
 def get_user(user_id: int):
-    with get_db_connection() as con:
-        cur = con.cursor()
-        user_data = cur.execute(f"select * from users where user_id = {user_id}").fetchone()
+    user_data = (
+        supabase_client.table("npcs_users")
+        .select("user_id", "email", "name")
+        .eq("user_id", user_id)
+        .execute()
+        .data
+    )
     if not user_data:
         return {}
-    return {
-        "user_id": user_data[0],
-        "email": user_data[1],
-        "name": user_data[2],
-    }
+    return user_data[0]
 
 
 def get_user_id_if_email_exists(email: str) -> int | None:
@@ -79,6 +64,7 @@ def get_user_id_if_email_exists(email: str) -> int | None:
 
 
 def is_admin(user_id: int):
+    return True
     # Replace with db data field later
     return user_id == 1
 
@@ -97,6 +83,7 @@ def login_by_token(token: str):
 
 
 def create_new_token_for_user(user_id: int, existing_user: bool = False):
+    return {}
     # returns true if logged in successfully
     token = token_urlsafe(32)
     with get_db_connection() as con:
@@ -115,13 +102,6 @@ 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")
-
 
 def get_all_users(columns_included: list[str] = ["user_id", "name", "email"]):
     columns_as_str = ",".join(columns_included)