diff --git a/apps/wpm/views.py b/apps/wpm/views.py index d45191c3..de959487 100644 --- a/apps/wpm/views.py +++ b/apps/wpm/views.py @@ -75,7 +75,7 @@ class SfLogViewSet(UpdateModelMixin, CustomListModelMixin, DestroyModelMixin, Cu def add_info_for_list(self, data): sflogs = [i['id'] for i in data] - sflogs_dict = EnStat.objects.filter(sflog_id__in=sflogs).values('sflog_id', 'total_production','qua_data','production_hour','elec_consume_unit','run_rate','production_cost_unit') + sflogs_dict = EnStat.objects.filter(sflog_id__in=sflogs, type='sflog').values('sflog_id', 'total_production','qua_data','production_hour','elec_consume_unit','run_rate','production_cost_unit') # 创建索引 sflogs_dict = {x['sflog_id']: x for x in sflogs_dict} for item in data: