Skip to content

Commit 994c575

Browse files
♻️ Remove deprecated from_orm method (Pydantic v2) (#6869)
1 parent 3dded08 commit 994c575

File tree

5 files changed

+17
-17
lines changed

5 files changed

+17
-17
lines changed

packages/postgres-database/src/simcore_postgres_database/utils_projects_metadata.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ async def get(connection: SAConnection, project_uuid: uuid.UUID) -> ProjectMetad
9393
row: RowProxy | None = await result.first()
9494
if row is None:
9595
raise DBProjectNotFoundError(project_uuid=project_uuid)
96-
return ProjectMetadata.from_orm(row)
96+
return ProjectMetadata.model_validate(row)
9797

9898

9999
def _check_valid_ancestors_combination(
@@ -202,7 +202,7 @@ async def set_project_ancestors(
202202
result: ResultProxy = await connection.execute(upsert_stmt)
203203
row: RowProxy | None = await result.first()
204204
assert row # nosec
205-
return ProjectMetadata.from_orm(row)
205+
return ProjectMetadata.model_validate(row)
206206

207207
except ForeignKeyViolation as err:
208208
assert err.pgerror is not None # nosec # noqa: PT017
@@ -234,7 +234,7 @@ async def set_project_custom_metadata(
234234
result: ResultProxy = await connection.execute(upsert_stmt)
235235
row: RowProxy | None = await result.first()
236236
assert row # nosec
237-
return ProjectMetadata.from_orm(row)
237+
return ProjectMetadata.model_validate(row)
238238

239239
except ForeignKeyViolation as err:
240240
raise DBProjectNotFoundError(project_uuid=project_uuid) from err

packages/postgres-database/src/simcore_postgres_database/utils_projects_nodes.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,7 @@ async def add(
102102
assert result # nosec
103103
rows = await result.fetchall()
104104
assert rows is not None # nosec
105-
return [ProjectNode.from_orm(r) for r in rows]
105+
return [ProjectNode.model_validate(r) for r in rows]
106106
except ForeignKeyViolation as exc:
107107
# this happens when the project does not exist, as we first check the node exists
108108
raise ProjectNodesProjectNotFoundError(
@@ -128,7 +128,7 @@ async def list(self, connection: SAConnection) -> list[ProjectNode]:
128128
assert result # nosec
129129
rows = await result.fetchall()
130130
assert rows is not None # nosec
131-
return [ProjectNode.from_orm(row) for row in rows]
131+
return [ProjectNode.model_validate(row) for row in rows]
132132

133133
async def get(self, connection: SAConnection, *, node_id: uuid.UUID) -> ProjectNode:
134134
"""get a node in the current project
@@ -154,7 +154,7 @@ async def get(self, connection: SAConnection, *, node_id: uuid.UUID) -> ProjectN
154154
project_uuid=self.project_uuid, node_id=node_id
155155
)
156156
assert row # nosec
157-
return ProjectNode.from_orm(row)
157+
return ProjectNode.model_validate(row)
158158

159159
async def update(
160160
self, connection: SAConnection, *, node_id: uuid.UUID, **values
@@ -184,7 +184,7 @@ async def update(
184184
project_uuid=self.project_uuid, node_id=node_id
185185
)
186186
assert row # nosec
187-
return ProjectNode.from_orm(row)
187+
return ProjectNode.model_validate(row)
188188

189189
async def delete(self, connection: SAConnection, *, node_id: uuid.UUID) -> None:
190190
"""delete a node in the current project

packages/pytest-simcore/src/pytest_simcore/db_entries_mocks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,7 +93,7 @@ async def creator(
9393
.returning(sa.literal_column("*"))
9494
)
9595

96-
inserted_project = ProjectAtDB.from_orm(await result.first())
96+
inserted_project = ProjectAtDB.model_validate(await result.first())
9797
project_nodes_repo = ProjectNodesRepo(project_uuid=project_uuid)
9898
# NOTE: currently no resources is passed until it becomes necessary
9999
default_node_config = {"required_resources": {}}

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ async def get_everyone_group(self) -> GroupAtDB:
3737
raise UninitializedGroupError(
3838
group=GroupType.EVERYONE, repo_cls=GroupsRepository
3939
)
40-
return GroupAtDB.from_orm(row)
40+
return GroupAtDB.model_validate(row)
4141

4242
async def get_user_gid_from_email(
4343
self, user_email: LowerCaseEmailStr

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

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ async def list_services(
8383

8484
async with self.db_engine.connect() as conn:
8585
return [
86-
ServiceMetaDataAtDB.from_orm(row)
86+
ServiceMetaDataAtDB.model_validate(row)
8787
async for row in await conn.stream(
8888
list_services_stmt(
8989
gids=gids,
@@ -134,7 +134,7 @@ async def list_service_releases(
134134

135135
async with self.db_engine.connect() as conn:
136136
releases = [
137-
ServiceMetaDataAtDB.from_orm(row)
137+
ServiceMetaDataAtDB.model_validate(row)
138138
async for row in await conn.stream(query)
139139
]
140140

@@ -163,7 +163,7 @@ async def get_latest_release(self, key: str) -> ServiceMetaDataAtDB | None:
163163
result = await conn.execute(query)
164164
row = result.first()
165165
if row:
166-
return ServiceMetaDataAtDB.from_orm(row)
166+
return ServiceMetaDataAtDB.model_validate(row)
167167
return None # mypy
168168

169169
async def get_service(
@@ -208,7 +208,7 @@ async def get_service(
208208
result = await conn.execute(query)
209209
row = result.first()
210210
if row:
211-
return ServiceMetaDataAtDB.from_orm(row)
211+
return ServiceMetaDataAtDB.model_validate(row)
212212
return None # mypy
213213

214214
async def create_or_update_service(
@@ -234,7 +234,7 @@ async def create_or_update_service(
234234
)
235235
row = result.first()
236236
assert row # nosec
237-
created_service = ServiceMetaDataAtDB.from_orm(row)
237+
created_service = ServiceMetaDataAtDB.model_validate(row)
238238

239239
for access_rights in new_service_access_rights:
240240
insert_stmt = pg_insert(services_access_rights).values(
@@ -468,7 +468,7 @@ async def get_service_access_rights(
468468

469469
async with self.db_engine.connect() as conn:
470470
return [
471-
ServiceAccessRightsAtDB.from_orm(row)
471+
ServiceAccessRightsAtDB.model_validate(row)
472472
async for row in await conn.stream(query)
473473
]
474474

@@ -494,7 +494,7 @@ async def list_services_access_rights(
494494
async with self.db_engine.connect() as conn:
495495
async for row in await conn.stream(query):
496496
service_to_access_rights[(row.key, row.version)].append(
497-
ServiceAccessRightsAtDB.from_orm(row)
497+
ServiceAccessRightsAtDB.model_validate(row)
498498
)
499499
return service_to_access_rights
500500

@@ -585,7 +585,7 @@ async def get_service_specifications(
585585
try:
586586
_logger.debug("found following %s", f"{row=}")
587587
# validate the specs first
588-
db_service_spec = ServiceSpecificationsAtDB.from_orm(row)
588+
db_service_spec = ServiceSpecificationsAtDB.model_validate(row)
589589
db_spec_version = packaging.version.parse(
590590
db_service_spec.service_version
591591
)

0 commit comments

Comments
 (0)