|
12 | 12 | from aiohttp import web |
13 | 13 | from aiohttp.test_utils import TestClient |
14 | 14 | from common_library.dict_tools import remap_keys |
15 | | -from models_library.projects_nodes import Node |
16 | 15 | from models_library.projects_nodes_io import NodeID |
17 | 16 | from models_library.services_resources import ServiceResourcesDictHelpers |
18 | | -from pydantic import TypeAdapter |
19 | 17 | from simcore_postgres_database.utils_projects_nodes import ProjectNodeCreate |
20 | 18 | from simcore_service_webserver.projects._groups_repository import ( |
21 | 19 | update_or_insert_project_group, |
@@ -77,30 +75,49 @@ async def create_project( |
77 | 75 |
|
78 | 76 | db: ProjectDBAPI = app[APP_PROJECT_DBAPI] |
79 | 77 |
|
80 | | - raw_workbench = project_data.pop("workbench", {}) |
| 78 | + raw_workbench: dict[str, Any] = project_data.pop("workbench", {}) |
| 79 | + for raw_node in raw_workbench.values(): |
| 80 | + if "position" in raw_node: |
| 81 | + del raw_node["position"] |
| 82 | + |
| 83 | + # Get valid ProjectNodeCreate fields, excluding node_id since it's set separately |
| 84 | + valid_fields = ProjectNodeCreate.get_field_names(exclude={"node_id"}) |
| 85 | + |
| 86 | + # Mapping from camelCase (workbench) to snake_case (ProjectNodeCreate) |
| 87 | + field_mapping = { |
| 88 | + "inputAccess": "input_access", |
| 89 | + "inputNodes": "input_nodes", |
| 90 | + "inputsUnits": "inputs_units", |
| 91 | + "outputNodes": "output_nodes", |
| 92 | + "runHash": "run_hash", |
| 93 | + "bootOptions": "boot_options", |
| 94 | + } |
81 | 95 |
|
82 | | - workbench = TypeAdapter(dict[NodeID, Node]).validate_python(raw_workbench) |
83 | 96 | fake_required_resources: dict[str, Any] = ServiceResourcesDictHelpers.model_config[ |
84 | 97 | "json_schema_extra" |
85 | 98 | ]["examples"][0] |
86 | 99 |
|
| 100 | + project_nodes = { |
| 101 | + NodeID(node_id): ProjectNodeCreate( |
| 102 | + node_id=NodeID(node_id), |
| 103 | + # NOTE: fake initial resources until more is needed |
| 104 | + required_resources=fake_required_resources, |
| 105 | + **{ |
| 106 | + str(field_mapping.get(field, field)): value |
| 107 | + for field, value in raw_node.items() |
| 108 | + if field_mapping.get(field, field) in valid_fields |
| 109 | + }, |
| 110 | + ) |
| 111 | + for node_id, raw_node in raw_workbench.items() |
| 112 | + } |
| 113 | + |
87 | 114 | project_created = await db.insert_project( |
88 | 115 | project_data, |
89 | 116 | user_id, |
90 | 117 | product_name=product_name, |
91 | 118 | force_project_uuid=force_uuid, |
92 | 119 | force_as_template=as_template, |
93 | | - # NOTE: fake initial resources until more is needed |
94 | | - project_nodes={ |
95 | | - node_id: ProjectNodeCreate( |
96 | | - node_id=node_id, |
97 | | - required_resources=fake_required_resources, |
98 | | - **node_model.model_dump( |
99 | | - by_alias=False, exclude_unset=True, mode="json" |
100 | | - ), |
101 | | - ) |
102 | | - for node_id, node_model in workbench.items() |
103 | | - }, |
| 120 | + project_nodes=project_nodes, |
104 | 121 | ) |
105 | 122 |
|
106 | 123 | if params_override and ( |
|
0 commit comments