Extend methods to generate Selection Schema for CVE and other projects

src/README.md

@@ -97,6 +97,28 @@ For usage in vulnerability management scenarios consider the following popular S
     from ssvc.decision_tables.helpers import ascii_tree
     print(ascii_tree(CISACoordinate))
 
+    #Creating an SSVC Selection for publish/export to external providers like CSAF or CVE
+    from datetime import datetime, timezone
+    from ssvc.decision_tables.cisa.cisa_coordinate_dt import LATEST as decision_table
+    from ssvc import selection
+    namespace = "ssvc"
+    decision_points = ["Exploitation"]
+    values = [["Public PoC"]]
+    timestamp = datetime.now()
+    selections = []
+
+    for dp in decision_table.decision_points.values():
+        if dp.namespace == namespace and dp.name in decision_points:
+           dp_index = decision_points.index(dp.name)
+           selected = selection.Selection.from_decision_point(dp)
+           selected.values = tuple(selection.MinimalDecisionPointValue(key=val.key,
+	   name=val.name) for val in dp.values if val.name in values[dp_index])
+           selections.append(selected)
+
+    out = selection.SelectionList(selections=selections,timestamp=timestamp)
+    print(out.model_dump_json(exclude_none=True, indent=4))
+
+
 Resources
 ---------
 

src/ssvc/selection.py

@@ -312,7 +312,55 @@ def model_json_schema(cls, **kwargs):
         schema = strip_nullable_anyof(schema)
 
         return order_schema(schema)
+    def _post_process(self, data):
+        """
+        Ensures all Selection.values are lists and removes empty array elements.
+        """
+        def fix_selection(selection):
+            # Convert tuple to list and filter out empty items
+            values = selection.get("values", [])
+            # Ensure it's a list, filter out empty/falsy items
+            selection["values"] = [v for v in list(values) if v]
+            return selection
+
+        # If this is a dict with selections, process each selection
+        if isinstance(data, dict) and "selections" in data:
+            data["selections"] = [
+                fix_selection(sel) for sel in data["selections"] if sel
+            ]
+        # Remove empty array fields from the top level
+        keys_to_delete = [k for k, v in data.items() if isinstance(v, list) and not v]
+        for k in keys_to_delete:
+            del data[k]
+        return data
 
+    def model_dump(self, *args, **kwargs):
+        data = super().model_dump(*args, **kwargs)
+        return self._post_process(data)
+
+    def model_dump_json(self, *args, **kwargs):
+        # Dump to python dict first, post-process, then dump to JSON
+        import json
+        from datetime import timezone
+        model_dump_kwargs = kwargs.copy()
+        json_kwargs = {}
+        # List of json.dumps kwargs you want to support
+        json_kwarg_names = ['indent', 'sort_keys', 'separators', 'ensure_ascii']
+        for key in json_kwarg_names:
+            if key in model_dump_kwargs:
+                json_kwargs[key] = model_dump_kwargs.pop(key)
+        # Get dict with Pydantic's processing (exclude_none, etc.)
+        data = super().model_dump(*args, **model_dump_kwargs)
+        data = self._post_process(data)
+        # Format timestamp as UTC RFC3339 string
+        if "timestamp" in data and isinstance(data["timestamp"], datetime):
+            ts = data["timestamp"]
+            if ts.tzinfo is None or ts.tzinfo.utcoffset(ts) is None:
+                utc_dt = ts.replace(tzinfo=timezone.utc)
+            else:
+                utc_dt = ts.astimezone(timezone.utc)
+            data["timestamp"] = utc_dt.strftime("%Y-%m-%dT%H:%M:%SZ")
+        return json.dumps(data, **json_kwargs)
 
 def main() -> None:
     print(