Skip to content

Commit e326307

Browse files
authored
Merge branch 'master' into is34/fix-catalog-service-inherit-group-1
2 parents 7d833a3 + 2098965 commit e326307

File tree

53 files changed

+1165
-348
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

53 files changed

+1165
-348
lines changed
File renamed without changes.

api/specs/web-server/_auth.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -153,17 +153,15 @@ async def logout(_body: LogoutBody):
153153

154154
@router.get(
155155
"/auth:check",
156-
operation_id="check_authentication",
157156
status_code=status.HTTP_204_NO_CONTENT,
158157
responses={
159158
status.HTTP_401_UNAUTHORIZED: {
160159
"model": EnvelopedError,
161-
"description": "unauthorized reset due to invalid token code",
162160
}
163161
},
164162
)
165163
async def check_auth():
166-
"""checks if user is authenticated in the platform"""
164+
"""checks whether user request is authenticated"""
167165

168166

169167
@router.post(

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

Whitespace-only changes.

services/storage/src/simcore_service_storage/api/rpc/_async_jobs.py renamed to packages/celery-library/src/celery_library/rpc/_async_jobs.py

Lines changed: 20 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -3,11 +3,6 @@
33
import logging
44

55
from celery.exceptions import CeleryError # type: ignore[import-untyped]
6-
from celery_library.errors import (
7-
TransferrableCeleryError,
8-
decode_celery_transferrable_error,
9-
)
10-
from fastapi import FastAPI
116
from models_library.api_schemas_rpc_async_jobs.async_jobs import (
127
AsyncJobGet,
138
AsyncJobId,
@@ -22,21 +17,27 @@
2217
JobSchedulerError,
2318
)
2419
from servicelib.celery.models import TaskState
20+
from servicelib.celery.task_manager import TaskManager
2521
from servicelib.logging_utils import log_catch
2622
from servicelib.rabbitmq import RPCRouter
2723

28-
from ...modules.celery import get_task_manager_from_app
24+
from ..errors import (
25+
TransferrableCeleryError,
26+
decode_celery_transferrable_error,
27+
)
2928

3029
_logger = logging.getLogger(__name__)
3130
router = RPCRouter()
3231

3332

3433
@router.expose(reraise_if_error_type=(JobSchedulerError,))
35-
async def cancel(app: FastAPI, job_id: AsyncJobId, job_id_data: AsyncJobNameData):
36-
assert app # nosec
34+
async def cancel(
35+
task_manager: TaskManager, job_id: AsyncJobId, job_id_data: AsyncJobNameData
36+
):
37+
assert task_manager # nosec
3738
assert job_id_data # nosec
3839
try:
39-
await get_task_manager_from_app(app).cancel_task(
40+
await task_manager.cancel_task(
4041
task_context=job_id_data.model_dump(),
4142
task_uuid=job_id,
4243
)
@@ -46,13 +47,13 @@ async def cancel(app: FastAPI, job_id: AsyncJobId, job_id_data: AsyncJobNameData
4647

4748
@router.expose(reraise_if_error_type=(JobSchedulerError,))
4849
async def status(
49-
app: FastAPI, job_id: AsyncJobId, job_id_data: AsyncJobNameData
50+
task_manager: TaskManager, job_id: AsyncJobId, job_id_data: AsyncJobNameData
5051
) -> AsyncJobStatus:
51-
assert app # nosec
52+
assert task_manager # nosec
5253
assert job_id_data # nosec
5354

5455
try:
55-
task_status = await get_task_manager_from_app(app).get_task_status(
56+
task_status = await task_manager.get_task_status(
5657
task_context=job_id_data.model_dump(),
5758
task_uuid=job_id,
5859
)
@@ -75,20 +76,20 @@ async def status(
7576
)
7677
)
7778
async def result(
78-
app: FastAPI, job_id: AsyncJobId, job_id_data: AsyncJobNameData
79+
task_manager: TaskManager, job_id: AsyncJobId, job_id_data: AsyncJobNameData
7980
) -> AsyncJobResult:
80-
assert app # nosec
81+
assert task_manager # nosec
8182
assert job_id # nosec
8283
assert job_id_data # nosec
8384

8485
try:
85-
_status = await get_task_manager_from_app(app).get_task_status(
86+
_status = await task_manager.get_task_status(
8687
task_context=job_id_data.model_dump(),
8788
task_uuid=job_id,
8889
)
8990
if not _status.is_done:
9091
raise JobNotDoneError(job_id=job_id)
91-
_result = await get_task_manager_from_app(app).get_task_result(
92+
_result = await task_manager.get_task_result(
9293
task_context=job_id_data.model_dump(),
9394
task_uuid=job_id,
9495
)
@@ -122,12 +123,12 @@ async def result(
122123

123124
@router.expose(reraise_if_error_type=(JobSchedulerError,))
124125
async def list_jobs(
125-
app: FastAPI, filter_: str, job_id_data: AsyncJobNameData
126+
task_manager: TaskManager, filter_: str, job_id_data: AsyncJobNameData
126127
) -> list[AsyncJobGet]:
127128
_ = filter_
128-
assert app # nosec
129+
assert task_manager # nosec
129130
try:
130-
tasks = await get_task_manager_from_app(app).list_tasks(
131+
tasks = await task_manager.list_tasks(
131132
task_context=job_id_data.model_dump(),
132133
)
133134
except CeleryError as exc:

packages/celery-library/tests/conftest.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
from celery_library.common import create_task_manager
1616
from celery_library.signals import on_worker_init, on_worker_shutdown
1717
from celery_library.task_manager import CeleryTaskManager
18+
from celery_library.types import register_celery_types
1819
from pytest_simcore.helpers.monkeypatch_envs import setenvs_from_dict
1920
from pytest_simcore.helpers.typing_env import EnvVarsDict
2021
from servicelib.celery.app_server import BaseAppServer
@@ -25,6 +26,7 @@
2526
"pytest_simcore.docker_compose",
2627
"pytest_simcore.docker_swarm",
2728
"pytest_simcore.environment_configs",
29+
"pytest_simcore.rabbit_service",
2830
"pytest_simcore.redis_service",
2931
"pytest_simcore.repository_paths",
3032
]
@@ -123,6 +125,8 @@ async def celery_task_manager(
123125
celery_settings: CelerySettings,
124126
with_celery_worker: TestWorkController,
125127
) -> CeleryTaskManager:
128+
register_celery_types()
129+
126130
return await create_task_manager(
127131
celery_app,
128132
celery_settings,

0 commit comments

Comments
 (0)