123456789101112131415161718 |
- from sqlalchemy import create_engine
- from sqlalchemy.ext.declarative import declarative_base
- from sqlalchemy.orm import sessionmaker, scoped_session
- from core.config import settings
- import os
- pool_size = 20
- if os.getenv("ENV") == "prod":
- pool_size = 100
- engine = create_engine(settings.SQLALCHEMY_DATABASE_URI, pool_pre_ping=True, pool_size=pool_size)
- db_session = scoped_session(
- sessionmaker(autocommit=False, autoflush=False, bind=engine)
- )
- SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
- Base = declarative_base()
|