From 168b7958219b4be34a1922fa27c92a50a1f9d8d2 Mon Sep 17 00:00:00 2001 From: Cristiano Urban <cristiano.urban@inaf.it> Date: Mon, 20 Sep 2021 13:44:31 +0200 Subject: [PATCH] Allow to change the time to wait directly. Signed-off-by: Cristiano Urban <cristiano.urban@inaf.it> --- transfer_service/task_executor.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/transfer_service/task_executor.py b/transfer_service/task_executor.py index 50afd05..f5cdb18 100644 --- a/transfer_service/task_executor.py +++ b/transfer_service/task_executor.py @@ -29,8 +29,11 @@ class TaskExecutor(Process): self.destQueue = None super(TaskExecutor, self).__init__() - def wait(self): - time.sleep(self.execWaitTime) + def wait(self, timeout = None): + if timeout is None: + time.sleep(self.execWaitTime) + else: + time.sleep(timeout) def setSourceQueueName(self, srcQueueName): self.srcQueue = JobQueue(srcQueueName) -- GitLab