generate_lhe_condor.py

#! /usr/bin/env python

from general_condor_functions import getbasic_parser
from general_condor_functions import create_output_directory
from general_condor_functions import create_jdl_file_for_condor
from general_condor_functions import create_sh_file_for_condor

def getargs():
    parser = getbasic_parser()
    parser.add_argument('-cpu', '--ncpu',
                        default=8,
                        help='number of cpu to run'
                       )
    parser.add_argument('-r', '--randomnumber',
                        default=11,
                        help='random seed'
                       )
    return parser.parse_args()


def main():
    # get input arguments
    args = getargs()

    # list of input files to be added in jdl file
    inputlist = args.jdlfilename+".sh, "+args.InProcCardPath+"/"+args.tarfile

    # command to run
    command = './bin/mg5_aMC '+args.tarfile

    # Get Output directory name
    output_folder, output_log_path = create_output_directory(args)

    # create the jdl file for condor
    create_jdl_file_for_condor(args, inputlist, output_log_path)

    # create the sh file for condor
    create_sh_file_for_condor(args, command, output_folder)

    # running help
    print "===> Set Proxy Using:"
    print "\tvoms-proxy-init --voms cms --valid 168:00"
    print "\"condor_submit "+args.jdlfilename+".jdl\" to submit"


if __name__ == "__main__":
    main()