Organized

redmine_ticket_control_export.py

@@ -0,0 +1,67 @@
+#!/usr/bin/env python3
+"""Training/export helpers for local control tags."""
+
+from __future__ import annotations
+
+import json
+from pathlib import Path
+from typing import Any, Iterable
+
+
+def export_control_tag_jsonl(
+    *,
+    cache_dir: Path,
+    tags_payload: dict[str, Any],
+    issues_by_id: dict[int, dict[str, Any]],
+    tag: str | None = None,
+    out_path: Path,
+) -> int:
+    """Write one JSONL record per tagged ticket for later LLM refine / training.
+
+    Each line: {tag, issue_id, subject, status, assignee, contact_email,
+                properties, annotation, text_snippet}
+    """
+    out_path = Path(out_path)
+    out_path.parent.mkdir(parents=True, exist_ok=True)
+    tags = tags_payload.get("tags") if isinstance(tags_payload, dict) else {}
+    if not isinstance(tags, dict):
+        tags = {}
+    names = [tag] if tag else sorted(tags.keys())
+    count = 0
+    with out_path.open("w", encoding="utf-8") as fh:
+        for name in names:
+            bucket = tags.get(name) or {}
+            tickets = bucket.get("tickets") if isinstance(bucket, dict) else {}
+            if not isinstance(tickets, dict):
+                continue
+            for iid_s, meta in tickets.items():
+                try:
+                    iid = int(iid_s)
+                except (TypeError, ValueError):
+                    continue
+                issue = issues_by_id.get(iid) or {}
+                helpdesk_raw = issue.get("helpdesk_ticket")
+                helpdesk: dict[str, Any] = helpdesk_raw if isinstance(helpdesk_raw, dict) else {}
+                contact_raw = helpdesk.get("contact")
+                contact: dict[str, Any] = contact_raw if isinstance(contact_raw, dict) else {}
+                status_raw = issue.get("status")
+                status_obj: dict[str, Any] = status_raw if isinstance(status_raw, dict) else {}
+                assignee_raw = issue.get("assigned_to")
+                assignee_obj: dict[str, Any] = assignee_raw if isinstance(assignee_raw, dict) else {}
+                meta_d = meta if isinstance(meta, dict) else {}
+                rec = {
+                    "tag": name,
+                    "issue_id": iid,
+                    "subject": issue.get("subject"),
+                    "status": status_obj.get("name"),
+                    "assignee": assignee_obj.get("name"),
+                    "contact_email": contact.get("email") or helpdesk.get("from_address"),
+                    "properties": meta_d.get("properties") or [],
+                    "annotation": meta_d.get("annotation") or "",
+                    "text_snippet": (
+                        f"{issue.get('subject') or ''}\n{(issue.get('description') or '')[:500]}"
+                    ).strip(),
+                }
+                fh.write(json.dumps(rec, sort_keys=True) + "\n")
+                count += 1
+    return count