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}")