panda-resub-taskid

#!/usr/bin/env python

"""
Resubmit jobs with pbook

Runs on pure python beauty.

"""
_h_jobid = 'panda taskid, you can also pipe them'

epi='Thanks ATLAS. Thatlas.'

import sys, os
import argparse



def get_args():
    parser = argparse.ArgumentParser(
        description=__doc__,
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog=epi)
    parser.add_argument('taskids', type=int, nargs='*', help=_h_jobid)
    parser.add_argument('-b', '--build', action='store_true')
    parser.add_argument('-n', '--nfiles', type=int)
    args = parser.parse_args()
    if not args.taskids and sys.stdin.isatty():
        parser.print_usage()
        sys.exit('ERROR: need to pipe in a taskid')
    return args

def retry(jobs, args):
    #enforceEnter, verbose, restoreDB
    pbook = PBookCore(False, False, False)
    pbook.sync()
    retry_opts = {}
    if args.build:
        retry_opts['retryBuild'] = True
    if args.nfiles:
        retry_opts.setdefault('newOpts', {})['nFilesPerJob'] = args.nfiles
    for job in jobs:
        pbook.retry(job, **retry_opts)

def stdin_iter():
    for line in sys.stdin:
        for job in line.split():
            yield int(job)

if __name__ == '__main__':
    args = get_args()

    # this is a thing of beauty, who needs import when you have execfile?
    # --- begin beauty ---
    try:
        tmp, __name__ = __name__, 'pbook'
        execfile(os.path.join(os.environ["ATLAS_LOCAL_ROOT_BASE"],
                              "x86_64/PandaClient/current/bin/pbook") )
        __name__ = tmp
    except ImportError:
        print("Failed to load PandaClient, please set up locally")
        sys.exit(1)
    # --- end thing of beauty ---

    jobs = []
    if not args.taskids:
        jobs = stdin_iter()
    else:
        jobs = args.taskids

    retry(jobs, args)