Spaces:
Running
Running
from sqlalchemy import create_engine | |
from sqlalchemy.orm import sessionmaker, declarative_base | |
DATABASE_URL = "sqlite:///./users.db" | |
engine = create_engine(DATABASE_URL) | |
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) | |
Base = declarative_base() | |
def get_db(): | |
"""Dependency function to get the database session.""" | |
db = SessionLocal() | |
try: | |
yield db | |
finally: | |
db.close() | |