retry.py - mozsearch

Enable keyboard shortcuts

#!/usr/bin/env python3

import argparse

import subprocess

import time

import sys

def get_args() -> argparse.ArgumentParser:

    parser = argparse.ArgumentParser()

    parser.add_argument("--delay", action="store", type=float, default=3, help="Initial delay before retry, in seconds")

    parser.add_argument("--count", action="store", type=int, default=5, help="Total number of tries")

    parser.add_argument("--factor", action="store", type=float, default=2, help="Exponential backoff factor")

    parser.add_argument("cmd", nargs=argparse.REMAINDER)

    return parser

def log(value: str) -> None:

    print(value)

    sys.stdout.flush()

def main() -> None:

    args = get_args().parse_args()

    if not args.cmd:

        log("No command supplied")

        sys.exit(1)

    retcode = None

    for n in range(args.count):

        try:

            log("Running %s [try %d/%d]" % (" ".join(args.cmd), (n+1), args.count))

            subprocess.check_call(args.cmd)

        except subprocess.CalledProcessError as e:

            retcode = e.returncode

        else:

            log("Command succeeded")

            retcode = 0

            break

        if args.factor == 0:

            wait_time = (n+1) * args.delay

        else:

            wait_time = args.factor**n * args.delay

        if n < args.count - 1:

            log("Command failed, waiting %s seconds to retry" % wait_time)

            time.sleep(wait_time)

        else:

            log("Command failed, out of retries")

    sys.exit(retcode)

if __name__ == "__main__":

    main()