teams-transcripts 0.5.0__py3-none-any.whl

teams_transcripts/services/output.py

@@ -0,0 +1,84 @@
+"""Final output construction and emission for transcript downloads."""
+
+from __future__ import annotations
+
+import contextlib
+import json
+import os
+import sys
+import typing
+
+from teams_transcripts.domain.summary import build_result_summary
+from teams_transcripts.domain.vtt import parse_cues
+
+if typing.TYPE_CHECKING:
+    from teams_transcripts.domain.models import ChatMessage, MeetingConfig, MeetingParticipants
+
+
+def _build_config_dict(
+    config: MeetingConfig,
+    speaker_map: dict[str, str],
+    roster_members_raw: list[dict],
+    meeting_time_text: str,
+    *,
+    tenant_domain: str | None = None,
+) -> dict:
+    """Adapt ``MeetingConfig`` to the plain mapping used by domain functions."""
+    result: dict[str, object] = {
+        "meeting": config.meeting,
+        "date": config.date,
+        "time": config.time,
+        "output": config.output,
+        "output_to_stdout": config.output_to_stdout,
+        "speaker_map": speaker_map,
+        "roster_members": roster_members_raw,
+        "meeting_time_text": meeting_time_text,
+    }
+    if tenant_domain is not None:
+        result["tenant"] = tenant_domain
+    return result
+
+
+def _emit_output(
+    config: MeetingConfig,
+    config_dict: dict,
+    content: str,
+    num_parts: int,
+    verification: dict,
+    chat_count: int,
+    participants: MeetingParticipants | None = None,
+    *,
+    chat_messages: list[ChatMessage] | None = None,
+) -> None:
+    """Emit the final VTT or structured JSON to stdout or the output file."""
+    output_to_stdout = config.output_to_stdout
+
+    if config.format_json:
+        transcript_cues = parse_cues(content) if config.include_transcript else None
+        summary = build_result_summary(
+            config_dict,
+            num_parts=num_parts,
+            verification=verification,
+            chat_count=chat_count,
+            participants=participants,
+            chat_messages=chat_messages,
+            transcript_cues=transcript_cues,
+            include_transcript=config.include_transcript,
+            include_chat=config.include_chat,
+            include_metadata=config.include_metadata,
+        )
+
+        if output_to_stdout:
+            print(json.dumps(summary, ensure_ascii=False, indent=2), flush=True)
+        else:
+            assert config.output is not None
+            with open(config.output, "w") as output_file:
+                json.dump(summary, output_file, ensure_ascii=False, indent=2)
+                output_file.write("\n")
+    elif output_to_stdout:
+        sys.stdout.write(content)
+        sys.stdout.flush()
+
+    if output_to_stdout and config.output:
+        with contextlib.suppress(OSError):
+            os.remove(config.output)