From 33c94d64c130351474040bb5f91a4602ffab0018 Mon Sep 17 00:00:00 2001 From: Fred Boniface Date: Tue, 30 Apr 2024 20:58:23 +0100 Subject: [PATCH] Add speedyf/job_manager.py --- speedyf/job_manager.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 speedyf/job_manager.py diff --git a/speedyf/job_manager.py b/speedyf/job_manager.py new file mode 100644 index 0000000..43123cc --- /dev/null +++ b/speedyf/job_manager.py @@ -0,0 +1,18 @@ +class JobManager: + def __init__(self): + self.jobs = {} + + def create_job(self, job_id): + self.jobs[job_id] = {state: 'pending', fetch_url: None, error:None} + + def set_job_state(self, job_id, state): + self.jobs[job_id]['state'] = state + + def set_fetch_url(self, job_id, url): + self.jobs[job_id]['fetch_url'] = url + + def set_error(self, job_id, error): + self.jobs[job_id]['error'] = error + + def get_job_status(self, job_id): + return self.jobs.get(job_id, {}) \ No newline at end of file