def _configure_cli_token(profile, insecure): config = ProfileConfigProvider( profile).get_config() or DatabricksConfig.empty() host = click.prompt(PROMPT_HOST, default=config.host, type=_DbfsHost()) token = click.prompt(PROMPT_TOKEN, default=config.token) new_config = DatabricksConfig.from_token(host, token, insecure) update_and_persist_config(profile, new_config)
def _configure_cli_aad_token(profile, insecure, host, jobs_api_version): config = ProfileConfigProvider( profile).get_config() or DatabricksConfig.empty() if ENV_AAD_TOKEN not in os.environ: click.echo('[ERROR] Set Environment Variable \'%s\' with your ' 'AAD Token and run again.\n' % ENV_AAD_TOKEN) click.echo( 'Commands to run to get your AAD token:\n' '\t az login\n' '\t token_response=$(az account get-access-token ' '--resource 2ff814a6-3304-4ab8-85cb-cd0e6f879c1d)\n' '\t export %s=$(jq .accessToken -r <<< "$token_response")\n' % ENV_AAD_TOKEN) return if not host: host = click.prompt(PROMPT_HOST, default=config.host, type=_DbfsHost()) aad_token = os.environ.get(ENV_AAD_TOKEN) new_config = DatabricksConfig.from_token(host=host, token=aad_token, refresh_token=None, insecure=insecure, jobs_api_version=jobs_api_version) update_and_persist_config(profile, new_config)
def configure(version): profile = get_profile_from_context() config = ProfileConfigProvider( profile).get_config() if profile else get_config() new_config = config or DatabricksConfig.empty() new_config.jobs_api_version = version update_and_persist_config(profile, new_config)
def _configure_cli_token(profile, insecure): PROMPT_HOST = "Databricks Host (should begin with https://)" PROMPT_TOKEN = "Token" # NOQA config = ProfileConfigProvider( profile).get_config() or DatabricksConfig.empty() host = click.prompt(PROMPT_HOST, default=config.host, type=_DbfsHost()) token = click.prompt(PROMPT_TOKEN, default=config.token, hide_input=True) new_config = DatabricksConfig.from_token(host, token, insecure) update_and_persist_config(profile, new_config)
def _configure_cli_token(profile, insecure, host, jobs_api_version): config = ProfileConfigProvider( profile).get_config() or DatabricksConfig.empty() if not host: host = click.prompt(PROMPT_HOST, default=config.host, type=_DbfsHost()) token = click.prompt(PROMPT_TOKEN, default=config.token, hide_input=True) new_config = DatabricksConfig.from_token(host, token, insecure, jobs_api_version) update_and_persist_config(profile, new_config)
def _configure_cli_token_file(profile, token_file, host, insecure): if not path.exists(token_file): raise RuntimeError('Unable to read token from "{}"'.format(token_file)) with io.open(token_file, encoding='utf-8') as f: token = f.readline().strip() config = ProfileConfigProvider(profile).get_config() or DatabricksConfig.empty() if not host: host = click.prompt(PROMPT_HOST, default=config.host, type=_DbfsHost()) new_config = DatabricksConfig.from_token(host, token, insecure) update_and_persist_config(profile, new_config)
def _configure_cli_password(profile, insecure, host): config = ProfileConfigProvider(profile).get_config() or DatabricksConfig.empty() if config.password: default_password = '******' * len(config.password) else: default_password = None if not host: host = click.prompt(PROMPT_HOST, default=config.host, type=_DbfsHost()) username = click.prompt(PROMPT_USERNAME, default=config.username) password = click.prompt(PROMPT_PASSWORD, default=default_password, hide_input=True, confirmation_prompt=True) if password == default_password: password = config.password new_config = DatabricksConfig.from_password(host, username, password, insecure) update_and_persist_config(profile, new_config)
def _configure_cli_oauth(profile, insecure, host, scope, jobs_api_version): config = ProfileConfigProvider( profile).get_config() or DatabricksConfig.empty() if not host: host = click.prompt(PROMPT_HOST, default=config.host, type=_DbfsHost()) if not scope: scope = click.prompt( "Pick one or more comma-separated scopes from {scopes}".format( scopes=DEFAULT_SCOPES), value_proc=scope_format, type=click.Choice(DEFAULT_SCOPES), default=None) access_token, refresh_token = get_tokens(host, scope) new_config = DatabricksConfig.from_token(host=host, token=access_token, refresh_token=refresh_token, insecure=insecure, jobs_api_version=jobs_api_version) update_and_persist_config(profile, new_config)