Skip to content

Commit 2e15d30

Browse files
committed
Fix id not fetch issue
1 parent b985485 commit 2e15d30

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

dref/views.py

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -71,9 +71,10 @@ def get_queryset(self):
7171
dref_list.append(_queryset)
7272
drefs = []
7373
for dref in dref_list:
74-
id = list(dref.values_list('id', flat=True))
75-
new_dref = Dref.objects.filter(id__in=id).first()
76-
drefs.append(new_dref.id)
74+
ids = list(dref.values_list('id', flat=True))
75+
for id in ids:
76+
new_dref = Dref.objects.filter(id=id).first()
77+
drefs.append(new_dref.id)
7778
if len(drefs):
7879
return queryset.filter(models.Q(id__in=drefs) | models.Q(created_by=user)).distinct()
7980
else:
@@ -137,9 +138,10 @@ def get_queryset(self):
137138
dref_op_list.append(_queryset)
138139
ops = []
139140
for op in dref_op_list:
140-
id = list(op.values_list('id', flat=True))
141-
new_dref_op = DrefOperationalUpdate.objects.filter(id__in=id).first()
142-
ops.append(new_dref_op.id)
141+
ids = list(op.values_list('id', flat=True))
142+
for id in ids:
143+
new_dref_op = DrefOperationalUpdate.objects.filter(id=id).first()
144+
ops.append(new_dref_op.id)
143145
if len(ops):
144146
return queryset.filter(models.Q(id__in=ops) | models.Q(created_by=user)).distinct()
145147
else:
@@ -188,9 +190,10 @@ def get_queryset(self):
188190
dref_final_list.append(_queryset)
189191
finals = []
190192
for final in dref_final_list:
191-
id = list(final.values_list('id', flat=True))
192-
new_final = DrefFinalReport.objects.filter(id__in=id).first()
193-
finals.append(new_final.id)
193+
ids = list(final.values_list('id', flat=True))
194+
for id in ids:
195+
new_final = DrefFinalReport.objects.filter(id__in=id).first()
196+
finals.append(new_final.id)
194197
if len(finals):
195198
return queryset.filter(models.Q(id__in=finals) | models.Q(created_by=user)).distinct()
196199
else:

0 commit comments

Comments
 (0)