From 53b1e7cb181380a24aab4cbc7a9634b2d8dd2d29 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Fri, 19 Jan 2024 08:29:04 +0300
Subject: scripts: Process R/qtl2 bundle

Build script to start the processing of the R/qtl2 bundle.
---
 qc_app/jobs.py | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

(limited to 'qc_app/jobs.py')

diff --git a/qc_app/jobs.py b/qc_app/jobs.py
index c5bf5e5..a8257a3 100644
--- a/qc_app/jobs.py
+++ b/qc_app/jobs.py
@@ -3,11 +3,19 @@ import os
 import sys
 import shlex
 import subprocess
-from uuid import uuid4
+from typing import Union
+from uuid import UUID, uuid4
 from datetime import timedelta
 
 from redis import Redis
 
+class JobNotFound(Exception):
+    """Raised if we try to retrieve a non-existent job."""
+
+def raise_jobnotfound(jobid: Union[str,UUID]):
+    """Utility to raise a `NoSuchJobError`"""
+    raise JobNotFound(f"Could not retrieve job '{jobid}'.")
+
 def error_filename(job_id, error_dir):
     "Compute the path of the file where errors will be dumped."
     return f"{error_dir}/job_{job_id}.error"
@@ -70,6 +78,7 @@ def launch_job(the_job: dict, redisurl: str, error_dir):
 
     return the_job
 
-def job(redis_conn, job_id: str):
+def job(redis_conn, job_id: Union[str,UUID]):
     "Retrieve the job"
-    return redis_conn.hgetall(job_id)
+    thejob = redis_conn.hgetall(str(job_id)) or raise_jobnotfound(job_id)
+    return thejob
-- 
cgit v1.2.3