feat(import_project): new schema, rpc added to reset viewer and prepare imported datas

Author: MaxNumerique

src/opengeodeweb_viewer/rpc/schemas/reset_project.json

@@ -0,0 +1,7 @@
+{
+  "rpc": "reset_project",
+  "type": "object",
+  "properties": {},
+  "required": [],
+  "additionalProperties": false
+}

src/opengeodeweb_viewer/rpc/utils_protocols.py

@@ -7,6 +7,7 @@
 # Local application imports
 from opengeodeweb_microservice.schemas import get_schemas_dict
 from opengeodeweb_viewer.vtk_protocol import VtkView
+from opengeodeweb_microservice.database import connection
 
 
 class VtkUtilsView(VtkView):
@@ -25,3 +26,29 @@ def kill(self) -> None:
         )
         print("Manual viewer kill, shutting down...", flush=True)
         os._exit(0)
+
+    @exportRpc(utils_prefix + utils_schemas_dict["reset_project"]["rpc"])
+    def resetProject(self, rpc_params: dict | None = None) -> None:
+        print(
+            f"{self.utils_prefix + self.utils_schemas_dict['reset_project']['rpc']}",
+            flush=True,
+        )
+        renderWindow = self.getView("-1")
+        renderer = renderWindow.GetRenderers().GetFirstRenderer()
+
+        renderer.RemoveAllViewProps()
+
+        widget = self.get_widget()
+        if widget is not None:
+            try:
+                widget.EnabledOff()
+            except Exception:
+                pass
+        self.coreServer.setSharedObject("widget", None)
+        self.coreServer.setSharedObject("grid_scale", None)
+        self.coreServer.setSharedObject("axes", None)
+
+        self.get_data_base().clear()
+
+        db_full_path = os.path.join(self.DATA_FOLDER_PATH, "project.db")
+        connection.init_database(db_full_path, create_tables=False)

tests/data/images/viewer/import_project.jpeg

@@ -0,0 +1,46 @@
+from typing import Callable
+from opengeodeweb_viewer.rpc.utils_protocols import VtkUtilsView
+from opengeodeweb_viewer.rpc.viewer.viewer_protocols import VtkViewerView
+from opengeodeweb_viewer.rpc.mesh.mesh_protocols import VtkMeshView
+from opengeodeweb_microservice.database.connection import get_session
+from opengeodeweb_microservice.database.data import Data
+from tests.conftest import ServerMonitor
+
+
+def test_reset_project_after_import(
+    server: ServerMonitor, dataset_factory: Callable[..., str]
+) -> None:
+    # Mock
+    pre_id = "123456789"
+    dataset_factory(id=pre_id, viewable_file_name="hat.vtp")
+    server.call(
+        VtkMeshView.mesh_prefix + VtkMeshView.mesh_schemas_dict["register"]["rpc"],
+        [{"id": pre_id}],
+    )
+    assert server.compare_image(3, "mesh/register.jpeg") is True
+
+    # Import
+    with get_session() as session:
+        session.query(Data).delete()
+        session.commit()
+
+    post_id = "987654321"
+    dataset_factory(id=post_id, viewable_file_name="hat.vtp")
+
+    server.call(
+        VtkUtilsView.utils_prefix
+        + VtkUtilsView.utils_schemas_dict["reset_project"]["rpc"]
+    )
+    server.get_response()
+
+    server.call(
+        VtkViewerView.viewer_prefix
+        + VtkViewerView.viewer_schemas_dict["reset_visualization"]["rpc"]
+    )
+    assert server.compare_image(3, "viewer/reset_visualization.jpeg") is True
+    
+    server.call(
+        VtkMeshView.mesh_prefix + VtkMeshView.mesh_schemas_dict["register"]["rpc"],
+        [{"id": post_id}],
+    )
+    assert server.compare_image(3, "viewer/import_project.jpeg") is True