database/__init__.py

from .database import SessionLocal, engine, Base
from .schema import *
from .operation import *


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