Add serialization for the dataclass models

Signed-off-by: tdruez <[email protected]>

Author: tdruez

dje/outputs.py

@@ -6,6 +6,7 @@
 # See https://aboutcode.org for more information about AboutCode FOSS projects.
 #
 
+import dataclasses
 import json
 import re
 from datetime import UTC
@@ -406,3 +407,29 @@ def get_openvex_document(product):
         tooling=f"DejaCode-{dejacode_version}",
         statements=get_openvex_statements(product),
     )
+
+
+def to_json_key(field_name):
+    """
+    - field_id -> @id
+    - field_context -> @context
+    """
+    prefix = "field_"
+    if field_name.startswith(prefix):
+        return "@" + field_name.removeprefix(prefix)
+    return field_name
+
+
+def openvex_dict_factory(fields):
+    """Dict factory for dataclasses.asdict that converts field names to JSON keys."""
+    return {to_json_key(name): value for name, value in fields}
+
+
+def get_openvex_document_json(product, indent=2):
+    openvex_document = get_openvex_document(product)
+    openvex_document_dict = dataclasses.asdict(
+        openvex_document,
+        dict_factory=openvex_dict_factory,
+    )
+    openvex_document_json = json.dumps(openvex_document_dict, indent=indent)
+    return openvex_document_json

dje/views.py

@@ -2441,11 +2441,8 @@ class ExportOpenVEXView(
     BaseDetailView,
 ):
     def get(self, request, *args, **kwargs):
-        import json
-
         product = self.get_object()
-        openvex_document = outputs.get_openvex_document(product)
-        openvex_document_json = json.dumps(openvex_document.__dict__, indent=2)
+        openvex_document_json = outputs.get_openvex_document_json(product)
         filename = outputs.get_filename(product, extension="openvex.json")
 
         return outputs.get_attachment_response(