import getpass
import os

from app.db import SessionLocal
from app.services import UserService


def main():
    db = SessionLocal()
    try:
        service = UserService(db)
        username = os.environ.get("ADMIN_USERNAME") or input("Usuario admin: ")
        email = os.environ.get("ADMIN_EMAIL") or input("Email admin: ")
        password = os.environ.get("ADMIN_PASSWORD") or getpass.getpass("Contrasena admin: ")
        service.create_user(username=username, email=email, password=password, role="admin")
        print("Administrador creado")
    finally:
        db.close()


if __name__ == "__main__":
    main()