29 lines
723 B
Python
29 lines
723 B
Python
from sqlalchemy import create_engine
|
|
from dotenv import load_dotenv
|
|
import os
|
|
|
|
# Load .env vars
|
|
load_dotenv()
|
|
|
|
# Get env vars
|
|
USER = os.getenv("user")
|
|
PASSWORD = os.getenv("password")
|
|
HOST = os.getenv("host")
|
|
PORT = os.getenv("port")
|
|
DBNAME = os.getenv("dbname")
|
|
|
|
# Full SQLAlchemy URI for Session Pooler
|
|
DATABASE_URL = (
|
|
f"postgresql+psycopg2://{USER}:{PASSWORD}@{HOST}:{PORT}/{DBNAME}?sslmode=require"
|
|
)
|
|
|
|
# Use NullPool to defer to Supabase's pooler
|
|
from sqlalchemy.pool import NullPool
|
|
engine = create_engine(DATABASE_URL, poolclass=NullPool)
|
|
|
|
try:
|
|
with engine.connect() as conn:
|
|
print("✅ Supabase Session Pooler connection successful.")
|
|
except Exception as e:
|
|
print(f"❌ Connection failed: {e}")
|