18 lines
515 B
Python
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, {})
|