add uuid dict logic

Author: SpliiT

src/opengeodeweb_back/routes/models/blueprint_models.py

@@ -2,8 +2,8 @@
 import os
 import xml.etree.ElementTree as ET
 import flask
-
 from src.opengeodeweb_back import geode_functions, utils_functions
+from opengeode import model
 
 routes = flask.Blueprint("models", __name__)
 
@@ -23,13 +23,13 @@ def teardown_request(exception):
 
 schemas = os.path.join(os.path.dirname(__file__), "schemas")
 
-with open(os.path.join(schemas, "components.json"), "r") as file:
-    components_json = json.load(file)
+with open(os.path.join(schemas, "mesh_components.json"), "r") as file:
+    mesh_components_json = json.load(file)
 
 
-@routes.route(components_json["route"], methods=components_json["methods"])
-def print_vtm_file():
-    utils_functions.validate_request(flask.request, components_json)
+@routes.route(mesh_components_json["route"], methods=mesh_components_json["methods"])
+def uuid_to_flat_index():
+    utils_functions.validate_request(flask.request, mesh_components_json)
     vtm_file_path = os.path.join(
         flask.current_app.config["DATA_FOLDER_PATH"], flask.request.json["id"] + ".vtm"
     )
@@ -50,3 +50,45 @@ def print_vtm_file():
         {"uuid_to_flat_index": uuid_to_flat_index},
         200,
     )
+
+
+def extract_model_uuids(geode_object, file_path):
+    model = geode_functions.load(geode_object, file_path)
+    components = {
+        "blocks": getattr(model, "blocks", lambda: [])(),
+        "lines": getattr(model, "lines", lambda: [])(),
+        "surfaces": getattr(model, "surfaces", lambda: [])(),
+        "corners": getattr(model, "corners", lambda: [])(),
+    }
+
+    uuid_dict = {
+        key: [
+            component.id().string()
+            for component in components[key]
+            if hasattr(component, "id")
+        ]
+        for key in components
+        if components[key]
+    }
+
+    print(f"{uuid_dict=}", flush=True)
+    return uuid_dict
+
+
+with open(os.path.join(schemas, "components_types.json"), "r") as file:
+    components_types_json = json.load(file)
+
+
+@routes.route(components_types_json["route"], methods=components_types_json["methods"])
+def extract_uuids_endpoint():
+    utils_functions.validate_request(flask.request, components_types_json)
+
+    file_path = os.path.join(
+        flask.current_app.config["DATA_FOLDER_PATH"], flask.request.json["filename"]
+    )
+
+    if not os.path.exists(file_path):
+        return flask.make_response({"error": "File not found"}, 404)
+
+    uuid_dict = extract_model_uuids(flask.request.json["geode_object"], file_path)
+    return flask.make_response(uuid_dict, 200)

src/opengeodeweb_back/routes/models/schemas/components_types.json

@@ -0,0 +1,20 @@
+{
+  "route": "/components_types",
+  "methods": [
+    "POST"
+  ],
+  "type": "object",
+  "properties": {
+    "filename": {
+      "type": "string"
+    },
+    "geode_object": {
+      "type": "string"
+    }
+  },
+  "required": [
+    "filename",
+    "geode_object"
+  ],
+  "additionalProperties": false
+}

src/opengeodeweb_back/routes/models/schemas/components.json renamed to src/opengeodeweb_back/routes/models/schemas/mesh_components.json

@@ -1,5 +1,5 @@
 {
-  "route": "/components",
+  "route": "/mesh_components",
   "methods": [
     "POST"
   ],

tests/data/cube.og_brep

@@ -1,12 +1,10 @@
 import os
-
 from werkzeug.datastructures import FileStorage
-
 from src.opengeodeweb_back import geode_functions, geode_objects, test_utils
 
 
-def test_model_components(client):
-    route = f"/models/components"
+def test_model_mesh_components(client):
+    route = f"/models/mesh_components"
     get_full_data = lambda: {"id": "cube"}
     json = get_full_data()
     response = client.post(route, json=json)
@@ -19,5 +17,20 @@ def test_model_components(client):
     assert indices[0] == 1
     assert all(indices[i] == indices[i - 1] + 1 for i in range(1, len(indices)))
 
-    # Test all params
-    test_utils.test_route_wrong_params(client, route, get_full_data)
+
+def test_extract_brep_uuids(client):
+    route = "/models/components_types"
+    json_data = {"filename": "cube.og_brep", "geode_object": "BRep"}
+
+    response = client.post(route, json=json_data)
+
+    assert response.status_code == 200
+    uuid_dict = response.json
+    assert isinstance(uuid_dict, dict)
+    assert (
+        "blocks" in uuid_dict
+        or "lines" in uuid_dict
+        or "surfaces" in uuid_dict
+        or "corners" in uuid_dict
+    )
+    assert False