File tree Expand file tree Collapse file tree 3 files changed +20
-6
lines changed
postgres-database/src/simcore_postgres_database
pytest-simcore/src/pytest_simcore/helpers
services/web/server/src/simcore_service_webserver/projects Expand file tree Collapse file tree 3 files changed +20
-6
lines changed Original file line number Diff line number Diff line change @@ -40,6 +40,9 @@ class ProjectNodesDuplicateNodeError(BaseProjectNodesError):
4040class ProjectNodeCreate (BaseModel ):
4141 node_id : uuid .UUID
4242 required_resources : dict [str , Any ] = Field (default_factory = dict )
43+ key : str
44+ version : str
45+ label : str
4346
4447 @classmethod
4548 def get_field_names (cls , * , exclude : set [str ]) -> set [str ]:
@@ -65,7 +68,7 @@ async def add(
6568 * ,
6669 nodes : list [ProjectNodeCreate ],
6770 ) -> list [ProjectNode ]:
68- """creates a new entry in *projects_nodes* and *projects_to_projects_nodes* tables
71+ """Creates a new entry in *projects_nodes* table
6972
7073 NOTE: Do not use this in an asyncio.gather call as this will fail!
7174
Original file line number Diff line number Diff line change @@ -81,8 +81,11 @@ async def create_project(
8181 required_resources = ServiceResourcesDictHelpers .model_config [
8282 "json_schema_extra"
8383 ]["examples" ][0 ],
84+ key = node_info .get ("key" ),
85+ version = node_info .get ("version" ),
86+ label = node_info .get ("label" ),
8487 )
85- for node_id in project_data .get ("workbench" , {})
88+ for node_id , node_info in project_data .get ("workbench" , {}). items ( )
8689 },
8790 )
8891
Original file line number Diff line number Diff line change @@ -216,19 +216,27 @@ def _reraise_if_not_unique_uuid_error(err: UniqueViolation):
216216 if project_nodes is None :
217217 project_nodes = {
218218 NodeID (node_id ): ProjectNodeCreate (
219- node_id = NodeID (node_id ), required_resources = {}
219+ node_id = NodeID (node_id ),
220+ required_resources = {},
221+ key = node_info .get ("key" ),
222+ version = node_info .get ("version" ),
223+ label = node_info .get ("label" ),
220224 )
221- for node_id in selected_values ["workbench" ]
225+ for node_id , node_info in selected_values ["workbench" ]. items ()
222226 }
223227
224228 nodes = [
225229 project_nodes .get (
226230 NodeID (node_id ),
227231 ProjectNodeCreate (
228- node_id = NodeID (node_id ), required_resources = {}
232+ node_id = NodeID (node_id ),
233+ required_resources = {},
234+ key = node_info .get ("key" ),
235+ version = node_info .get ("version" ),
236+ label = node_info .get ("label" ),
229237 ),
230238 )
231- for node_id in selected_values ["workbench" ]
239+ for node_id , node_info in selected_values ["workbench" ]. items ()
232240 ]
233241 await project_nodes_repo .add (conn , nodes = nodes )
234242 return selected_values
You can’t perform that action at this time.
0 commit comments