session.py 477 B

12345678910111213141516
  1. """
  2. SQLAlchemy engine and sessions tools
  3. """
  4. from sqlalchemy import create_engine
  5. from sqlalchemy.orm import sessionmaker
  6. from app.core import config
  7. if config.settings.ENVIRONMENT == "PYTEST":
  8. sqlalchemy_database_uri = config.settings.TEST_SQLALCHEMY_DATABASE_URI
  9. else:
  10. sqlalchemy_database_uri = config.settings.DEFAULT_SQLALCHEMY_DATABASE_URI
  11. engine = create_engine(sqlalchemy_database_uri)
  12. session = sessionmaker(autocommit=False, autoflush=False, bind=engine)