#!/usr/bin/env python

from benchmark_utils import summarize, clean

from benchmark_utils import get_default_argparser, get_transfer_command

from benchmark_utils import create_random_subfolder, benchmark_command

def benchmark_cp(args):

destination = args.destination

if args.recursive:

destination = create_random_subfolder(destination)

command = 'cp %s %s' % (args.source, destination)

command = get_transfer_command(command, args.recursive, args.quiet)

def cleanup():

if not args.no_cleanup:

clean(destination, args.recursive)

benchmark_command(

command, args.benchmark_script, args.summarize_script,

args.result_dir, args.num_iterations, args.dry_run,

cleanup=cleanup

)

if __name__ == "__main__":

parser = get_default_argparser()

parser.add_argument(

'-s', '--source', required=True,

help='A local path or s3 path.'

)

parser.add_argument(

'-d', '--destination', required=True,

help='A local path or s3 path. A directory will be created in this '

'location to copy to in the case of a recursive transfer.'

)

benchmark_cp(parser.parse_args())