src/queries/supabase_db/client.py

import os

from supabase import create_client, Client

url: str = str(os.environ.get("SUPABASE_URL"))
key: str = str(os.environ.get("SUPABASE_KEY"))

supabase_client: Client = create_client(url, key)


def get_selected_keepers(manager_guid: str):
    """Returns a map of selected player id and actual keeper cost applied for logged in guid"""
    response_data = (
        supabase_client.table("selected_keepers").select("*").eq("manager_guid", manager_guid).execute().data
    )
    return {x["player_id"]: x["keeper_cost"] for x in response_data}


def get_all_selected_keepers():
    """Returns a map of selected player id and actual keeper cost applied for logged in guid"""
    response_data = supabase_client.table("selected_keepers").select("*").execute().data
    return response_data


def select_keeper(player_id: int, manager_guid: str, keeper_cost: int):
    return (
        supabase_client.table("selected_keepers")
        .insert({"player_id": player_id, "manager_guid": manager_guid, "keeper_cost": keeper_cost})
        .execute()
    )


def reset_keepers(manager_guid: str):
    return supabase_client.table("selected_keepers").delete().eq("manager_guid", manager_guid).execute().data