from app.services.auth import hash_password
from app.db.session import SessionLocal
from app.db.models import Admin


def main():
    db = SessionLocal()
    try:
        email = input("Email: ").strip()
        password = input("Password: ").strip()
        admin = db.query(Admin).filter(Admin.email == email).first()
        if admin:
            admin.password_hash = hash_password(password)
        else:
            admin = Admin(email=email, password_hash=hash_password(password), role="owner")
            db.add(admin)
        db.commit()
        print("Admin saved")
    finally:
        db.close()


if __name__ == "__main__":
    main()