|
18 | 18 | from servicelib.fastapi.dependencies import get_app |
19 | 19 | from servicelib.logging_errors import create_troubleshootting_log_kwargs |
20 | 20 |
|
| 21 | +from ...clients.celery_task_manager import get_task_filter |
21 | 22 | from ...models.schemas.base import ApiServerEnvelope |
22 | 23 | from ...models.schemas.errors import ErrorGet |
23 | 24 | from ..dependencies.authentication import get_current_user_id, get_product_name |
@@ -60,7 +61,7 @@ async def list_tasks( |
60 | 61 | task_manager = get_task_manager(app) |
61 | 62 |
|
62 | 63 | tasks = await task_manager.list_tasks( |
63 | | - task_filter=_get_task_filter(user_id, product_name), |
| 64 | + task_filter=get_task_filter(user_id, product_name), |
64 | 65 | ) |
65 | 66 |
|
66 | 67 | app_router = app.router |
@@ -102,7 +103,7 @@ async def get_task_status( |
102 | 103 | task_manager = get_task_manager(app) |
103 | 104 |
|
104 | 105 | task_status = await task_manager.get_task_status( |
105 | | - task_filter=_get_task_filter(user_id, product_name), |
| 106 | + task_filter=get_task_filter(user_id, product_name), |
106 | 107 | task_uuid=TaskUUID(f"{task_id}"), |
107 | 108 | ) |
108 | 109 |
|
@@ -137,7 +138,7 @@ async def cancel_task( |
137 | 138 | task_manager = get_task_manager(app) |
138 | 139 |
|
139 | 140 | await task_manager.cancel_task( |
140 | | - task_filter=_get_task_filter(user_id, product_name), |
| 141 | + task_filter=get_task_filter(user_id, product_name), |
141 | 142 | task_uuid=TaskUUID(f"{task_id}"), |
142 | 143 | ) |
143 | 144 |
|
@@ -171,7 +172,7 @@ async def get_task_result( |
171 | 172 | product_name: Annotated[ProductName, Depends(get_product_name)], |
172 | 173 | ): |
173 | 174 | task_manager = get_task_manager(app) |
174 | | - task_filter = _get_task_filter(user_id, product_name) |
| 175 | + task_filter = get_task_filter(user_id, product_name) |
175 | 176 |
|
176 | 177 | task_status = await task_manager.get_task_status( |
177 | 178 | task_filter=task_filter, |
|
0 commit comments