from sqlalchemy.orm import Session
from models import Post, SessionLocal

# Get database session
def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

# Save post to database
def save_post(db: Session, post_data: dict):
    post = Post(**post_data)
    db.add(post)
    db.commit()
    db.refresh(post)
    return post