function to switch aws profiles that supports 2fa

gistfile1.txt

aws-profile ()
{
  case "$1" in
        stage)
                aws --profile ops sts assume-role --role-arn "{role_arn_here}" --role-session-name "my_aws_session" > /tmp/assume-role-output.txt
        ;;
        dev)
                aws --profile dev sts assume-role --role-arn "{role_arn_here}" --role-session-name "my_aws_session" > /tmp/assume-role-output.txt
        ;;
        prod)
                aws --profile prod sts assume-role --role-arn "{role_arn_here}" --role-session-name "my_aws_session" > /tmp/assume-role-output.txt
        ;;
        *)
                echo "Usage: specify profile i.e. dev, stage, prod"
  esac
  export AWS_ACCESS_KEY_ID=`cat /tmp/assume-role-output.txt | jq -c '.Credentials.AccessKeyId' | tr -d '"' | tr -d ' '`
  export AWS_SECRET_ACCESS_KEY=`cat /tmp/assume-role-output.txt | jq -c '.Credentials.SecretAccessKey' | tr -d '"' | tr -d ' '`
  export AWS_SESSION_TOKEN=`cat /tmp/assume-role-output.txt | jq -c '.Credentials.SessionToken' | tr -d '"' | tr -d ' '`
  export AWS_SECURITY_TOKEN=`cat /tmp/assume-role-output.txt | jq -c '.Credentials.SessionToken' | tr -d '"' | tr -d ' '`
}