diff --git a/app/cli.py b/app/cli.py index 820addb..e14983a 100644 --- a/app/cli.py +++ b/app/cli.py @@ -13,7 +13,10 @@ def cli(): @cli.command("setup") @click.option("--path", required=True) @click.option("--api_server", required=True) -def setup(path, api_server): +@click.option("--name", required=True) +@click.option("--realm", required=True) +@click.option("--seed", required=True) +def setup(path, api_server, name, realm, seed): os.chdir(path) repo_name = "ucloud-cli" @@ -26,6 +29,9 @@ def setup(path, api_server): with open(os.path.join(repo_name, ".env"), "w") as f: content = ( f"UCLOUD_API_SERVER={api_server}\n" + f"OTP_NAME={name}\n" + f"OTP_REALM={realm}\n" + f"OTP_SEED={seed}\n" ) f.writelines(content)