rockybot / scripts /create_admin.py
youaremywind
Initial commit
1e7a182
Raw
History Blame Contribute Delete
649 Bytes
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()