From 0804b1753ab94bf4511598f5b750ad99340aa32a Mon Sep 17 00:00:00 2001 From: Robin Ole Heinemann Date: Thu, 28 Dec 2023 17:45:51 +0100 Subject: [PATCH] worker: kill keepalive task on exceptions --- worker/transcribee_worker/worker.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/worker/transcribee_worker/worker.py b/worker/transcribee_worker/worker.py index 1ecefa67..4572b658 100644 --- a/worker/transcribee_worker/worker.py +++ b/worker/transcribee_worker/worker.py @@ -365,14 +365,16 @@ async def _work(): self.keepalive(task_id, self.progress) except Exception as exc: logging.error("Keepliave failed", exc_info=exc) - await asyncio.sleep(seconds) + finally: + await asyncio.sleep(seconds) task = asyncio.create_task(_work()) - yield - - stop_event.set() - await task + try: + yield + finally: + stop_event.set() + await task async def run_task(self, mark_completed=True): task = self.claim_task()