From fcae82356d8b4d4c2fd3df81f68b09a130dc0ed4 Mon Sep 17 00:00:00 2001 From: EvaBardou <bardou@teklia.com> Date: Mon, 29 Jan 2024 18:03:36 +0100 Subject: [PATCH] Nits --- arkindex_worker/worker/transcription.py | 26 +++++++++++++++++-------- 1 file changed, 18 insertions(+), 8 deletions(-) diff --git a/arkindex_worker/worker/transcription.py b/arkindex_worker/worker/transcription.py index 6f53a126..80300cfc 100644 --- a/arkindex_worker/worker/transcription.py +++ b/arkindex_worker/worker/transcription.py @@ -450,17 +450,27 @@ class TranscriptionMixin: if worker_version is not None: # If worker_version=False, filter by manual worker_version e.g. None - worker_version_id = worker_version if worker_version else None - transcriptions = transcriptions.where( - CachedTranscription.worker_version_id == worker_version_id - ) + worker_version_id = worker_version or None + if worker_version_id: + transcriptions = transcriptions.where( + CachedTranscription.worker_version_id == worker_version_id + ) + else: + transcriptions = transcriptions.where( + CachedTranscription.worker_version_id.is_null() + ) if worker_run is not None: # If worker_run=False, filter by manual worker_run e.g. None - worker_run_id = worker_run if worker_run else None - transcriptions = transcriptions.where( - CachedTranscription.worker_run_id == worker_run_id - ) + worker_run_id = worker_run or None + if worker_run_id: + transcriptions = transcriptions.where( + CachedTranscription.worker_run_id == worker_run_id + ) + else: + transcriptions = transcriptions.where( + CachedTranscription.worker_run_id.is_null() + ) else: transcriptions = self.api_client.paginate( "ListTranscriptions", id=element.id, **query_params -- GitLab