diff --git a/longtasks/longtasks.py b/longtasks/longtasks.py index e804ddb..8537152 100644 --- a/longtasks/longtasks.py +++ b/longtasks/longtasks.py @@ -220,17 +220,16 @@ class LongTasks: async def get_redis_task(self, taskid): task = await self.redis.hgetall(f'{self.taskname}:task:{taskid}') - task = DictObject(**task) - if task.created_at: - task.created_at = float(task.created_at) - if task.started_at: - task.started_at = float(task.started_at) - if task.finished_at: - task.finished_at = float(task.finished_at) - if task.payload: - task.payload = json.loads(task.payload) - if task.status == 'SUCCEEDED' and task.result: - task.result = json.loads(task.result) + if task['created_at']: + task['created_at'] = float(task['created_at']) + if task['started_at']: + task['started_at'] = float(task['started_at']) + if task['finished_at']: + task['finished_at'] = float(task['finished_at']) + if task['payload']: + task['payload'] = json.loads(task['payload']) + if task['status'] == 'SUCCEEDED' and task['result']: + task['result'] = json.loads(task['result']) return task async def delete_redis_task(self, taskid):