diff --git a/netmon/entry.py b/netmon/entry.py index 7dbd11d..870f7d5 100644 --- a/netmon/entry.py +++ b/netmon/entry.py @@ -36,6 +36,7 @@ def record(target: str, interval: int, outage_interval: int): "INSERT INTO errors (from_time, to_time, host) VALUES (?, ?, ?);", (int(outage), int(time.time()), target), ) + outage = None time.sleep(interval if outage is None else outage_interval) except KeyboardInterrupt as exn: if outage is not None: @@ -50,11 +51,14 @@ def record(target: str, interval: int, outage_interval: int): def entry(): parser = argparse.ArgumentParser() parser.add_argument("host", help="Remote host to ping (IP address preferred)") - parser.add_argument("-i", "--interval", default=30, help="Time between two pings") + parser.add_argument( + "-i", "--interval", default=30, type=int, help="Time between two pings" + ) parser.add_argument( "-I", "--outage-interval", default=2, + type=int, help="Time between two pings while the network is down", ) args = parser.parse_args()