Chandima Prabhath commited on
Commit
1c5a919
Β·
1 Parent(s): d013b54

Refactor ensure_user_exists function to use maybe_single for user retrieval

Browse files
Files changed (2) hide show
  1. app.py +5 -1
  2. requirements.txt +1 -2
app.py CHANGED
@@ -76,7 +76,11 @@ def worker():
76
  # β€”β€”β€”β€”β€” Supabase Helpers β€”β€”β€”β€”β€”
77
  def ensure_user_exists(chat_id: str):
78
  """Create user record if not exists."""
79
- res = supabase.table("users").select("*").eq("chat_id", chat_id).single().execute()
 
 
 
 
80
  if res.data is None:
81
  supabase.table("users").insert({
82
  "chat_id": chat_id,
 
76
  # β€”β€”β€”β€”β€” Supabase Helpers β€”β€”β€”β€”β€”
77
  def ensure_user_exists(chat_id: str):
78
  """Create user record if not exists."""
79
+ res = supabase.table("users") \
80
+ .select("*") \
81
+ .eq("chat_id", chat_id) \
82
+ .maybe_single() \
83
+ .execute()
84
  if res.data is None:
85
  supabase.table("users").insert({
86
  "chat_id": chat_id,
requirements.txt CHANGED
@@ -4,5 +4,4 @@ openai
4
  flask
5
  pillow
6
  requests
7
- supabase
8
- psycopg2
 
4
  flask
5
  pillow
6
  requests
7
+ supabase