From f53aff4bfcfc28d29ca81a5006bc5b5ecdc41eea Mon Sep 17 00:00:00 2001 From: Kyrylo Kireiev <90455454+KyryloKireiev@users.noreply.github.com> Date: Tue, 23 Apr 2024 23:27:58 +0300 Subject: [PATCH] refactor: [AXM-288] Refactor get_course_assignments Co-authored-by: monteri <36768631+monteri@users.noreply.github.com> --- lms/djangoapps/mobile_api/users/serializers.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/lms/djangoapps/mobile_api/users/serializers.py b/lms/djangoapps/mobile_api/users/serializers.py index b14719a7cb5c..4e89c339d3f8 100644 --- a/lms/djangoapps/mobile_api/users/serializers.py +++ b/lms/djangoapps/mobile_api/users/serializers.py @@ -279,11 +279,13 @@ def get_course_assignments(self, model: CourseEnrollment) -> Dict[str, Optional[ if not assignment.complete: next_assignment_all.append(assignment) - if next_assignment_all: - future_assignment_date = next_assignment_all[0].date.date() - next_assignment = [ - assignment for assignment in next_assignment_all if assignment.date.date() == future_assignment_date - ] +if future_assignments: + future_assignment_date = future_assignments[0].date.date() + next_assignments = [ + assignment for assignment in future_assignments if assignment.date.date() == future_assignment_date + ] + else: + next_assignments = [] return { 'future_assignments': DateSummarySerializer(next_assignment, many=True).data,