Add optional prompt for apiKey

Author: BartoszCki

paperspace/cli/auth.py

@@ -1,6 +1,7 @@
 import click
 
 from paperspace import client, config
+from paperspace.cli import common
 from paperspace.cli.cli import cli
 from paperspace.cli.validators import validate_email
 from paperspace.commands import login as login_commands
@@ -39,7 +40,7 @@ def logout():
 
 
 @cli.command("apiKey", help="Save your api key")
-@click.argument("api_key")
+@click.argument("api_key", required=False, callback=common.prompt_for_secret("API Key: "))
 def save_api_key(api_key):
     command = login_commands.SetApiKeyCommand()
     command.execute(api_key)

paperspace/cli/common.py

@@ -1,3 +1,5 @@
+import getpass
+
 import click
 from click_didyoumean import DYMMixin
 from click_help_colors import HelpColorsGroup
@@ -18,3 +20,13 @@ def del_if_value_is_none(dict_):
 
 class ClickGroup(DYMMixin, HelpColorsGroup):
     pass
+
+
+def prompt_for_secret(prompt):
+    def callback_fun(ctx, param, value):
+        if value is None:
+            value = getpass.getpass(prompt)
+
+        return value
+
+    return callback_fun