Automate the AWS MFA code when running an AWS command

aws-mfa.py

# $ pip3 install pexpect
# $ pip3 install pyotp

import os
import sys

import pexpect
import pyotp

os.environ['AWS_DEFAULT_PROFILE'] = 'admin'  # Change this if you have a different profile


def main():
    print('Entering MFA...')
    enter_mfa()
    print('Done')


def generate_code() -> str:
    totp = pyotp.TOTP('#####')  # Your MFA secret
    return totp.now()


def enter_mfa():
    child = pexpect.spawn('aws iam get-account-summary')
    child.log_file_read = sys.stdout
    child.expect('Enter MFA code for')
    child.sendline(generate_code())


main()