Skip to content

Commit 6338615

Browse files
committed
cleanup
1 parent aac0db2 commit 6338615

File tree

5 files changed

+20
-7
lines changed

5 files changed

+20
-7
lines changed
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
from pydantic import BaseModel
2+
3+
4+
class CatalogInputSchema(BaseModel):
5+
...
6+
7+
8+
class CatalogOutputSchema(BaseModel):
9+
...

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

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from typing import Any, TypeAlias
33

44
from models_library.rpc_pagination import PageRpc
5-
from pydantic import BaseModel, ConfigDict, Field, HttpUrl, NonNegativeInt
5+
from pydantic import ConfigDict, Field, HttpUrl, NonNegativeInt
66
from pydantic.config import JsonDict
77

88
from ..boot_options import BootOptions
@@ -21,6 +21,7 @@
2121
from ..services_resources import ServiceResourcesDict
2222
from ..services_types import ServiceKey, ServiceVersion
2323
from ..utils.change_case import snake_to_camel
24+
from ._base import CatalogInputSchema, CatalogOutputSchema
2425

2526
_EXAMPLE_FILEPICKER: dict[str, Any] = {
2627
"name": "File Picker",
@@ -168,7 +169,8 @@ def _update_json_schema_extra(schema: JsonDict) -> None:
168169
)
169170

170171

171-
class ServiceGetV2(BaseModel):
172+
class ServiceGetV2(CatalogOutputSchema):
173+
# Model used in catalog's rpc and rest interfaces
172174
key: ServiceKey
173175
version: ServiceVersion
174176

@@ -282,7 +284,7 @@ def _update_json_schema_extra(schema: JsonDict) -> None:
282284
ServiceResourcesGet: TypeAlias = ServiceResourcesDict
283285

284286

285-
class ServiceUpdateV2(BaseModel):
287+
class ServiceUpdateV2(CatalogInputSchema):
286288
name: str | None = None
287289
thumbnail: HttpUrl | None = None
288290
icon: HttpUrl | None = None

services/catalog/openapi.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
"info": {
44
"title": "simcore-service-catalog",
55
"description": "Manages and maintains a catalog of all published components (e.g. macro-algorithms, scripts, etc)",
6-
"version": "0.7.0"
6+
"version": "0.8.0"
77
},
88
"paths": {
99
"/": {

services/catalog/src/simcore_service_catalog/core/background_tasks.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
from ..db.repositories.groups import GroupsRepository
2929
from ..db.repositories.projects import ProjectsRepository
3030
from ..db.repositories.services import ServicesRepository
31-
from ..models.services_db import ServiceAccessRightsAtDB, ServiceMetaDataDBGet
31+
from ..models.services_db import ServiceAccessRightsAtDB, ServiceMetaDataDBCreate
3232
from ..services import access_rights
3333

3434
_logger = logging.getLogger(__name__)
@@ -89,7 +89,9 @@ def _by_version(t: tuple[ServiceKey, ServiceVersion]) -> Version:
8989

9090
# set the service in the DB
9191
await services_repo.create_or_update_service(
92-
ServiceMetaDataDBGet(**service_metadata.model_dump(), owner=owner_gid),
92+
ServiceMetaDataDBCreate(
93+
**service_metadata.model_dump(exclude_unset=True), owner=owner_gid
94+
),
9395
service_access_rights,
9496
)
9597

services/catalog/src/simcore_service_catalog/db/repositories/services.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ async def create_or_update_service(
231231
result = await conn.execute(
232232
# pylint: disable=no-value-for-parameter
233233
services_meta_data.insert()
234-
.values(**new_service.model_dump(by_alias=True, exclude_unset=True))
234+
.values(**new_service.model_dump(exclude_unset=True))
235235
.returning(*SERVICES_META_DATA_COLS)
236236
)
237237
row = result.first()

0 commit comments

Comments
 (0)