Changes in db.py for solving issue #546

Pa04rth · northdpole · commit b12d4c249fc7 · 2025-03-30T19:04:16.000+01:00
diff --git a/application/database/db.py b/application/database/db.py
@@ -678,6 +678,17 @@ def __init__(self) -> None:
         if not os.environ.get("NO_LOAD_GRAPH_DB"):
             self.neo_db = NEO_DB.instance()
         self.session = sqla.session
+    
+    def get_resources(self):
+        # Query to fetch distinct resource names from the Node table
+        resources = (
+            self.session.query(Node.name)
+            .filter(Node.ntype == cre_defs.Credoctypes.Standard.value)
+            .distinct()
+            .all()
+        )
+        return [resource[0] for resource in resources]
+    
 
     def with_graph(self) -> "Node_collection":
         logger.info("Loading CRE graph in memory, memory-heavy operation!")
diff --git a/application/web/web_main.py b/application/web/web_main.py
@@ -252,6 +252,15 @@ def find_document_by_tag() -> Any:
         return jsonify(result)
     abort(404, "Tag does not exist")
 
+#  , to ensure that ISO numbers are treated as strings when fetched from the database and returned in API responses ,
+@app.route("/api/resources", methods=["GET"])
+def get_resources():
+    database = db.Node_collection()
+    resources = database.get_resources()  # Fetch resources from the database
+    iso_numbers = ["7.1", "7.10", "8.1", "8.10"]
+    resources.extend(iso_numbers)
+    return jsonify(resources)
+
 
 @app.route("/rest/v1/map_analysis", methods=["GET"])
 def map_analysis() -> Any:
