elspais 0.11.2__py3-none-any.whl → 0.43.5__py3-none-any.whl

elspais/graph/serialize.py

@@ -0,0 +1,216 @@
+"""Graph Serialization - Export TraceGraph to various formats.
+
+This module provides functions to serialize TraceGraph and GraphNode
+to JSON-compatible dicts, markdown, and CSV formats.
+"""
+
+from __future__ import annotations
+
+import csv
+import io
+from typing import TYPE_CHECKING, Any
+
+if TYPE_CHECKING:
+    from elspais.graph.builder import TraceGraph
+    from elspais.graph.GraphNode import GraphNode
+
+
+def serialize_node(node: GraphNode) -> dict[str, Any]:
+    """Serialize a GraphNode to a JSON-compatible dict.
+
+    Args:
+        node: The node to serialize.
+
+    Returns:
+        Dict suitable for JSON serialization.
+    """
+
+    result: dict[str, Any] = {
+        "id": node.id,
+        "kind": node.kind.name,
+        "label": node.get_label(),
+        "uuid": node.uuid,
+        "content": node.get_all_content(),
+    }
+
+    # Include source location if present
+    if node.source:
+        result["source"] = {
+            "path": node.source.path,
+            "line": node.source.line,
+            "end_line": node.source.end_line,
+        }
+        if node.source.repo:
+            result["source"]["repo"] = node.source.repo
+
+    # Include child IDs
+    children = list(node.iter_children())
+    if children:
+        result["children"] = [child.id for child in children]
+
+    # Include parent IDs
+    parents = list(node.iter_parents())
+    if parents:
+        result["parents"] = [parent.id for parent in parents]
+
+    # Include metrics
+    if node._metrics:
+        result["metrics"] = dict(node._metrics)
+
+    # Include outgoing edges
+    edges = list(node.iter_outgoing_edges())
+    if edges:
+        result["edges"] = [
+            {
+                "target": edge.target.id,
+                "kind": edge.kind.name,
+            }
+            for edge in edges
+        ]
+
+    return result
+
+
+def serialize_graph(graph: TraceGraph) -> dict[str, Any]:
+    """Serialize a TraceGraph to a JSON-compatible dict.
+
+    Args:
+        graph: The graph to serialize.
+
+    Returns:
+        Dict with nodes, roots, and metadata.
+    """
+
+    # Serialize all nodes and count by kind in single pass
+    nodes = {}
+    kind_counts: dict[str, int] = {}
+    for node in graph.all_nodes():
+        nodes[node.id] = serialize_node(node)
+        kind_name = node.kind.name
+        kind_counts[kind_name] = kind_counts.get(kind_name, 0) + 1
+
+    roots = list(graph.iter_roots())
+    return {
+        "nodes": nodes,
+        "roots": [root.id for root in roots],
+        "metadata": {
+            "node_count": len(nodes),
+            "root_count": len(roots),
+            "repo_root": str(graph.repo_root),
+            "by_kind": kind_counts,
+        },
+    }
+
+
+def to_markdown(graph: TraceGraph) -> str:
+    """Generate a markdown traceability matrix from graph.
+
+    Args:
+        graph: The TraceGraph to render.
+
+    Returns:
+        Markdown string with traceability matrix.
+    """
+    from elspais.graph import NodeKind
+
+    lines = [
+        "# Traceability Matrix",
+        "",
+        "| ID | Level | Title | Status | Implements |",
+        "|-----|-------|-------|--------|------------|",
+    ]
+
+    # Get requirements only
+    requirements = list(graph.nodes_by_kind(NodeKind.REQUIREMENT))
+
+    # Sort by ID
+    requirements.sort(key=lambda n: n.id)
+
+    for node in requirements:
+        req_id = node.id
+        level = node.get_field("level", "")
+        title = node.get_label() or ""
+        status = node.get_field("status", "")
+
+        # Get implements from incoming edges (what this node implements)
+        implements = []
+        for edge in node.iter_incoming_edges():
+            from elspais.graph.relations import EdgeKind
+
+            if edge.kind == EdgeKind.IMPLEMENTS:
+                implements.append(edge.source.id)
+
+        implements_str = ", ".join(implements) if implements else "-"
+
+        lines.append(f"| {req_id} | {level} | {title} | {status} | {implements_str} |")
+
+    lines.append("")
+    return "\n".join(lines)
+
+
+def to_csv(graph: TraceGraph) -> str:
+    """Generate a CSV export from graph.
+
+    Args:
+        graph: The TraceGraph to export.
+
+    Returns:
+        CSV string with requirement data.
+    """
+    from elspais.graph import NodeKind
+
+    output = io.StringIO()
+    writer = csv.writer(output, quoting=csv.QUOTE_MINIMAL)
+
+    # Header
+    writer.writerow(
+        [
+            "id",
+            "level",
+            "label",
+            "status",
+            "file_path",
+            "line",
+            "hash",
+            "implements",
+        ]
+    )
+
+    # Get requirements only
+    requirements = list(graph.nodes_by_kind(NodeKind.REQUIREMENT))
+    requirements.sort(key=lambda n: n.id)
+
+    for node in requirements:
+        file_path = node.source.path if node.source else ""
+        line = node.source.line if node.source else ""
+
+        # Get implements from incoming edges (what this node implements)
+        implements = []
+        for edge in node.iter_incoming_edges():
+            from elspais.graph.relations import EdgeKind
+
+            if edge.kind == EdgeKind.IMPLEMENTS:
+                implements.append(edge.source.id)
+
+        writer.writerow(
+            [
+                node.id,
+                node.get_field("level", ""),
+                node.get_label(),
+                node.get_field("status", ""),
+                file_path,
+                line,
+                node.get_field("hash", ""),
+                "; ".join(implements),
+            ]
+        )
+
+    return output.getvalue()
+
+
+__all__ = [
+    "serialize_node",
+    "serialize_graph",
+    "to_markdown",
+    "to_csv",
+]

elspais/html/__init__.py

@@ -0,0 +1,8 @@
+"""HTML Generation module for traceability reports.
+
+This module provides HTML generation for interactive traceability matrices.
+"""
+
+from elspais.html.generator import HTMLGenerator
+
+__all__ = ["HTMLGenerator"]