Files
FitTrack_GarminSync/examples/Garmin_Analyser/db/session.py

19 lines
480 B
Python

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker, declarative_base
from config.settings import DATABASE_URL
# Create the SQLAlchemy engine
engine = create_engine(DATABASE_URL)
# Create a SessionLocal class
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
# Create a Base class for declarative models
Base = declarative_base()
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()