wag-managment-api-service-v.../databases/sql_models/postgres_database.py

23 lines
586 B
Python

from api_configs import WagDatabase
from sqlalchemy import create_engine
from sqlalchemy.orm import scoped_session, sessionmaker
from sqlalchemy.ext.declarative import declarative_base
engine_config = {
"url": WagDatabase.DATABASE_URL,
"pool_size": 10,
"max_overflow": 0,
"echo": False,
}
engine = create_engine(**engine_config)
session_config = {"autoflush": True, "bind": engine, "echo": True}
SessionLocal = sessionmaker(**session_config)
session = scoped_session(sessionmaker(bind=engine))
session.expunge_all()
Base = declarative_base()
Base.session = session