From fc4f9a1c7a0734a190b99265ca50d72014173859 Mon Sep 17 00:00:00 2001 From: Aldo Cortesi Date: Sat, 25 Oct 2014 17:58:59 +1300 Subject: pathoc -n 0 repeats forever --- libpathod/cmdline.py | 2 +- libpathod/pathoc.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) (limited to 'libpathod') diff --git a/libpathod/cmdline.py b/libpathod/cmdline.py index a21fa218..f80b7aae 100644 --- a/libpathod/cmdline.py +++ b/libpathod/cmdline.py @@ -36,7 +36,7 @@ def go_pathoc(): ) parser.add_argument( "-n", dest='repeat', default=1, type=int, metavar="N", - help='Repeat N times' + help='Repeat N times. If 0 repeat for ever.' ) parser.add_argument( "-r", dest="random", action="store_true", default=False, diff --git a/libpathod/pathoc.py b/libpathod/pathoc.py index c85c207a..02d0c06d 100644 --- a/libpathod/pathoc.py +++ b/libpathod/pathoc.py @@ -224,7 +224,9 @@ class Pathoc(tcp.TCPClient): def main(args): try: - for i in range(args.repeat): + cnt = 0 + while 1: + cnt += 1 p = Pathoc( (args.host, args.port), ssl=args.ssl, @@ -258,5 +260,7 @@ def main(args): sys.stdout.flush() if ret and args.oneshot: sys.exit(0) + if cnt == args.repeat: + break except KeyboardInterrupt: pass -- cgit v1.2.3