from sqlalchemy.orm import sessionmaker

from backend.models import Base, User
from backend.database import engine
from backend.security import get_password_hash


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

    Session = sessionmaker(bind=engine)
    db = Session()
    try:
        if not db.query(User).filter(User.username == 'admin').first():
            db.add(User(
                username='admin',
                password_hash=get_password_hash('admin'),
                role='admin',
            ))
            db.commit()
            print("Created default admin user (login: admin / password: admin)")
    finally:
        db.close()


if __name__ == "__main__":
    init_db()