from __future__ import annotations

from django.http import Http404, JsonResponse
from django.views.decorators.http import require_http_methods
from django.contrib.auth.decorators import login_required

from review_agent.models import RegulatoryReviewBatch, WorkflowNodeRun


@require_http_methods(["GET"])
@login_required
def batch_status(request, batch_id: int):
    batch = RegulatoryReviewBatch.objects.filter(pk=batch_id, user=request.user).first()
    if not batch:
        raise Http404("批次不存在。")
    nodes = WorkflowNodeRun.objects.filter(
        workflow_type="regulatory_review",
        workflow_batch_id=batch.pk,
    ).order_by("id")
    return JsonResponse(
        {
            "batch": {
                "id": batch.pk,
                "workflow_type": "regulatory_review",
                "batch_no": batch.batch_no,
                "status": batch.status,
                "source_summary_batch_id": batch.source_summary_batch_id,
                "risk_summary": batch.risk_summary,
                "error_message": batch.error_message,
            },
            "nodes": [
                {
                    "node_code": node.node_code,
                    "node_name": node.node_name,
                    "status": node.status,
                    "progress": node.progress,
                    "message": node.message,
                }
                for node in nodes
            ],
        }
    )