Update scripts (#847)

cscenter · Jun 18, 2024 · 3d85ec7 · 3d85ec7
1 parent 9580bb3
commit 3d85ec7
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/apps/admission/management/commands/import_exam_results_from_csv.py b/apps/admission/management/commands/import_exam_results_from_csv.py
@@ -53,17 +53,14 @@ def handle(self, *args, **options):
                     score = Decimal(row[1])
 
                     try:
-                        applicant = Applicant.objects.get(yandex_login=yandex_login, campaign__in=campaigns)
+                        applicant = Applicant.objects.get(yandex_login=yandex_login, campaign__in=campaigns,
+                                                          status__in=before_exam)
                     except Applicant.DoesNotExist:
                         self.stdout.write(self.style.ERROR(f'Applicant with login {yandex_login} does not exist'))
                         continue
                     except Applicant.MultipleObjectsReturned:
                         self.stdout.write(self.style.ERROR(f'There are many applicants with login {yandex_login}'))
                         continue
-                    if applicant.status not in before_exam:
-                        self.stdout.write(self.style.ERROR(f'Invalid status for applicant {yandex_login}: '
-                                                           f'{applicant.status}'))
-                        continue
                     total_by_compaign[applicant.campaign] += 1
                     exam, created = Exam.objects.get_or_create(applicant=applicant,
                                                                defaults={