def set(**kwargs): # pylint:disable=redefined-builtin
"""Set the global config values.
Example:
\b
$ polyaxon config set --host=localhost
"""
try:
_config = ClientConfigManager.get_config_or_default()
except Exception as e:
handle_cli_error(e, message="Polyaxon load configuration.")
Printer.print_header(
"You can reset your config by running: polyaxon config purge"
)
sys.exit(1)
for key, value in kwargs.items():
if value is not None:
setattr(_config, key, value)
ClientConfigManager.set_config(_config)
Printer.print_success("Config was updated.")
# Reset cli config
CliConfigManager.purge()
def purge():
"""Purge the global config values."""
ClientConfigManager.purge()
CliConfigManager.purge()
AuthConfigManager.purge()
ProjectManager.purge()
RunManager.purge()
Printer.print_success("Config was removed.")
def set_cli_config():
from polyaxon.managers.cli import CliConfigManager
global CLI_CONFIG
try:
CLI_CONFIG = CliConfigManager.get_config_or_default()
except (TypeError, ValidationError):
CliConfigManager.purge()
Printer.print_warning("Your CLI Configuration was purged!")
def purge(cache_only):
"""Purge the global config values."""
if not cache_only:
ClientConfigManager.purge()
CliConfigManager.purge()
AuthConfigManager.purge()
UserConfigManager.purge()
ProjectConfigManager.purge()
RunConfigManager.purge()
Printer.print_success("Configs was removed.")
def port_forward(port, namespace, deployment_type, release_name):
"""If you deploy Polyaxon using ClusterIP, you can use this command
to access the gateway through `localhost:port`.
"""
from polyaxon.deploy.operators.kubectl import KubectlOperator
if not port and deployment_type in [
DeploymentTypes.MICRO_K8S,
DeploymentTypes.MINIKUBE,
]:
port = 31833
port = port or 8000
namespace = namespace or "polyaxon"
release_name = release_name or "polyaxon"
kubectl = KubectlOperator()
args = [
"port-forward",
"-n",
namespace,
"svc/{}-polyaxon-gateway".format(release_name),
"{}:80".format(port),
]
try:
_config = ClientConfigManager.get_config_or_default()
except Exception as e:
handle_cli_error(e, message="Polyaxon load configuration.")
Printer.print_header(
"You can reset your config by running: polyaxon config purge")
sys.exit(1)
_config.host = "http://localhost:{}".format(port)
ClientConfigManager.set_config(_config)
CliConfigManager.purge()
AuthConfigManager.purge()
UserConfigManager.purge()
Printer.print_header("Client configuration is updated!")
Printer.print_success("Polyaxon will be available at: {}".format(
_config.host))
stdout = kubectl.execute(args=args,
is_json=False,
stream=settings.CLIENT_CONFIG.debug)
click.echo(stdout)
def login(token, username, password):
"""Login to Polyaxon."""
polyaxon_client = PolyaxonClient()
if username and not token:
# Use user or email / password login
if not password:
password = click.prompt(
"Please enter your password", type=str, hide_input=True
)
password = password.strip()
if not password:
logger.info(
"You entered an empty string. "
"Please make sure you enter your password correctly."
)
sys.exit(1)
try:
body = V1Credentials(username=username, password=password)
access_auth = polyaxon_client.auth_v1.login(body=body)
except (ApiException, HTTPError) as e:
AuthConfigManager.purge()
CliConfigManager.purge()
handle_cli_error(e, message="Could not login.")
sys.exit(1)
if not access_auth.token:
Printer.print_error("Failed to login")
return
else:
if not token:
token_url = "{}/profile/token".format(
clean_host(polyaxon_client.config.host)
)
click.confirm(
"Authentication token page will now open in your browser. Continue?",
abort=True,
default=True,
)
click.launch(token_url)
logger.info("Please copy and paste the authentication token.")
token = click.prompt(
"This is an invisible field. Paste token and press ENTER",
type=str,
hide_input=True,
)
if not token:
logger.info(
"Empty token received. "
"Make sure your shell is handling the token appropriately."
)
logger.info(
"See docs for help: http://polyaxon.com/docs/polyaxon_cli/commands/auth"
)
return
access_auth = polyaxon_sdk.models.V1Auth(token=token.strip(" "))
# Set user
try:
AuthConfigManager.purge()
polyaxon_client = PolyaxonClient(token=access_auth.token)
user = polyaxon_client.users_v1.get_user()
except (ApiException, HTTPError) as e:
handle_cli_error(e, message="Could not load user info.")
sys.exit(1)
access_token = AccessTokenConfig(username=user.username, token=access_auth.token)
AuthConfigManager.set_config(access_token)
polyaxon_client.config.token = access_auth.token
Printer.print_success("Login successful")
# Reset current cli
server_versions = get_server_versions(polyaxon_client=polyaxon_client)
current_version = get_current_version()
log_handler = get_log_handler(polyaxon_client=polyaxon_client)
CliConfigManager.reset(
check_count=0,
current_version=current_version,
server_versions=server_versions.to_dict(),
log_handler=log_handler,
)
def logout():
"""Logout of Polyaxon."""
AuthConfigManager.purge()
CliConfigManager.purge()
Printer.print_success("You are logged out")
def logout():
"""Logout from Polyaxon Cloud or Polyaxon EE."""
AuthConfigManager.purge()
UserConfigManager.purge()
CliConfigManager.purge()
Printer.print_success("You are logged out")
def session_expired():
AuthConfigManager.purge()
CliConfigManager.purge()
click.echo("Session has expired, please try again.")
sys.exit(1)
def purge():
"""Purge the global config values."""
ClientConfigManager.purge()
Printer.print_success("Config was removed.")
# Reset cli config
CliConfigManager.purge()
