3535class CeleryTaskClient :
3636 _celery_app : Celery
3737 _celery_settings : CelerySettings
38- _task_store : TaskInfoStore
38+ _task_info_store : TaskInfoStore
3939
4040 async def submit_task (
4141 self ,
@@ -63,7 +63,9 @@ async def submit_task(
6363 if task_metadata .ephemeral
6464 else self ._celery_settings .CELERY_RESULT_EXPIRES
6565 )
66- await self ._task_store .create_task (task_id , task_metadata , expiry = expiry )
66+ await self ._task_info_store .create_task (
67+ task_id , task_metadata , expiry = expiry
68+ )
6769 return task_uuid
6870
6971 @make_async ()
@@ -96,9 +98,9 @@ async def get_task_result(
9698 async_result = self ._celery_app .AsyncResult (task_id )
9799 result = async_result .result
98100 if async_result .ready ():
99- task_metadata = await self ._task_store .get_task_metadata (task_id )
101+ task_metadata = await self ._task_info_store .get_task_metadata (task_id )
100102 if task_metadata is not None and task_metadata .ephemeral :
101- await self ._task_store .remove_task (task_id )
103+ await self ._task_info_store .remove_task (task_id )
102104 await self ._forget_task (task_id )
103105 return result
104106
@@ -107,7 +109,7 @@ async def _get_task_progress_report(
107109 ) -> ProgressReport :
108110 if task_state in (TaskState .STARTED , TaskState .RETRY , TaskState .ABORTED ):
109111 task_id = build_task_id (task_context , task_uuid )
110- progress = await self ._task_store .get_task_progress (task_id )
112+ progress = await self ._task_info_store .get_task_progress (task_id )
111113 if progress is not None :
112114 return progress
113115 if task_state in (
@@ -153,4 +155,4 @@ async def list_tasks(self, task_context: TaskContext) -> list[Task]:
153155 logging .DEBUG ,
154156 msg = f"Listing tasks: { task_context = } " ,
155157 ):
156- return await self ._task_store .list_tasks (task_context )
158+ return await self ._task_info_store .list_tasks (task_context )
0 commit comments