| 
8 | 8 | from fastapi.encoders import jsonable_encoder  | 
9 | 9 | from models_library.aiodocker_api import AioDockerServiceSpec  | 
10 | 10 | from models_library.progress_bar import ProgressReport  | 
11 |  | -from models_library.projects import ProjectAtDB  | 
 | 11 | +from models_library.projects import NodesDict  | 
12 | 12 | from models_library.projects_nodes import Node  | 
13 | 13 | from models_library.projects_nodes_io import NodeIDStr  | 
14 | 14 | from models_library.rabbitmq_messages import (  | 
 | 
35 | 35 | from .....utils.dict_utils import nested_update  | 
36 | 36 | from ....catalog import CatalogClient  | 
37 | 37 | from ....db.repositories.groups_extra_properties import GroupsExtraPropertiesRepository  | 
38 |  | -from ....db.repositories.projects import ProjectsRepository  | 
 | 38 | +from ....db.repositories.projects_nodes import ProjectsNodesRepository  | 
39 | 39 | from ...docker_api import (  | 
40 | 40 |     constrain_service_to_node,  | 
41 | 41 |     create_network,  | 
@@ -172,14 +172,14 @@ async def action(cls, app: FastAPI, scheduler_data: SchedulerData) -> None:  | 
172 | 172 |         # also other encodes the env vars to target the proper container  | 
173 | 173 | 
 
  | 
174 | 174 |         # fetching project form DB and fetching user settings  | 
175 |  | -        projects_repository = get_repository(app, ProjectsRepository)  | 
 | 175 | +        project_nodes_repository = get_repository(app, ProjectsNodesRepository)  | 
176 | 176 | 
 
  | 
177 |  | -        project: ProjectAtDB = await projects_repository.get_project(  | 
 | 177 | +        workbench: NodesDict = await project_nodes_repository.get_nodes(  | 
178 | 178 |             project_id=scheduler_data.project_id  | 
179 | 179 |         )  | 
180 | 180 | 
 
  | 
181 | 181 |         node_uuid_str = NodeIDStr(scheduler_data.node_uuid)  | 
182 |  | -        node: Node | None = project.workbench.get(node_uuid_str)  | 
 | 182 | +        node: Node | None = workbench.get(node_uuid_str)  | 
183 | 183 |         boot_options = (  | 
184 | 184 |             node.boot_options  | 
185 | 185 |             if node is not None and node.boot_options is not None  | 
 | 
0 commit comments