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)
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()
def __init__(self, session):
self.session = session
self.section = _get_profile_str(session, ' ')
self.config_file_writer = ConfigFileWriter()
