fix the api part of #531 by providing an api call to export all CREs …

…in an instance in a visually pleasing format

application/frontend/src/pages/Explorer/explorer.tsx

@@ -8,9 +8,9 @@ import { LoadingAndErrorIndicator } from '../../components/LoadingAndErrorIndica
 import { TYPE_CONTAINS, TYPE_LINKED_TO } from '../../const';
 import { useDataStore } from '../../providers/DataProvider';
 import { LinkedTreeDocument, TreeDocument } from '../../types';
-import { LinkedStandards } from './LinkedStandards';
 import { getDocumentDisplayName } from '../../utils';
 import { getInternalUrl } from '../../utils/document';
+import { LinkedStandards } from './LinkedStandards';
 
 export const Explorer = () => {
   const { dataLoading, dataTree } = useDataStore();

application/tests/spreadsheet_test.py

@@ -4,7 +4,7 @@
 from application import create_app, sqla  # type: ignore
 from application.database import db
 from application.defs import cre_defs as defs
-from application.utils.spreadsheet import prepare_spreadsheet
+from application.utils.spreadsheet import prepare_spreadsheet, generate_staggered_csv
 
 
 class TestDB(unittest.TestCase):
@@ -530,6 +530,60 @@ def test_prepare_spreadsheet_simple(self) -> None:
 
         self.assertCountEqual(result, expected)
 
+    def test_generate_staggered_csv(self) -> None:
+        """
+        Given: a CRE structure with 4 depth levels and 2 root cres
+        prepare a staggered csv accordingly
+        """
+        # empty string means temporary db
+        collection = db.Node_collection().with_graph()
+        roots = []
+        for j in range(2):
+            root = defs.CRE(description=f"root{j}", name=f"root{j}", id=f"123-30{j}")
+            db_root = collection.add_cre(root)
+            roots.append(root)
+            previous_db = db_root
+            previous_cre = root
+
+            for i in range(4):
+                c = defs.CRE(
+                    description=f"CREdesc{i}-{j}",
+                    name=f"CREname{i}-{j}",
+                    id=f"123-4{j}{i}",
+                )
+                dbcre = collection.add_cre(c)
+                collection.add_internal_link(
+                    higher=previous_db, lower=dbcre, type=defs.LinkTypes.Contains
+                )
+                previous_cre.add_link(
+                    defs.Link(document=c, ltype=defs.LinkTypes.Contains)
+                )
+                previous_cre = c
+                previous_db = dbcre
+        csv = generate_staggered_csv(database=collection, docs=roots)
+        self.assertEqual(
+            csv,
+            [
+                {
+                    "CRE0-ID": "",
+                    "CRE1-ID": "",
+                    "CRE2-ID": "",
+                    "CRE3-ID": "",
+                    "CRE4-ID": "",
+                },
+                {"CRE0-ID": "123-300"},
+                {"CRE1-ID": "123-400"},
+                {"CRE2-ID": "123-401"},
+                {"CRE3-ID": "123-402"},
+                {"CRE4-ID": "123-403"},
+                {"CRE0-ID": "123-301"},
+                {"CRE1-ID": "123-410"},
+                {"CRE2-ID": "123-411"},
+                {"CRE3-ID": "123-412"},
+                {"CRE4-ID": "123-413"},
+            ],
+        )
+
 
 if __name__ == "__main__":
     unittest.main()

application/tests/web_main_test.py

@@ -1,22 +1,21 @@
+import io
 import csv
 import random
 import string
 import re
 import json
 import unittest
 from unittest.mock import patch
-
 import redis
 import rq
+import os
 
 from application import create_app, sqla  # type: ignore
 from application.database import db
 from application.defs import cre_defs as defs
-from application.defs import osib_defs
 from application.web import web_main
 from application.utils.gap_analysis import GAP_ANALYSIS_TIMEOUT
-
-import os
+from application.utils import spreadsheet
 
 
 class MockJob:
@@ -886,3 +885,61 @@ def test_all_cres(self, db_mock) -> None:
                 {"data": expected, "page": 1, "total_pages": 1},
                 json.loads(response.data),
             )
+
+    def test_get_cre_csv(self) -> None:
+        # empty string means temporary db
+        collection = db.Node_collection().with_graph()
+        roots = []
+        for j in range(2):
+            root = defs.CRE(description=f"root{j}", name=f"root{j}", id=f"123-30{j}")
+            db_root = collection.add_cre(root)
+            roots.append(root)
+            previous_db = db_root
+            previous_cre = root
+
+            for i in range(4):
+                c = defs.CRE(
+                    description=f"CREdesc{i}-{j}",
+                    name=f"CREname{i}-{j}",
+                    id=f"123-4{j}{i}",
+                )
+                dbcre = collection.add_cre(c)
+                collection.add_internal_link(
+                    higher=previous_db, lower=dbcre, type=defs.LinkTypes.Contains
+                )
+                previous_cre.add_link(
+                    defs.Link(document=c, ltype=defs.LinkTypes.Contains)
+                )
+                previous_cre = c
+                previous_db = dbcre
+
+        with self.app.test_client() as client:
+            response = client.get(
+                "/rest/v1/cre_csv",
+                headers={"Content-Type": "application/json"},
+            )
+            self.assertEqual(200, response.status_code)
+            expected_out = [
+                {
+                    "CRE0-ID": "",
+                    "CRE1-ID": "",
+                    "CRE2-ID": "",
+                    "CRE3-ID": "",
+                    "CRE4-ID": "",
+                },
+                {"CRE0-ID": "123-300"},
+                {"CRE1-ID": "123-400"},
+                {"CRE2-ID": "123-401"},
+                {"CRE3-ID": "123-402"},
+                {"CRE4-ID": "123-403"},
+                {"CRE0-ID": "123-301"},
+                {"CRE1-ID": "123-410"},
+                {"CRE2-ID": "123-411"},
+                {"CRE3-ID": "123-412"},
+                {"CRE4-ID": "123-413"},
+            ]
+            data = spreadsheet.write_csv(expected_out)
+            self.assertEqual(
+                data.getvalue(),
+                response.data.decode(),
+            )

application/utils/spreadsheet.py

@@ -2,7 +2,7 @@
 import io
 import logging
 from copy import deepcopy
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Set
 import os
 import gspread
 import yaml
@@ -237,3 +237,59 @@ def write_spreadsheet(title: str, docs: List[Dict[str, Any]], emails: List[str])
     for email in emails:
         sh.share(email, perm_type="user", role="writer")
     return "https://docs.google.com/spreadsheets/d/%s" % sh.id
+
+
+def generate_staggered_csv(database: db.Node_collection, docs: List[defs.CRE]) -> str:
+    maxOffset = 0
+    related = set()
+
+    def add_offset_cre(
+        cre: defs.CRE, database: db.Node_collection, offset: int, visited_cres: Set
+    ) -> List[Dict[str, str]]:
+        nonlocal maxOffset, related
+        maxOffset = max(maxOffset, offset)
+        rows = []
+
+        rows.append({f"CRE{offset}-ID": cre.id})
+        visited_cres.add(cre.id)
+        dbcre = database.get_CREs(external_id=cre.id)
+        if not dbcre:
+            raise ValueError(f"CRE with id {cre.id} not found in the database")
+        cre = dbcre[0]
+        for link in cre.links:
+            if (
+                link.document.doctype == defs.Credoctypes.CRE
+                and link.document.id not in visited_cres
+            ):
+                if link.ltype == defs.LinkTypes.Contains:
+                    rows.extend(
+                        add_offset_cre(
+                            cre=link.document,
+                            database=database,
+                            offset=offset + 1,
+                            visited_cres=visited_cres,
+                        )
+                    )
+                elif link.ltype == defs.LinkTypes.Related:
+                    related.add(link.document.id)
+        return rows
+
+    visited_cres = set()
+    csv: List[Dict[str, str]] = []
+
+    for cre in docs:
+        csv.extend(
+            add_offset_cre(
+                cre=cre, database=database, offset=0, visited_cres=visited_cres
+            )
+        )
+    result = [{f"CRE{offset}-ID": "" for offset in range(0, maxOffset + 1)}]
+    result.extend(csv)
+
+    orphaned_documents = [doc for doc in related if doc not in visited_cres]
+    if len(orphaned_documents):
+        raise ValueError(
+            "found CREs with only related links not provided in the root_cre list, unless you are really sure for this use case, this is a bug"
+        )
+
+    return result

application/web/web_main.py

@@ -684,6 +684,16 @@ def all_cres() -> Any:
     abort(404)
 
 
+@app.route("/rest/v1/cre_csv", methods=["GET"])
+def get_cre_csv() -> Any:
+    database = db.Node_collection()
+    root_cres = database.get_root_cres()
+    if root_cres:
+        docs = sheet_utils.generate_staggered_csv(database=database, docs=root_cres)
+        return write_csv(docs=docs).getvalue().encode("utf-8")
+    abort(404)
+
+
 # @app.route("/rest/v1/all_nodes", methods=["GET"])
 # def all_nodes() -> Any:
 #     database = db.Node_collection()