From 47ad8a1ae2427c228c344157ff65f3d0566a9287 Mon Sep 17 00:00:00 2001 From: Rob Austein Date: Sun, 23 Dec 2007 19:55:29 +0000 Subject: sleep command svn path=/scripts/testdb.py; revision=1430 --- scripts/testdb.py | 18 ++++++++---------- 1 file changed, 8 insertions(+), 10 deletions(-) (limited to 'scripts/testdb.py') diff --git a/scripts/testdb.py b/scripts/testdb.py index e60311a6..fee0339e 100644 --- a/scripts/testdb.py +++ b/scripts/testdb.py @@ -166,19 +166,17 @@ def wakeup(signum, frame): signal.signal(signal.SIGALRM, wakeup) -def cmd_pause(): - """Do nothing until a signal arrives.""" - rpki.log.info("Pausing indefinitely, send a SIGALRM to wake me up") - signal.pause() - -def cmd_sleep(seconds): +def cmd_sleep(seconds = None): """Set an alarm, then wait for it to go off.""" - rpki.log.info("Sleeping %s seconds" % seconds) - signal.alarm(int(seconds)) + + if seconds is None: + rpki.log.info("Pausing indefinitely, send a SIGALRM to wake me up") + else: + rpki.log.info("Sleeping %s seconds" % seconds) + signal.alarm(int(seconds)) signal.pause() -cmds = { "pause" : cmd_pause, - "sleep" : cmd_sleep } +cmds = { "sleep" : cmd_sleep } class allocation_db(list): -- cgit v1.2.3