def __init__(self, fetch=FETCH_PORT, status=STATUS_PORT, redisp=REDIS_PORT): super(TaskQueue, self).__init__() self.__redis_connection = redis.Redis(host='redis', port=redisp, db=0) self.__util = TaskUtil()
class Task(Resource): """ Show task status """ def __init__(self, redisp=6379): super(Task, self).__init__() self.__redis_connection = redis.Redis(host='redis', port=redisp, db=0) self.__util = TaskUtil() def get(self, job_id): """ GET status of the task. :return: response """ status = self.__redis_connection.hget(name='status', key=job_id) logging.debug('Got status: ' + str(status)) result = { TAG_JOB_ID: job_id, TAG_STATUS: status } return result, 200 def delete(self, job_id): """Delete both job and its result. :return: Success message """ return self.__util.delete_job(job_id), 200
class TaskQueue(Resource): """API for job management. """ def __init__(self, fetch=FETCH_PORT, status=STATUS_PORT, redisp=REDIS_PORT): super(TaskQueue, self).__init__() self.__redis_connection = redis.Redis(host='redis', port=redisp, db=0) self.__util = TaskUtil() def __get_status(self): status_hash = self.__redis_connection.hgetall(name=TAG_STATUS) serializable = [] keys = status_hash.keys() for key in keys: serializable.append( { TAG_JOB_ID: key.decode("utf-8"), TAG_STATUS: status_hash[key].decode("utf-8") } ) return serializable def get(self): status_message = self.__get_status() return status_message, 200 def delete(self): """Delete all jobs :return: """ status_hash = self.__redis_connection.hgetall(name=TAG_STATUS) keys = status_hash.keys() deleted_jobs = [] for key in keys: result = self.__util.delete_job(key) deleted_jobs.append(result[TAG_JOB_ID]) message = { 'deletedJobs': deleted_jobs } return message, 200
class TaskQueue(Resource): """API for job management. """ def __init__(self, fetch=FETCH_PORT, status=STATUS_PORT, redisp=REDIS_PORT): super(TaskQueue, self).__init__() self.__redis_connection = redis.Redis(host='redis', port=redisp, db=0) self.__util = TaskUtil() def __get_status(self): status_hash = self.__redis_connection.hgetall(name=TAG_STATUS) serializable = [] keys = status_hash.keys() for key in keys: serializable.append({ TAG_JOB_ID: key.decode("utf-8"), TAG_STATUS: status_hash[key].decode("utf-8") }) return serializable def get(self): status_message = self.__get_status() return status_message, 200 def delete(self): """Delete all jobs :return: """ status_hash = self.__redis_connection.hgetall(name=TAG_STATUS) keys = status_hash.keys() deleted_jobs = [] for key in keys: result = self.__util.delete_job(key) deleted_jobs.append(result[TAG_JOB_ID]) message = {'deletedJobs': deleted_jobs} return message, 200
class Task(Resource): """ Show task status """ def __init__(self, redisp=6379): super(Task, self).__init__() self.__redis_connection = redis.Redis(host='redis', port=redisp, db=0) self.__util = TaskUtil() def get(self, job_id): """ GET status of the task. :return: response """ status = self.__redis_connection.hget(name='status', key=job_id) logging.debug('Got status: ' + str(status)) result = {TAG_JOB_ID: job_id, TAG_STATUS: status} return result, 200 def delete(self, job_id): """Delete both job and its result. :return: Success message """ return self.__util.delete_job(job_id), 200
def __init__(self, redisp=6379): super(Task, self).__init__() self.__redis_connection = redis.Redis(host='redis', port=redisp, db=0) self.__util = TaskUtil()