Skip to content

Commit 5b67d46

Browse files
author
Andrei Neagu
committed
renamed run_id to service_run_id and RunID to ServiceRunID
1 parent b0d715c commit 5b67d46

File tree

46 files changed

+276
-248
lines changed

Some content is hidden

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

46 files changed

+276
-248
lines changed

packages/models-library/src/models_library/api_schemas_resource_usage_tracker/service_runs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,13 @@
88
from ..projects_nodes_io import NodeID
99
from ..resource_tracker import CreditTransactionStatus, ServiceRunStatus
1010
from ..services import ServiceKey, ServiceVersion
11-
from ..services_types import RunID
11+
from ..services_types import ServiceRunID
1212
from ..users import UserID
1313
from ..wallets import WalletID
1414

1515

1616
class ServiceRunGet(BaseModel):
17-
service_run_id: RunID
17+
service_run_id: ServiceRunID
1818
wallet_id: WalletID | None
1919
wallet_name: str | None
2020
user_id: UserID

packages/models-library/src/models_library/api_schemas_webserver/resource_usage.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
UnitExtraInfo,
1717
)
1818
from ..services import ServiceKey, ServiceVersion
19-
from ..services_types import RunID
19+
from ..services_types import ServiceRunID
2020
from ..users import UserID
2121
from ..wallets import WalletID
2222
from ._base import InputSchema, OutputSchema
@@ -27,7 +27,7 @@
2727
class ServiceRunGet(
2828
BaseModel
2929
): # NOTE: this is already in use so I didnt modidy inheritance from OutputSchema
30-
service_run_id: RunID
30+
service_run_id: ServiceRunID
3131
wallet_id: WalletID | None
3232
wallet_name: str | None
3333
user_id: UserID

packages/models-library/src/models_library/rabbitmq_messages.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from .projects_state import RunningState
1616
from .services import ServiceKey, ServiceType, ServiceVersion
1717
from .services_resources import ServiceResourcesDict
18-
from .services_types import RunID
18+
from .services_types import ServiceRunID
1919
from .users import UserID
2020
from .utils.enums import StrAutoEnum
2121
from .wallets import WalletID
@@ -179,7 +179,7 @@ class RabbitResourceTrackingMessageType(StrAutoEnum):
179179
class RabbitResourceTrackingBaseMessage(RabbitMessageBase):
180180
channel_name: Literal["io.simcore.service.tracking"] = "io.simcore.service.tracking"
181181

182-
service_run_id: RunID = Field(
182+
service_run_id: ServiceRunID = Field(
183183
..., description="uniquely identitifies the service run"
184184
)
185185
created_at: datetime.datetime = Field(

packages/models-library/src/models_library/services.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,9 @@
77
from .services_metadata_published import ServiceInputsDict, ServiceMetaDataPublished
88
from .services_types import (
99
DynamicServiceKey,
10-
RunID,
1110
ServiceKey,
1211
ServicePortKey,
12+
ServiceRunID,
1313
ServiceVersion,
1414
)
1515

@@ -21,14 +21,14 @@
2121
"BootOptions",
2222
"DynamicServiceKey",
2323
"LATEST_INTEGRATION_VERSION",
24-
"RunID",
2524
"ServiceInput",
2625
"ServiceInputsDict",
2726
"ServiceKey",
2827
"ServiceKeyVersion",
2928
"ServiceMetaDataPublished",
3029
"ServiceOutput",
3130
"ServicePortKey",
31+
"ServiceRunID",
3232
"ServiceType",
3333
"ServiceVersion",
3434
)

packages/models-library/src/models_library/services_types.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@
4545
ServiceVersion: TypeAlias = Annotated[str, StringConstraints(pattern=SIMPLE_VERSION_RE)]
4646

4747

48-
class RunID(str):
48+
class ServiceRunID(str):
4949
"""
5050
Used to assign a unique identifier to the run of a service.
5151
@@ -92,7 +92,7 @@ def __get_pydantic_core_schema__(
9292
return core_schema.no_info_after_validator_function(cls, handler(str))
9393

9494
@classmethod
95-
def validate(cls, v: "RunID | str", _: ValidationInfo) -> "RunID":
95+
def validate(cls, v: "ServiceRunID | str", _: ValidationInfo) -> "ServiceRunID":
9696
if isinstance(v, cls):
9797
return v
9898
if isinstance(v, str):

packages/models-library/tests/test_services_types.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import pytest
22
from models_library.projects import ProjectID
33
from models_library.projects_nodes import NodeID
4-
from models_library.services_types import RunID
4+
from models_library.services_types import ServiceRunID
55
from models_library.users import UserID
66
from pydantic import PositiveInt
77

@@ -25,12 +25,12 @@ def test_run_id_get_resource_tracking_run_id(
2525
iteration: PositiveInt,
2626
expected_result: str,
2727
):
28-
resource_tracking_run_id = RunID.get_resource_tracking_run_id(
28+
resource_tracking_service_run_id = ServiceRunID.get_resource_tracking_run_id(
2929
user_id, project_id, node_id, iteration
3030
)
31-
assert isinstance(resource_tracking_run_id, RunID)
32-
assert resource_tracking_run_id == expected_result
31+
assert isinstance(resource_tracking_service_run_id, ServiceRunID)
32+
assert resource_tracking_service_run_id == expected_result
3333

3434

3535
def test_run_id_create_for_dynamic_sidecar():
36-
assert isinstance(RunID.create_for_dynamic_sidecar(), RunID)
36+
assert isinstance(ServiceRunID.create_for_dynamic_sidecar(), ServiceRunID)

packages/service-library/src/servicelib/rabbitmq/rpc_interfaces/webserver/licenses/licensed_items.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from models_library.licensed_items import LicensedItemID
99
from models_library.products import ProductName
1010
from models_library.rabbitmq_basic_types import RPCMethodName
11-
from models_library.services_types import RunID
11+
from models_library.services_types import ServiceRunID
1212
from models_library.users import UserID
1313
from models_library.wallets import WalletID
1414
from pydantic import TypeAdapter
@@ -65,7 +65,7 @@ async def checkout_licensed_item_for_wallet(
6565
wallet_id: WalletID,
6666
licensed_item_id: LicensedItemID,
6767
num_of_seats: int,
68-
service_run_id: RunID,
68+
service_run_id: ServiceRunID,
6969
) -> None:
7070
result = await rabbitmq_rpc_client.request(
7171
WEBSERVER_RPC_NAMESPACE,
@@ -89,7 +89,7 @@ async def release_licensed_item_for_wallet(
8989
wallet_id: WalletID,
9090
licensed_item_id: LicensedItemID,
9191
num_of_seats: int,
92-
service_run_id: RunID,
92+
service_run_id: ServiceRunID,
9393
) -> None:
9494
result = await rabbitmq_rpc_client.request(
9595
WEBSERVER_RPC_NAMESPACE,

services/agent/src/simcore_service_agent/models/volumes.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,14 @@
66
)
77
from models_library.projects import ProjectID
88
from models_library.projects_nodes_io import NodeID
9-
from models_library.services_types import RunID
9+
from models_library.services_types import ServiceRunID
1010
from models_library.users import UserID
1111
from pydantic import BaseModel, ConfigDict, Field, TypeAdapter
1212

1313

1414
class DynamicServiceVolumeLabels(BaseModel):
1515
node_uuid: NodeID
16-
run_id: RunID
16+
run_id: ServiceRunID
1717
source: str
1818
study_id: ProjectID
1919
swarm_stack_name: str

services/agent/tests/unit/conftest.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
from fastapi.testclient import TestClient
1616
from models_library.projects import ProjectID
1717
from models_library.projects_nodes_io import NodeID
18-
from models_library.services_types import RunID
18+
from models_library.services_types import ServiceRunID
1919
from models_library.users import UserID
2020
from pytest_simcore.helpers.monkeypatch_envs import EnvVarsDict, setenvs_from_dict
2121
from settings_library.rabbit import RabbitSettings
@@ -56,8 +56,8 @@ def test_client(initialized_app: FastAPI) -> TestClient:
5656

5757

5858
@pytest.fixture
59-
def run_id() -> RunID:
60-
return RunID.create_for_dynamic_sidecar()
59+
def service_run_id() -> ServiceRunID:
60+
return ServiceRunID.create_for_dynamic_sidecar()
6161

6262

6363
@pytest.fixture
@@ -77,7 +77,7 @@ def volumes_path(tmp_path: Path) -> Path:
7777

7878
@pytest.fixture
7979
async def create_dynamic_sidecar_volume(
80-
run_id: RunID,
80+
service_run_id: ServiceRunID,
8181
project_id: ProjectID,
8282
swarm_stack_name: str,
8383
user_id: UserID,
@@ -89,13 +89,13 @@ async def create_dynamic_sidecar_volume(
8989
async with aiodocker.Docker() as docker_client:
9090

9191
async def _(node_id: NodeID, in_use: bool, volume_name: str) -> str:
92-
source = get_source(run_id, node_id, volumes_path / volume_name)
92+
source = get_source(service_run_id, node_id, volumes_path / volume_name)
9393
volume = await docker_client.volumes.create(
9494
{
9595
"Name": source,
9696
"Labels": {
9797
"node_uuid": f"{node_id}",
98-
"run_id": run_id,
98+
"run_id": service_run_id,
9999
"source": source,
100100
"study_id": f"{project_id}",
101101
"swarm_stack_name": swarm_stack_name,

services/agent/tests/unit/test_services_backup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from fastapi import FastAPI
1212
from models_library.projects import ProjectID
1313
from models_library.projects_nodes_io import NodeID
14-
from models_library.services_types import RunID
14+
from models_library.services_types import ServiceRunID
1515
from pydantic import NonNegativeInt
1616
from simcore_service_agent.core.settings import ApplicationSettings
1717
from simcore_service_agent.services.backup import backup_volume
@@ -48,7 +48,7 @@ async def test_backup_volume(
4848
volume_content: Path,
4949
project_id: ProjectID,
5050
swarm_stack_name: str,
51-
run_id: RunID,
51+
service_run_id: ServiceRunID,
5252
downlaoded_from_s3: Path,
5353
create_dynamic_sidecar_volumes: Callable[[NodeID, bool], Awaitable[set[str]]],
5454
initialized_app: FastAPI,
@@ -80,7 +80,7 @@ async def test_backup_volume(
8080
async with session.client("s3", endpoint_url=f"{settings.AGENT_VOLUMES_CLEANUP_S3_ENDPOINT}") as s3_client: # type: ignore
8181
list_response = await s3_client.list_objects_v2(
8282
Bucket=settings.AGENT_VOLUMES_CLEANUP_S3_BUCKET,
83-
Prefix=f"{swarm_stack_name}/{project_id}/{node_id}/{run_id}",
83+
Prefix=f"{swarm_stack_name}/{project_id}/{node_id}/{service_run_id}",
8484
)
8585
synced_keys: list[str] = [o["Key"] for o in list_response["Contents"]]
8686

0 commit comments

Comments
 (0)