1616 JobNotDoneError ,
1717 JobSchedulerError ,
1818)
19- from servicelib .celery .models import TaskState
19+ from servicelib .celery .models import TaskFilter , TaskState
2020from servicelib .celery .task_manager import TaskManager
2121from servicelib .logging_utils import log_catch
2222from servicelib .rabbitmq import RPCRouter
@@ -38,7 +38,7 @@ async def cancel(
3838 assert job_filter # nosec
3939 try :
4040 await task_manager .cancel_task (
41- task_filter = job_filter ,
41+ task_filter = TaskFilter . model_validate ( job_filter . model_dump ()) ,
4242 task_uuid = job_id ,
4343 )
4444 except CeleryError as exc :
@@ -54,7 +54,7 @@ async def status(
5454
5555 try :
5656 task_status = await task_manager .get_task_status (
57- task_filter = job_filter ,
57+ task_filter = TaskFilter . model_validate ( job_filter . model_dump ()) ,
5858 task_uuid = job_id ,
5959 )
6060 except CeleryError as exc :
@@ -82,15 +82,17 @@ async def result(
8282 assert job_id # nosec
8383 assert job_filter # nosec
8484
85+ task_filter = TaskFilter .model_validate (job_filter .model_dump ())
86+
8587 try :
8688 _status = await task_manager .get_task_status (
87- task_filter = job_filter ,
89+ task_filter = task_filter ,
8890 task_uuid = job_id ,
8991 )
9092 if not _status .is_done :
9193 raise JobNotDoneError (job_id = job_id )
9294 _result = await task_manager .get_task_result (
93- task_filter = job_filter ,
95+ task_filter = task_filter ,
9496 task_uuid = job_id ,
9597 )
9698 except CeleryError as exc :
@@ -129,7 +131,7 @@ async def list_jobs(
129131 assert task_manager # nosec
130132 try :
131133 tasks = await task_manager .list_tasks (
132- task_filter = job_filter ,
134+ task_filter = TaskFilter . model_validate ( job_filter . model_dump ()) ,
133135 )
134136 except CeleryError as exc :
135137 raise JobSchedulerError (exc = f"{ exc } " ) from exc
0 commit comments