更新

2026-10-17 16:51:48 +08:00
parent 56f85f7d89
commit ba2e2e4087
41 changed files with 2851 additions and 96 deletions
@@ -0,0 +1,111 @@
+"""Capture agreement authorization states using synthetic data and no network."""
+
+from __future__ import annotations
+
+import os
+from copy import deepcopy
+from pathlib import Path
+from typing import Any
+
+os.environ.setdefault("QT_QPA_PLATFORM", "offscreen")
+
+from PySide6.QtWidgets import QApplication
+from render_issued_prescription_ai import PreviewRepository
+
+from doctor_workstation.services import DemoDoctorRepository
+from doctor_workstation.ui.dialogs import issued_prescription_ai as ai
+from doctor_workstation.ui.pages import prescriptions
+from doctor_workstation.ui.theme import apply_theme
+
+
+class VisibilityPreviewRepository(DemoDoctorRepository, PreviewRepository):
+    def __init__(self) -> None:
+        DemoDoctorRepository.__init__(self)
+        PreviewRepository.__init__(self)
+        self.visible = True
+
+    def get_prescription_ai_capabilities(self) -> dict[str, Any]:
+        return {"agreement_visible": self.visible, "can_manage_visibility": True}
+
+    def list_prescription_ai_statuses(self, ids: list[int]) -> dict[str, Any]:
+        return {"enabled": True, "agreement_visible": self.visible, "items": [
+            {"prescription_id": value, "agreement_visible": self.visible, "status": "success",
+             "models": {"qwen": {"status": "success", "score": 68.5},
+                        "openai": {"status": "success", "score": 55.2}}} for value in ids]}
+
+    def list_prescription_ai_reports(self, **_params: Any) -> dict[str, Any]:
+        rows = [{**row, "agreement_visible": self.visible} for row in self.history()]
+        return {"agreement_visible": self.visible, "enabled": True, "lists": deepcopy(rows), "count": len(rows)}
+
+    def get_prescription_ai_report(self, _batch_id: int) -> dict[str, Any]:
+        return {**deepcopy(self.batch), "agreement_visible": self.visible}
+
+
+def main() -> None:
+    application = QApplication.instance() or QApplication([])
+    apply_theme(application)
+    output = Path(__file__).resolve().parents[2] / "artifacts" / "prescription-ai-visibility"
+    output.mkdir(parents=True, exist_ok=True)
+
+    def immediate(function: Any, **callbacks: Any) -> None:
+        try:
+            result = function()
+        except Exception as error:
+            if callbacks.get("on_error"):
+                callbacks["on_error"](error)
+            else:
+                raise
+        else:
+            if callbacks.get("on_success"):
+                callbacks["on_success"](result)
+        finally:
+            if callbacks.get("on_finished"):
+                callbacks["on_finished"]()
+
+    ai.run_async = immediate
+    prescriptions.run_async = immediate
+    repository = VisibilityPreviewRepository()
+
+    def capture(widget: Any, filename: str) -> None:
+        for _ in range(4):
+            application.processEvents()
+        assert widget.grab().save(str(output / filename))
+
+    page = prescriptions.PrescriptionsPage(repository, ["*"], {"name": "演示医生", "root": 1})
+    page.resize(1366, 800)
+    page.show()
+    capture(page, "list-authorized.png")
+    repository.visible = False
+    page.refresh()
+    capture(page, "list-restricted.png")
+    page.close()
+
+    repository.visible = True
+    dialog = ai.IssuedPrescriptionAiDialog(repository, ["*"], prescription_id=7556,
+                                          current_user={"name": "演示医生", "root": 1})
+    dialog.resize(1280, 860)
+    dialog.show()
+    capture(dialog, "report-authorized.png")
+    repository.visible = False
+    repository.batch["status"] = "partial"
+    repository.batch["models"]["openai"].update(
+        status="failed", report=None, candidate=None, comparison=None,
+        error_code="upstream_timeout", error_message="演示：模型连接超时，可重试。")
+    dialog.refresh()
+    dialog._goto("report")
+    capture(dialog, "report-restricted.png")
+    dialog._goto("overview")
+    capture(dialog, "review-restricted.png")
+    dialog._goto("per_herb")
+    capture(dialog, "candidate-restricted.png")
+    dialog.close()
+    statistics = ai.PrescriptionAiStatisticsDialog(repository, ["*"])
+    statistics.resize(1180, 860)
+    statistics.show()
+    capture(statistics, "statistics-restricted.png")
+    statistics.close()
+    print(output)
+
+
+if __name__ == "__main__":
+    main()