diff --git a/apps/wpm/services.py b/apps/wpm/services.py index 74448541..4dab0779 100644 --- a/apps/wpm/services.py +++ b/apps/wpm/services.py @@ -69,8 +69,8 @@ def get_pcoal_heat(year_s: int, month_s: int, day_s: int): return pcoal_heat else: try: - qs = SfLog.objects.get(end_time__year=year_s, end_time__month=month_s, end_time__day=day_s, - mgroup__name='回转窑', shift__name='白班') + qs = SfLog.objects.get(work_date__year=year_s, work_date__month=month_s, work_date__day=day_s, + mgroup__name='回转窑', shift__name__in=['白班', '早班']) # hardcode if qs.pcoal_heat is None: qs.pcoal_heat = 0 qs.save()