diff options
Diffstat (limited to 'scripts/testdb.py')
-rw-r--r-- | scripts/testdb.py | 18 |
1 files changed, 8 insertions, 10 deletions
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): |