#!/usr/bin/env python3
import argparse
import subprocess
import time
import sys


def get_args() -> argparse.ArgumentParser:
    parser = argparse.ArgumentParser()
    parser.add_argument("--delay", type=float, default=3,
                        help="Initial delay before retry, in seconds")
    parser.add_argument("--count", type=int, default=5, help="Total number of tries")
    parser.add_argument("--factor", 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()