Global project lock and unlock

gns3server/api/routes/controller/projects.py

@@ -36,7 +36,6 @@ from fastapi.responses import StreamingResponse, FileResponse
 from websockets.exceptions import ConnectionClosed, WebSocketException
 from typing import List, Optional
 from uuid import UUID
-from pathlib import Path
 
 from gns3server import schemas
 from gns3server.controller import Controller
@@ -46,7 +45,6 @@ from gns3server.controller.import_project import import_project as import_contro
 from gns3server.controller.export_project import export_project as export_controller_project
 from gns3server.utils.asyncio import aiozipstream
 from gns3server.utils.path import is_safe_path
-from gns3server.config import Config
 from gns3server.db.repositories.rbac import RbacRepository
 from gns3server.db.repositories.templates import TemplatesRepository
 from gns3server.services.templates import TemplatesService
@@ -397,6 +395,24 @@ async def duplicate_project(
     return new_project.asdict()
 
 
+@router.post("/{project_id}/lock", status_code=status.HTTP_204_NO_CONTENT)
+async def lock_project(project: Project = Depends(dep_project)) -> None:
+    """
+    Lock all drawings and nodes in a given project.
+    """
+
+    project.lock()
+
+
+@router.post("/{project_id}/unlock", status_code=status.HTTP_204_NO_CONTENT)
+async def unlock_project(project: Project = Depends(dep_project)) -> None:
+    """
+    Unlock all drawings and nodes in a given project.
+    """
+
+    project.unlock()
+
+
 @router.get("/{project_id}/files/{file_path:path}")
 async def get_file(file_path: str, project: Project = Depends(dep_project)) -> FileResponse:
     """

gns3server/controller/project.py

@@ -1112,6 +1112,38 @@ class Project:
                 return True
         return False
 
+    @open_required
+    def lock(self):
+        """
+        Lock all drawings and nodes
+        """
+
+        for drawing in self._drawings.values():
+            if not drawing.locked:
+                drawing.locked = True
+                self.emit_notification("drawing.updated", drawing.asdict())
+        for node in self.nodes.values():
+            if not node.locked:
+                node.locked = True
+                self.emit_notification("node.updated", node.asdict())
+        self.dump()
+
+    @open_required
+    def unlock(self):
+        """
+        Unlock all drawings and nodes
+        """
+
+        for drawing in self._drawings.values():
+            if drawing.locked:
+                drawing.locked = False
+                self.emit_notification("drawing.updated", drawing.asdict())
+        for node in self.nodes.values():
+            if node.locked:
+                node.locked = False
+                self.emit_notification("node.updated", node.asdict())
+        self.dump()
+
     def dump(self):
         """
         Dump topology to disk