adaptors/db.py

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from models.story import Base, Story
from sqlalchemy.orm import declarative_base
from loguru import logger

engine = create_engine('sqlite:///data/stories.db')

Base.metadata.create_all(engine)

Session = sessionmaker(bind=engine)
session = Session()

def get_story(uuid):
    return session.query(Story).filter_by(uuid=uuid).first()

def get_all_stories():
    stories = session.query(Story).all()
    return stories

def save_story(story):
    session.add(story)
    session.commit()