app/db.py

from app.sql.models import Base
from app.sql.models import User
from app.core.crud import hash_password
from app.sql.database import SessionLocal, engine
from app.config import SUPERUSER_USERNAME, SUPERUSER_PASSWORD


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


def init_db():
    db = SessionLocal()
    Base.metadata.create_all(bind=engine)

    hashed_superuser_password = hash_password(SUPERUSER_PASSWORD)
    superuser = User(username=SUPERUSER_USERNAME,
                     password=hashed_superuser_password, is_superuser=True, is_active=True)

    if db.query(User).filter(User.username == SUPERUSER_USERNAME).first() is None:
        db.add(superuser)
        db.commit()
        db.refresh(superuser)