Add relations in XLSX output #659

Signed-off-by: Thomas Druez <[email protected]>
aboutcode-org · Apr 17, 2023 · 74a1b76 · 74a1b76
1 parent c4f1b0d
commit 74a1b76
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/scanpipe/api/serializers.py b/scanpipe/api/serializers.py
@@ -384,6 +384,7 @@ def get_model_serializer(model_class):
         CodebaseResource: CodebaseResourceSerializer,
         DiscoveredPackage: DiscoveredPackageSerializer,
         DiscoveredDependency: DiscoveredDependencySerializer,
+        CodebaseRelation: CodebaseRelationSerializer,
         ProjectError: ProjectErrorSerializer,
     }.get(model_class, None)
 

diff --git a/scanpipe/pipes/output.py b/scanpipe/pipes/output.py
@@ -272,6 +272,7 @@ def to_json(project):
     "discoveredpackage": "PACKAGES",
     "discovereddependency": "DEPENDENCIES",
     "codebaseresource": "RESOURCES",
+    "codebaserelation": "RELATIONS",
     "projecterror": "ERRORS",
 }
 
@@ -447,6 +448,7 @@ def to_xlsx(project):
         "discoveredpackage",
         "discovereddependency",
         "codebaseresource",
+        "codebaserelation",
         "projecterror",
     ]
 

diff --git a/scanpipe/tests/pipes/test_output.py b/scanpipe/tests/pipes/test_output.py
@@ -194,7 +194,7 @@ def test_scanpipe_pipes_outputs_to_xlsx(self):
 
         # Make sure the output can be generated even if the work_directory was wiped
         shutil.rmtree(project.work_directory)
-        with self.assertNumQueries(7):
+        with self.assertNumQueries(8):
             output_file = output.to_xlsx(project=project)
         self.assertIn(output_file.name, project.output_root)