Skip to content

Commit 1fbd307

Browse files
committed
cleanup celery rpc interface
1 parent c4c4fe8 commit 1fbd307

File tree

1 file changed

+8
-6
lines changed

1 file changed

+8
-6
lines changed

packages/celery-library/src/celery_library/rpc/_async_jobs.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
JobNotDoneError,
1717
JobSchedulerError,
1818
)
19-
from servicelib.celery.models import TaskState
19+
from servicelib.celery.models import TaskFilter, TaskState
2020
from servicelib.celery.task_manager import TaskManager
2121
from servicelib.logging_utils import log_catch
2222
from 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

Comments
 (0)