25 lines
662 B
Python
25 lines
662 B
Python
import hashlib
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.db.base import Base
|
|
from app.db.session import SessionLocal, engine
|
|
from app.models import TestUser
|
|
|
|
|
|
def init_db() -> None:
|
|
Base.metadata.create_all(bind=engine)
|
|
seed_test_data()
|
|
|
|
|
|
def seed_test_data() -> None:
|
|
with SessionLocal() as db:
|
|
existing_user = db.scalar(select(TestUser).where(TestUser.username == "admin_test"))
|
|
if existing_user:
|
|
return
|
|
|
|
password_hash = hashlib.sha256("change_me_123".encode("utf-8")).hexdigest()
|
|
db.add(TestUser(username="admin_test", password_hash=password_hash))
|
|
db.commit()
|