def cli_mfa_to_qrcode():
""" Generates a QR code to import a token to other devices. """
parser = get_parser()
parser.add_argument("token_name",
help="Name of the token to use.").completer = \
ChoicesCompleter(list_mfa_tokens())
argcomplete.autocomplete(parser)
args = parser.parse_args()
mfa_to_qrcode(args.token_name)
def cli_mfa_code():
""" Generates a TOTP code using an MFA token. """
parser = get_parser()
parser.add_argument("token_name",
help="Name of the token to use.").completer = \
ChoicesCompleter(list_mfa_tokens())
argcomplete.autocomplete(parser)
args = parser.parse_args()
print(mfa_generate_code(args.token_name))
def cli_mfa_delete_token():
""" Deletes an MFA token file from the .ndt subdirectory in the user's
home directory """
parser = get_parser()
parser.add_argument("token_name",
help="Name of the token to delete.").completer = \
ChoicesCompleter(list_mfa_tokens())
argcomplete.autocomplete(parser)
args = parser.parse_args()
mfa_delete_token(args.token_name)
def session_to_env():
""" Export current session as environment variables """
parser = get_parser()
parser.add_argument("-t", "--token-name",
help="Name of the mfs token to use.").completer = \
ChoicesCompleter(list_mfa_tokens())
parser.add_argument("-d", "--duration-minutes", type=int, default=60,
help="Duration in minutes for the session token. Default to 60")
argcomplete.autocomplete(parser)
args = parser.parse_args()
call_args = {"duration_minutes": args.duration_minutes}
if args.token_name:
call_args["token_arn"] = mfa_read_token(args.token_name)["token_arn"]
call_args["token_value"] = mfa_generate_code(args.token_name)
creds = session_token(**call_args)
if creds:
print("AWS_ACCESS_KEY_ID=\"" + creds['AccessKeyId'] + "\"")
print("AWS_SECRET_ACCESS_KEY=\"" + creds['SecretAccessKey'] + "\"")
print("AWS_SESSION_TOKEN=\"" + creds['SessionToken'] + "\"")
print("AWS_SESSION_EXPIRATION=\"" + creds['Expiration'].strftime("%a, %d %b %Y %H:%M:%S +0000") + "\"")
print("export AWS_ACCESS_KEY_ID AWS_SECRET_ACCESS_KEY AWS_SESSION_TOKEN AWS_SESSION_EXPIRATION")