# database.py

from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine
from base import Base
from models import User

# Your database-related code here
DATABASE_URL = "sqlite:///./test.db"  # Update this with your database URL

engine = create_engine(DATABASE_URL)

SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

# Function to get a user by email
def get_user_by_email(db, email):
    return db.query(models.User).filter(models.User.email == email).first()