class ConfigWriter(object): def __init__(self, session): self.session = session self.section = _get_profile_str(session, ' ') self.config_file_writer = ConfigFileWriter() def update_config(self, key, value): config_filename = \ os.path.expanduser(self.session.get_config_variable('config_file')) updated_config = {'__section__': self.section, 'emr': {key: value}} self.config_file_writer.update_config(updated_config, config_filename)
def _run_main(self, parsed_args, parsed_globals): current_key = None current_secret = None masked_current_secret = None if self._session._credentials: current_key = self._session._credentials.access_key current_secret = self._session._credentials.secret_key if current_secret is not None: masked_current_secret = "*" * (len(current_secret) - 4) + current_secret[-4:] key = parsed_args.key if key is None: import getpass key = getpass.getpass("AWS Access Key ID [%s]: " % current_key) if key is None or key == "": key = current_key secret = parsed_args.secret if secret is None: import getpass secret = getpass.getpass("AWS Secret Access Key [%s]: " % masked_current_secret) if secret is None or secret == "": secret = current_secret profile = self._session.profile if profile is None: profile = "default" config_section = "default" else: config_section = "profile {0}".format(profile) persistence.set_credentials(profile, key, secret) config_update = {"__section__": config_section, "keyring": "true"} config_filename = os.path.expanduser( self._session.get_config_variable("config_file")) config_writer = ConfigFileWriter() config_writer.update_config(config_update, config_filename) return 0
def _run_main(self, parsed_args, parsed_globals): current_key = None current_secret = None masked_current_secret = None if self._session._credentials: current_key = self._session._credentials.access_key current_secret = self._session._credentials.secret_key if current_secret is not None: masked_current_secret = "*" * (len(current_secret) - 4) + current_secret[-4:] key = parsed_args.key if key is None: import getpass key = getpass.getpass("AWS Access Key ID [%s]: " % current_key) if key is None or key == "": key = current_key secret = parsed_args.secret if secret is None: import getpass secret = getpass.getpass("AWS Secret Access Key [%s]: " % masked_current_secret) if secret is None or secret == "": secret = current_secret profile = self._session.profile if profile is None: profile = "default" config_section = "default" else: config_section = "profile {0}".format(profile) persistence.set_credentials(profile, key, secret) config_update = {"__section__": config_section, "keyring": "true"} config_filename = os.path.expanduser(self._session.get_config_variable("config_file")) config_writer = ConfigFileWriter() config_writer.update_config(config_update, config_filename) return 0
def __init__(self, session): self.session = session self.section = _get_profile_str(session, ' ') self.config_file_writer = ConfigFileWriter()