diff --git a/apps/wpm/services.py b/apps/wpm/services.py index c1061103..825fb63d 100644 --- a/apps/wpm/services.py +++ b/apps/wpm/services.py @@ -961,9 +961,9 @@ def mlog_audit_end(ticket: Ticket): mlog_submit(ins, ticket.create_by, now) -def get_batch_dag(batch_number: str): +def get_batch_dag(batch_number: str, version=1): try: - batch_ins = BatchSt.objects.get(batch=batch_number) + batch_ins = BatchSt.objects.get(batch=batch_number, version=version) except Exception: raise ParseError("该批次号未构建关系链") # 收集所有相关批次和边 diff --git a/apps/wpm/views.py b/apps/wpm/views.py index 2abd9cbb..653bab38 100644 --- a/apps/wpm/views.py +++ b/apps/wpm/views.py @@ -816,6 +816,7 @@ class BatchLogViewSet(ListModelMixin, CustomGenericViewSet): 获取该批次的DAG图数据 """ batch = request.data.get("batch", None) + version = request.data.get("version", 1) if not batch: raise ParseError("缺少batch参数") - return Response(get_batch_dag(batch)) \ No newline at end of file + return Response(get_batch_dag(batch, version)) \ No newline at end of file