def wait_for()

in aws/util.py [0:0]

• 20 lines of code
• 4 McCabe index (conditional complexity)

def wait_for(msg, timeout=300, interval=1, print_spinner=True):
    """
    for _ in wait_for("asg to provision", timeout_sec, interval_sec):
        if check_condition():.
            break
    """
    spin = ["-", "/", "|", "\\", "-", "/", "|", "\\"]
    idx = 0
    start = time.time()
    max_time = start + timeout
    while True:
        if print_spinner:
            elapsed = time.time() - start
            print(
                f"Waiting for {msg}"
                f" ({elapsed:03.0f}/{timeout:3.0f}s elapsed) {spin[idx]}\r",
                end="",
            )
            sys.stdout.flush()
            idx = (idx + 1) % len(spin)

        if time.time() >= max_time:
            raise RuntimeError(f"Timed out while waiting for: {msg}")
        else:
            time.sleep(interval)
            yield