SpeedyF/speedyf/job_manager.py

18 lines
515 B
Python

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, {})