Fix connection closing

subscriptions.py

@@ -2,6 +2,10 @@ import os
 import psycopg
 from psycopg.rows import dict_row
 from supabase import create_client, Client
+import asyncio
+
+conn = None
+conn_lock = asyncio.Lock()
 
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_KEY = os.getenv("SUPABASE_SERVICE_ROLE_KEY")
@@ -11,6 +15,15 @@ POSTGRE_SECRET = os.getenv("POSTGRE_SECRET")
 conn = psycopg.connect(POSTGRE_SECRET, row_factory=dict_row)
 
 
+async def get_conn():
+    global conn
+
+    if conn is None or conn.closed:
+        conn = psycopg.connect(POSTGRE_SECRET, row_factory=dict_row)
+
+    return conn
+
+
 def normalize_plan_key(plan_name: str | None) -> str:
     if not plan_name:
         return "free"
@@ -27,7 +40,6 @@ def normalize_plan_key(plan_name: str | None) -> str:
 
 
 async def fetch_subscription(jwt: str):
-    # You still need Supabase auth to get the user email
     auth_res = supabase.auth.get_user(jwt)
     if auth_res.user is None:
         return {"error": "Invalid or expired session"}
@@ -35,39 +47,48 @@ async def fetch_subscription(jwt: str):
     user = auth_res.user
     email = user.email
 
-    with conn.cursor() as cur:
-        cur.execute("""
-            with cust as (
-                select id 
-                from stripe.customers 
-                where email = %s
-            ),
-            subs as (
-                select
-                    s.id as subscription_id,
-                    s.status,
-                    s.current_period_end,
-                    s.items->'data'->0->'price'->>'id' as price_id
-                from stripe.subscriptions s
-                join cust on s.customer = cust.id
-                where s.status in ('active', 'trialing', 'past_due')
-            )
-            select
-                subs.subscription_id,
-                subs.status,
-                subs.current_period_end,
-                subs.price_id,
-                prices.nickname,
-                prices.product as product_id,
-                products.name as product_name
-            from subs
-            left join stripe.prices prices
-              on prices.id = subs.price_id
-            left join stripe.products products
-              on prices.product = products.id;
-        """, (email,))
-
-        rows = cur.fetchall()
+    async with conn_lock:
+        connection = await get_conn()
+
+        try:
+            with connection.cursor() as cur:
+                cur.execute("""
+                    with cust as (
+                        select id 
+                        from stripe.customers 
+                        where email = %s
+                    ),
+                    subs as (
+                        select
+                            s.id as subscription_id,
+                            s.status,
+                            s.current_period_end,
+                            s.items->'data'->0->'price'->>'id' as price_id
+                        from stripe.subscriptions s
+                        join cust on s.customer = cust.id
+                        where s.status in ('active', 'trialing', 'past_due')
+                    )
+                    select
+                        subs.subscription_id,
+                        subs.status,
+                        subs.current_period_end,
+                        subs.price_id,
+                        prices.nickname,
+                        prices.product as product_id,
+                        products.name as product_name
+                    from subs
+                    left join stripe.prices prices
+                      on prices.id = subs.price_id
+                    left join stripe.products products
+                      on prices.product = products.id;
+                """, (email,))
+                rows = cur.fetchall()
+
+        except psycopg.OperationalError:
+            connection = psycopg.connect(POSTGRE_SECRET, row_factory=dict_row)
+            with connection.cursor() as cur:
+                cur.execute(""" ... same SQL ... """, (email,))
+                rows = cur.fetchall()
 
     if not rows:
         return {