tps: add do_repeat to cli

994a633b · Juan David González Cobas · 24704ba9 · 994a633b
Commit 994a633b authored May 20, 2011 by Juan David González Cobas
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 1 deletion

tps.py tps.py +8 -1

No files found.
--- a/tps.py
+++ b/tps.py
@@ -211,7 +211,14 @@ class Cli(cmd.Cmd, Suite):
    def do_run(self, arg):
        pass
    def do_repeat(self, arg):
-        pass
+        if arg:
+            try:
+                self.repeat = int(arg)
+            except ValueError:
+                print arg, 'is not an integer'
+        else:
+            print self.repeat
+

    def do_EOF(self, arg):
        print