Jon Solow commited on
Commit
d9e7c6f
·
1 Parent(s): e3c54e2

Copy supabase query client from other app

Browse files
src/queries/supabase_db/__init__.py ADDED
File without changes
src/queries/supabase_db/client.py ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+
3
+ from supabase import create_client, Client
4
+
5
+ url: str = str(os.environ.get("SUPABASE_URL"))
6
+ key: str = str(os.environ.get("SUPABASE_KEY"))
7
+
8
+ supabase_client: Client = create_client(url, key)
9
+
10
+
11
+ def get_selected_keepers(manager_guid: str):
12
+ """Returns a map of selected player id and actual keeper cost applied for logged in guid"""
13
+ response_data = (
14
+ supabase_client.table("selected_keepers").select("*").eq("manager_guid", manager_guid).execute().data
15
+ )
16
+ return {x["player_id"]: x["keeper_cost"] for x in response_data}
17
+
18
+
19
+ def get_all_selected_keepers():
20
+ """Returns a map of selected player id and actual keeper cost applied for logged in guid"""
21
+ response_data = supabase_client.table("selected_keepers").select("*").execute().data
22
+ return response_data
23
+
24
+
25
+ def select_keeper(player_id: int, manager_guid: str, keeper_cost: int):
26
+ return (
27
+ supabase_client.table("selected_keepers")
28
+ .insert({"player_id": player_id, "manager_guid": manager_guid, "keeper_cost": keeper_cost})
29
+ .execute()
30
+ )
31
+
32
+
33
+ def reset_keepers(manager_guid: str):
34
+ return supabase_client.table("selected_keepers").delete().eq("manager_guid", manager_guid).execute().data