OpenGeodeWeb-Back 5.10.4__py3-none-any.whl → 5.14.1__py3-none-any.whl

opengeodeweb_back/routes/blueprint_routes.py

@@ -1,17 +1,31 @@
 # Standard library imports
-import json
 import os
 import time
+import shutil
+from typing import Any
 
 # Third party imports
 import flask
-import opengeode
 import werkzeug
+import zipfile
+import opengeode_geosciences as og_geosciences
+from opengeodeweb_microservice.schemas import get_schemas_dict
+from opengeodeweb_microservice.database.data import Data
+from opengeodeweb_microservice.database.connection import get_session
+from opengeodeweb_microservice.database import connection
 
 # Local application imports
-from .. import geode_functions, utils_functions
-
 from .models import blueprint_models
+from . import schemas
+from opengeodeweb_back import geode_functions, utils_functions
+from opengeodeweb_back.geode_objects import geode_objects
+from opengeodeweb_back.geode_objects.types import geode_object_type
+from opengeodeweb_back.geode_objects.geode_mesh import GeodeMesh
+from opengeodeweb_back.geode_objects.geode_grid2d import GeodeGrid2D
+from opengeodeweb_back.geode_objects.geode_grid3d import GeodeGrid3D
+from opengeodeweb_back.geode_objects.geode_surface_mesh2d import GeodeSurfaceMesh2D
+from opengeodeweb_back.geode_objects.geode_surface_mesh3d import GeodeSurfaceMesh3D
+from opengeodeweb_back.geode_objects.geode_solid_mesh3d import GeodeSolidMesh3D
 
 routes = flask.Blueprint("routes", __name__, url_prefix="/opengeodeweb_back")
 
@@ -22,108 +36,87 @@ routes.register_blueprint(
     name=blueprint_models.routes.name,
 )
 
-
-schemas = os.path.join(os.path.dirname(__file__), "schemas")
-
-with open(
-    os.path.join(schemas, "allowed_files.json"),
-    "r",
-) as file:
-    allowed_files_json = json.load(file)
+schemas_dict = get_schemas_dict(os.path.join(os.path.dirname(__file__), "schemas"))
 
 
 @routes.route(
-    allowed_files_json["route"],
-    methods=allowed_files_json["methods"],
+    schemas_dict["allowed_files"]["route"],
+    methods=schemas_dict["allowed_files"]["methods"],
 )
-def allowed_files():
-    utils_functions.validate_request(flask.request, allowed_files_json)
-    extensions = geode_functions.list_input_extensions(
-        flask.request.get_json()["supported_feature"]
-    )
-    return flask.make_response({"extensions": extensions}, 200)
-
-
-with open(
-    os.path.join(schemas, "upload_file.json"),
-    "r",
-) as file:
-    upload_file_json = json.load(file)
+def allowed_files() -> flask.Response:
+    utils_functions.validate_request(flask.request, schemas_dict["allowed_files"])
+    extensions: set[str] = set()
+    for geode_object in geode_objects.values():
+        for extension in geode_object.input_extensions():
+            extensions.add(extension)
+    return flask.make_response({"extensions": list(extensions)}, 200)
 
 
 @routes.route(
-    upload_file_json["route"],
-    methods=upload_file_json["methods"],
+    schemas_dict["upload_file"]["route"],
+    methods=schemas_dict["upload_file"]["methods"],
 )
-def upload_file():
-    if flask.request.method == "OPTIONS":
-        return flask.make_response({}, 200)
-
+def upload_file() -> flask.Response:
     UPLOAD_FOLDER = flask.current_app.config["UPLOAD_FOLDER"]
     if not os.path.exists(UPLOAD_FOLDER):
-        os.mkdir(UPLOAD_FOLDER)
+        os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+
     file = flask.request.files["file"]
     filename = werkzeug.utils.secure_filename(os.path.basename(file.filename))
     file.save(os.path.join(UPLOAD_FOLDER, filename))
     return flask.make_response({"message": "File uploaded"}, 201)
 
 
-with open(
-    os.path.join(schemas, "allowed_objects.json"),
-    "r",
-) as file:
-    allowed_objects_json = json.load(file)
-
-
 @routes.route(
-    allowed_objects_json["route"],
-    methods=allowed_objects_json["methods"],
+    schemas_dict["allowed_objects"]["route"],
+    methods=schemas_dict["allowed_objects"]["methods"],
 )
-def allowed_objects():
-    if flask.request.method == "OPTIONS":
-        return flask.make_response({}, 200)
-
-    utils_functions.validate_request(flask.request, allowed_objects_json)
-    file_absolute_path = geode_functions.upload_file_path(
-        flask.request.get_json()["filename"]
+def allowed_objects() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["allowed_objects"]
     )
-    allowed_objects = geode_functions.list_geode_objects(
-        file_absolute_path, flask.request.get_json()["supported_feature"]
+    params = schemas.AllowedObjects.from_dict(json_data)
+    file_absolute_path = geode_functions.upload_file_path(params.filename)
+    file_extension = utils_functions.extension_from_filename(
+        os.path.basename(file_absolute_path)
     )
+    allowed_objects = {}
+    for geode_object_type, geode_object in geode_objects.items():
+        if file_extension not in geode_object.input_extensions():
+            continue
+        loadability_score = geode_object.is_loadable(file_absolute_path)
+        priority_score = geode_object.object_priority(file_absolute_path)
+        allowed_objects[geode_object_type] = {
+            "is_loadable": loadability_score.value(),
+            "object_priority": priority_score,
+        }
     return flask.make_response({"allowed_objects": allowed_objects}, 200)
 
 
-with open(
-    os.path.join(schemas, "missing_files.json"),
-    "r",
-) as file:
-    missing_files_json = json.load(file)
-
-
 @routes.route(
-    missing_files_json["route"],
-    methods=missing_files_json["methods"],
+    schemas_dict["missing_files"]["route"],
+    methods=schemas_dict["missing_files"]["methods"],
 )
-def missing_files():
-    utils_functions.validate_request(flask.request, missing_files_json)
-    file_path = geode_functions.upload_file_path(flask.request.get_json()["filename"])
-
-    additional_files = geode_functions.additional_files(
-        flask.request.get_json()["input_geode_object"],
+def missing_files() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["missing_files"]
+    )
+    params = schemas.MissingFiles.from_dict(json_data)
+    file_path = geode_functions.upload_file_path(params.filename)
+    geode_object = geode_functions.geode_object_from_string(params.geode_object_type)
+    additional_files = geode_object.additional_files(
         file_path,
     )
-
     has_missing_files = any(
         file.is_missing
         for file in additional_files.mandatory_files + additional_files.optional_files
     )
-
     mandatory_files = [
         os.path.basename(file.filename)
         for file in additional_files.mandatory_files
         if file.is_missing
     ]
-    additional_files = [
+    additional_files_array = [
         os.path.basename(file.filename)
         for file in additional_files.optional_files
         if file.is_missing
@@ -133,27 +126,26 @@ def missing_files():
         {
             "has_missing_files": has_missing_files,
             "mandatory_files": mandatory_files,
-            "additional_files": additional_files,
+            "additional_files": additional_files_array,
         },
         200,
     )
 
 
-with open(
-    os.path.join(schemas, "geographic_coordinate_systems.json"),
-    "r",
-) as file:
-    geographic_coordinate_systems_json = json.load(file)
-
-
 @routes.route(
-    geographic_coordinate_systems_json["route"],
-    methods=geographic_coordinate_systems_json["methods"],
+    schemas_dict["geographic_coordinate_systems"]["route"],
+    methods=schemas_dict["geographic_coordinate_systems"]["methods"],
 )
-def crs_converter_geographic_coordinate_systems():
-    utils_functions.validate_request(flask.request, geographic_coordinate_systems_json)
-    infos = geode_functions.geographic_coordinate_systems(
-        flask.request.get_json()["input_geode_object"]
+def crs_converter_geographic_coordinate_systems() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["geographic_coordinate_systems"]
+    )
+    params = schemas.GeographicCoordinateSystems.from_dict(json_data)
+    geode_object = geode_functions.geode_object_from_string(params.geode_object_type)
+    infos = (
+        og_geosciences.GeographicCoordinateSystem3D.geographic_coordinate_systems()
+        if geode_object.is_3D()
+        else og_geosciences.GeographicCoordinateSystem2D.geographic_coordinate_systems()
     )
     crs_list = []
     for info in infos:
@@ -162,59 +154,70 @@ def crs_converter_geographic_coordinate_systems():
         crs["code"] = info.code
         crs["authority"] = info.authority
         crs_list.append(crs)
-
     return flask.make_response({"crs_list": crs_list}, 200)
 
 
-with open(
-    os.path.join(schemas, "inspect_file.json"),
-    "r",
-) as file:
-    inspect_file_json = json.load(file)
-
-
 @routes.route(
-    inspect_file_json["route"],
-    methods=inspect_file_json["methods"],
+    schemas_dict["inspect_file"]["route"],
+    methods=schemas_dict["inspect_file"]["methods"],
 )
-def inspect_file():
-    utils_functions.validate_request(flask.request, inspect_file_json)
-
-    file_path = geode_functions.upload_file_path(flask.request.get_json()["filename"])
-    data = geode_functions.load(
-        flask.request.get_json()["input_geode_object"], file_path
-    )
-    class_inspector = geode_functions.inspect(
-        flask.request.get_json()["input_geode_object"], data
+def inspect_file() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["inspect_file"]
     )
-    inspection_result = geode_functions.get_inspector_children(class_inspector)
+    params = schemas.InspectFile.from_dict(json_data)
+    file_path = geode_functions.upload_file_path(params.filename)
+    geode_object = geode_functions.geode_object_from_string(
+        params.geode_object_type
+    ).load(file_path)
+    inspection_data = geode_object.inspect()
+    inspection_result = extract_inspector_result(inspection_data)
     return flask.make_response({"inspection_result": inspection_result}, 200)
 
 
-with open(
-    os.path.join(schemas, "geode_objects_and_output_extensions.json"),
-    "r",
-) as file:
-    geode_objects_and_output_extensions_json = json.load(file)
+def extract_inspector_result(inspection_data: Any) -> object:
+    new_object = {}
+
+    if hasattr(inspection_data, "inspection_type"):
+        new_object["title"] = inspection_data.inspection_type()
+        new_object["nb_issues"] = 0
+        new_object["children"] = []
+        for child in dir(inspection_data):
+            if child.startswith("__") or child in [
+                "inspection_type",
+                "string",
+            ]:
+                continue
+            child_instance = getattr(inspection_data, child)
+            child_object = extract_inspector_result(child_instance)
+            new_object["children"].append(child_object)
+            if hasattr(child_object, "nb_issues"):
+                new_object["nb_issues"] += child_object.nb_issues()
+    else:
+        new_object["title"] = inspection_data.description()
+        nb_issues = inspection_data.nb_issues()
+        new_object["nb_issues"] = nb_issues
+        if nb_issues > 0:
+            issues = inspection_data.string().split("\n")
+            new_object["issues"] = issues
+    return new_object
 
 
 @routes.route(
-    geode_objects_and_output_extensions_json["route"],
-    methods=geode_objects_and_output_extensions_json["methods"],
+    schemas_dict["geode_objects_and_output_extensions"]["route"],
+    methods=schemas_dict["geode_objects_and_output_extensions"]["methods"],
 )
-def geode_objects_and_output_extensions():
-    utils_functions.validate_request(
-        flask.request, geode_objects_and_output_extensions_json
-    )
-    file_path = geode_functions.upload_file_path(flask.request.get_json()["filename"])
-    data = geode_functions.load(
-        flask.request.get_json()["input_geode_object"],
-        file_path,
+def geode_objects_and_output_extensions() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["geode_objects_and_output_extensions"]
     )
+    params = schemas.GeodeObjectsAndOutputExtensions.from_dict(json_data)
+    file_path = geode_functions.upload_file_path(params.filename)
+    geode_object = geode_functions.geode_object_from_string(
+        params.geode_object_type
+    ).load(file_path)
     geode_objects_and_output_extensions = (
-        geode_functions.geode_objects_output_extensions(
-            flask.request.get_json()["input_geode_object"], data
-        )
+        geode_functions.geode_object_output_extensions(geode_object)
     )
     return flask.make_response(
         {"geode_objects_and_output_extensions": geode_objects_and_output_extensions},
@@ -222,82 +225,53 @@ def geode_objects_and_output_extensions():
     )
 
 
-with open(
-    os.path.join(schemas, "save_viewable_file.json"),
-    "r",
-) as file:
-    save_viewable_file_json = json.load(file)
-
-
 @routes.route(
-    save_viewable_file_json["route"],
-    methods=save_viewable_file_json["methods"],
+    schemas_dict["save_viewable_file"]["route"],
+    methods=schemas_dict["save_viewable_file"]["methods"],
 )
-def save_viewable_file():
-    utils_functions.validate_request(flask.request, save_viewable_file_json)
-    return flask.make_response(
-        utils_functions.generate_native_viewable_and_light_viewable_from_file(
-            geode_object=flask.request.get_json()["input_geode_object"],
-            input_filename=flask.request.get_json()["filename"],
-        ),
-        200,
+def save_viewable_file() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["save_viewable_file"]
     )
-
-
-with open(os.path.join(schemas, "create_point.json"), "r") as file:
-    create_point_json = json.load(file)
-
-
-@routes.route(create_point_json["route"], methods=create_point_json["methods"])
-def create_point():
-    utils_functions.validate_request(flask.request, create_point_json)
-    title = flask.request.get_json()["title"]
-    x = flask.request.get_json()["x"]
-    y = flask.request.get_json()["y"]
-    z = flask.request.get_json()["z"]
-    class_ = geode_functions.geode_object_class("PointSet3D")
-    PointSet3D = class_.create()
-    builder = geode_functions.create_builder("PointSet3D", PointSet3D)
-    builder.create_point(opengeode.Point3D([x, y, z]))
-    builder.set_name(title)
+    params = schemas.SaveViewableFile.from_dict(json_data)
     return flask.make_response(
-        utils_functions.generate_native_viewable_and_light_viewable_from_object(
-            "PointSet3D", PointSet3D
+        utils_functions.generate_native_viewable_and_light_viewable_from_file(
+            geode_object_type=geode_object_type(params.geode_object_type),
+            input_file=params.filename,
         ),
         200,
     )
 
 
-with open(os.path.join(schemas, "texture_coordinates.json"), "r") as file:
-    texture_coordinates_json = json.load(file)
-
-
 @routes.route(
-    texture_coordinates_json["route"],
-    methods=texture_coordinates_json["methods"],
+    schemas_dict["texture_coordinates"]["route"],
+    methods=schemas_dict["texture_coordinates"]["methods"],
 )
-def texture_coordinates():
-    utils_functions.validate_request(flask.request, texture_coordinates_json)
-    data = geode_functions.load_data(flask.request.get_json().get("id"))
-    texture_coordinates = data.texture_manager().texture_names()
+def texture_coordinates() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["texture_coordinates"]
+    )
+    params = schemas.TextureCoordinates.from_dict(json_data)
+    geode_object = geode_functions.load_geode_object(params.id)
+    if not isinstance(geode_object, GeodeSurfaceMesh2D | GeodeSurfaceMesh3D):
+        flask.abort(400, f"{params.id} is not a GeodeSurfaceMesh")
+    texture_coordinates = geode_object.texture_manager().texture_names()
     return flask.make_response({"texture_coordinates": texture_coordinates}, 200)
 
 
-with open(
-    os.path.join(schemas, "vertex_attribute_names.json"),
-    "r",
-) as file:
-    vertex_attribute_names_json = json.load(file)
-
-
 @routes.route(
-    vertex_attribute_names_json["route"],
-    methods=vertex_attribute_names_json["methods"],
+    schemas_dict["vertex_attribute_names"]["route"],
+    methods=schemas_dict["vertex_attribute_names"]["methods"],
 )
-def vertex_attribute_names():
-    utils_functions.validate_request(flask.request, vertex_attribute_names_json)
-    data = geode_functions.load_data(flask.request.get_json().get("id"))
-    vertex_attribute_names = data.vertex_attribute_manager().attribute_names()
+def vertex_attribute_names() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["vertex_attribute_names"]
+    )
+    params = schemas.VertexAttributeNames.from_dict(json_data)
+    geode_object = geode_functions.load_geode_object(params.id)
+    if not isinstance(geode_object, GeodeMesh):
+        flask.abort(400, f"{params.id} is not a GeodeMesh")
+    vertex_attribute_names = geode_object.vertex_attribute_manager().attribute_names()
     return flask.make_response(
         {
             "vertex_attribute_names": vertex_attribute_names,
@@ -306,21 +280,40 @@ def vertex_attribute_names():
     )
 
 
-with open(
-    os.path.join(schemas, "polygon_attribute_names.json"),
-    "r",
-) as file:
-    polygon_attribute_names_json = json.load(file)
+@routes.route(
+    schemas_dict["cell_attribute_names"]["route"],
+    methods=schemas_dict["cell_attribute_names"]["methods"],
+)
+def cell_attribute_names() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["cell_attribute_names"]
+    )
+    params = schemas.PolygonAttributeNames.from_dict(json_data)
+    geode_object = geode_functions.load_geode_object(params.id)
+    if not isinstance(geode_object, GeodeGrid2D | GeodeGrid3D):
+        flask.abort(400, f"{params.id} is not a GeodeGrid")
+    cell_attribute_names = geode_object.cell_attribute_manager().attribute_names()
+    return flask.make_response(
+        {
+            "cell_attribute_names": cell_attribute_names,
+        },
+        200,
+    )
 
 
 @routes.route(
-    polygon_attribute_names_json["route"],
-    methods=polygon_attribute_names_json["methods"],
+    schemas_dict["polygon_attribute_names"]["route"],
+    methods=schemas_dict["polygon_attribute_names"]["methods"],
 )
-def polygon_attribute_names():
-    utils_functions.validate_request(flask.request, polygon_attribute_names_json)
-    data = geode_functions.load_data(flask.request.get_json().get("id"))
-    polygon_attribute_names = data.polygon_attribute_manager().attribute_names()
+def polygon_attribute_names() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["polygon_attribute_names"]
+    )
+    params = schemas.PolygonAttributeNames.from_dict(json_data)
+    geode_object = geode_functions.load_geode_object(params.id)
+    if not isinstance(geode_object, GeodeSurfaceMesh2D | GeodeSurfaceMesh3D):
+        flask.abort(400, f"{params.id} is not a GeodeSurfaceMesh")
+    polygon_attribute_names = geode_object.polygon_attribute_manager().attribute_names()
     return flask.make_response(
         {
             "polygon_attribute_names": polygon_attribute_names,
@@ -329,21 +322,21 @@ def polygon_attribute_names():
     )
 
 
-with open(
-    os.path.join(schemas, "polyhedron_attribute_names.json"),
-    "r",
-) as file:
-    polyhedron_attribute_names_json = json.load(file)
-
-
 @routes.route(
-    polyhedron_attribute_names_json["route"],
-    methods=polyhedron_attribute_names_json["methods"],
+    schemas_dict["polyhedron_attribute_names"]["route"],
+    methods=schemas_dict["polyhedron_attribute_names"]["methods"],
 )
-def polyhedron_attribute_names():
-    utils_functions.validate_request(flask.request, polyhedron_attribute_names_json)
-    data = geode_functions.load_data(flask.request.get_json().get("id"))
-    polyhedron_attribute_names = data.polyhedron_attribute_manager().attribute_names()
+def polyhedron_attribute_names() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["polyhedron_attribute_names"]
+    )
+    params = schemas.PolyhedronAttributeNames.from_dict(json_data)
+    geode_object = geode_functions.load_geode_object(params.id)
+    if not isinstance(geode_object, GeodeSolidMesh3D):
+        flask.abort(400, f"{params.id} is not a GeodeSolidMesh")
+    polyhedron_attribute_names = (
+        geode_object.polyhedron_attribute_manager().attribute_names()
+    )
     return flask.make_response(
         {
             "polyhedron_attribute_names": polyhedron_attribute_names,
@@ -352,32 +345,160 @@ def polyhedron_attribute_names():
     )
 
 
-with open(
-    os.path.join(schemas, "ping.json"),
-    "r",
-) as file:
-    ping_json = json.load(file)
-
-
 @routes.route(
-    ping_json["route"],
-    methods=ping_json["methods"],
+    schemas_dict["ping"]["route"],
+    methods=schemas_dict["ping"]["methods"],
 )
-def ping():
-    utils_functions.validate_request(flask.request, ping_json)
+def ping() -> flask.Response:
+    utils_functions.validate_request(flask.request, schemas_dict["ping"])
     flask.current_app.config.update(LAST_PING_TIME=time.time())
     return flask.make_response({"message": "Flask server is running"}, 200)
 
 
-with open(
-    os.path.join(schemas, "kill.json"),
-    "r",
-) as file:
-    kill_json = json.load(file)
-
-
-@routes.route(kill_json["route"], methods=kill_json["methods"])
+@routes.route(schemas_dict["kill"]["route"], methods=schemas_dict["kill"]["methods"])
 def kill() -> flask.Response:
     print("Manual server kill, shutting down...", flush=True)
     os._exit(0)
     return flask.make_response({"message": "Flask server is dead"}, 200)
+
+
+@routes.route(
+    schemas_dict["export_project"]["route"],
+    methods=schemas_dict["export_project"]["methods"],
+)
+def export_project() -> flask.Response:
+    json_data = utils_functions.validate_request(
+        flask.request, schemas_dict["export_project"]
+    )
+    params = schemas.ExportProject.from_dict(json_data)
+
+    project_folder: str = flask.current_app.config["DATA_FOLDER_PATH"]
+    os.makedirs(project_folder, exist_ok=True)
+
+    filename: str = werkzeug.utils.secure_filename(os.path.basename(params.filename))
+    if not filename.lower().endswith(".vease"):
+        flask.abort(400, "Requested filename must end with .vease")
+    export_vease_path = os.path.join(project_folder, filename)
+
+    with get_session() as session:
+        rows = session.query(Data.id, Data.input_file, Data.additional_files).all()
+
+    with zipfile.ZipFile(
+        export_vease_path, "w", compression=zipfile.ZIP_DEFLATED
+    ) as zip_file:
+        database_root_path = os.path.join(project_folder, "project.db")
+        if os.path.isfile(database_root_path):
+            zip_file.write(database_root_path, "project.db")
+
+        for data_id, input_file, additional_files in rows:
+            base_dir = os.path.join(project_folder, data_id)
+
+            input_path = os.path.join(base_dir, str(input_file))
+            if os.path.isfile(input_path):
+                zip_file.write(input_path, os.path.join(data_id, str(input_file)))
+
+            for relative_path in (
+                additional_files if isinstance(additional_files, list) else []
+            ):
+                additional_path = os.path.join(base_dir, relative_path)
+                if os.path.isfile(additional_path):
+                    zip_file.write(
+                        additional_path, os.path.join(data_id, relative_path)
+                    )
+
+        zip_file.writestr("snapshot.json", flask.json.dumps(params.snapshot))
+
+    return utils_functions.send_file(project_folder, [export_vease_path], filename)
+
+
+@routes.route(
+    schemas_dict["import_project"]["route"],
+    methods=schemas_dict["import_project"]["methods"],
+)
+def import_project() -> flask.Response:
+    utils_functions.validate_request(flask.request, schemas_dict["import_project"])
+    if "file" not in flask.request.files:
+        flask.abort(400, "No .vease file provided under 'file'")
+    zip_file = flask.request.files["file"]
+    assert zip_file.filename is not None
+    filename = werkzeug.utils.secure_filename(os.path.basename(zip_file.filename))
+    if not filename.lower().endswith(".vease"):
+        flask.abort(400, "Uploaded file must be a .vease")
+
+    data_folder_path: str = flask.current_app.config["DATA_FOLDER_PATH"]
+
+    # 423 Locked bypass : remove stopped requests
+    if connection.scoped_session_registry:
+        connection.scoped_session_registry.remove()
+    if connection.engine:
+        connection.engine.dispose()
+    connection.engine = connection.session_factory = (
+        connection.scoped_session_registry
+    ) = None
+
+    try:
+        if os.path.exists(data_folder_path):
+            shutil.rmtree(data_folder_path)
+        os.makedirs(data_folder_path, exist_ok=True)
+    except PermissionError:
+        flask.abort(423, "Project files are locked; cannot overwrite")
+
+    zip_file.stream.seek(0)
+    with zipfile.ZipFile(zip_file.stream) as zip_archive:
+        project_folder = os.path.abspath(data_folder_path)
+        for member in zip_archive.namelist():
+            target = os.path.abspath(
+                os.path.normpath(os.path.join(project_folder, member))
+            )
+            if not (
+                target == project_folder or target.startswith(project_folder + os.sep)
+            ):
+                flask.abort(400, "Vease file contains unsafe paths")
+        zip_archive.extractall(project_folder)
+
+        database_root_path = os.path.join(project_folder, "project.db")
+        if not os.path.isfile(database_root_path):
+            flask.abort(400, "Missing project.db at project root")
+
+        connection.init_database(database_root_path, create_tables=False)
+
+        try:
+            with get_session() as session:
+                rows = session.query(Data).all()
+        except Exception:
+            connection.init_database(database_root_path, create_tables=True)
+            with get_session() as session:
+                rows = session.query(Data).all()
+
+        with get_session() as session:
+            for data in rows:
+                data_path = geode_functions.data_file_path(data.id)
+                viewable_name = data.viewable_file
+                if viewable_name:
+                    vpath = geode_functions.data_file_path(data.id, viewable_name)
+                    if os.path.isfile(vpath):
+                        continue
+
+                input_file = str(data.input_file or "")
+                if not input_file:
+                    continue
+
+                input_full = geode_functions.data_file_path(data.id, input_file)
+                if not os.path.isfile(input_full):
+                    continue
+
+                geode_object = geode_functions.geode_object_from_string(
+                    data.geode_object
+                ).load(input_full)
+                utils_functions.save_all_viewables_and_return_info(
+                    geode_object, data, data_path
+                )
+            session.commit()
+
+        snapshot = {}
+        try:
+            raw = zip_archive.read("snapshot.json").decode("utf-8")
+            snapshot = flask.json.loads(raw)
+        except KeyError:
+            snapshot = {}
+    return flask.make_response({"snapshot": snapshot}, 200)