From 0e8cce066975834d52891e4fc4f6b1cf0204f38b Mon Sep 17 00:00:00 2001
From: caoqianming <caoqianming@foxmail.com>
Date: Tue, 25 Mar 2025 09:54:57 +0800
Subject: [PATCH] feat: get_batch_dag bug2

---
 apps/wpm/services.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/apps/wpm/services.py b/apps/wpm/services.py
index 61609fb3..8e9872c1 100644
--- a/apps/wpm/services.py
+++ b/apps/wpm/services.py
@@ -915,7 +915,9 @@ def get_batch_dag(batch_number: str):
         while len(nodes_set) > prev_size:
             prev_size = len(nodes_set)
             # 查询所有与当前批次相关的记录（作为source或target）
-            logs = BatchLog.objects.filter(Q(source__id__in=nodes_set) | Q(target__ide__in=nodes_set)).order_by("create_time")
+            logs = BatchLog.objects.filter(Q(source__id__in=nodes_set) | Q(target__id__in=nodes_set)).select_related(
+                "source", "target"
+            ).order_by("create_time")
             # 处理每条记录，扩展节点和边
             for log in logs:
                 source = log.source.id