|
|
|
from sqlalchemy import create_engine |
|
from sqlalchemy.ext.declarative import declarative_base |
|
from sqlalchemy.orm import sessionmaker |
|
from core.config import get_settings |
|
|
|
DATABASE_URL = "postgresql://snapfeast_db_owner:4fLiupkCsN5E@ep-fancy-glade-a57hqsn3-pooler.us-east-2.aws.neon.tech/feastdb?sslmode=require" |
|
|
|
engine = create_engine( |
|
DATABASE_URL, |
|
pool_size=5, |
|
pool_pre_ping=True, |
|
pool_recycle=300, |
|
max_overflow=0 |
|
) |
|
|
|
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) |
|
|
|
Base = declarative_base() |
|
|
|
def get_db(): |
|
db = SessionLocal() |
|
try: |
|
yield db |
|
finally: |
|
db.close() |
