<type>(scope): <description>

[body]

[footer(s)]

services/backend/apps/core/management/commands/generate_pretty_data.py

@@ -14,7 +14,8 @@ from apps.review.models import Reviewer
 from apps.task.models import (
     CompetitionTask,
     CompetitionTaskCriteria,
-    CompetitionTaskSubmission, CompetitionTaskAttachment,
+    CompetitionTaskSubmission,
+    CompetitionTaskAttachment,
 )
 from apps.user.models import User, UserRole
 
@@ -44,6 +45,15 @@ dataset2 = ContentFile(
     name=f"dataset-{uuid.uuid4().hex}.csv",
 )
 
+correct_answer_file = ContentFile(
+    b"42",
+    name=f"answer.txt",
+)
+correct2_answer_file = ContentFile(
+    b"it is a dataset",
+    name=f"answer.txt",
+)
+
 now = timezone.now()
 
 image_dir = f"{settings.BASE_DIR}/apps/core/contents/images"
@@ -103,8 +113,9 @@ E — коэффициент чувствительности количеств
             {
                 "obj": None,
                 "title": "Задача 2",
-                "description": "Найдите максимальную зарплату программиста из датасета на питоне",
+                "description": "Найдите максимальную зарплату программиста из датасета на питоне. Программа должна вывести содержимое файла по пути /dataset",
                 "type": CompetitionTask.CompetitionTaskType.CHECKER.value,
+                "correct_answer_file": correct2_answer_file,
                 "attachment": dataset,
                 "attachment_path": "/dataset",
                 "points": 25,
@@ -317,9 +328,10 @@ B — пользователи, которым доступен только о
 и его характеристики. Однако оплатить товар можно и без захода на карточку
 товара.
 Задача: сравните группы по каждой метрике и сделайте вывод о том, стоит ли
-продолжить внедрение обновленного магазина или нужно вернуть старый
+продолжить внедрение обновленного магазина или нужно вернуть старый. Ответ 42.
                 """.strip(),
                 "type": CompetitionTask.CompetitionTaskType.CHECKER.value,
+                "correct_answer_file": correct_answer_file,
                 "points": 30,
                 "submission_reviewers_count": 2,
             },
@@ -338,7 +350,7 @@ B — пользователи, которым доступен только о
         "tasks": [
             {
                 "obj": None,
-                "title": "Анализ трендов", # TODO сюда добавить бд
+                "title": "Анализ трендов",  # TODO сюда добавить бд
                 "description": """ 
 Скачайте базу данных со специальной страницы (https://dano.hse.ru/data), изучите ее более
 внимательно: посмотрите на переменные, посчитайте описательные статистики, постройте
@@ -414,7 +426,7 @@ B — пользователи, которым доступен только о
                 "points": 15,
                 "submission_reviewers_count": 2,
                 "max_attempts": 50,
-                "correct_answer_file": ans3
+                "correct_answer_file": ans3,
             },
             {
                 "obj": None,
@@ -601,6 +613,7 @@ reviewers = [
     },
 ]
 
+
 class Command(BaseCommand):
     help = "Generate sample data for Users, Competitions, Tasks, Submissions, and States."
 
@@ -619,9 +632,9 @@ class Command(BaseCommand):
     def create_reviewers(self, count):
         reviewers_objs = []
         for reviewer in reviewers:
-            name = reviewer['name']
+            name = reviewer["name"]
-            surname = reviewer['surname']
+            surname = reviewer["surname"]
-            token = reviewer['token']
+            token = reviewer["token"]
             reviewer_obj = Reviewer(name=name, surname=surname, token=token)
             reviewer_obj.save()
             reviewers_objs.append(reviewer_obj)
@@ -631,11 +644,11 @@ class Command(BaseCommand):
         users_objs = []
         for user in users:
             user_obj, created = User.objects.get_or_create(
-                email=user['email'],
+                email=user["email"],
                 defaults={
-                    "username": user['username'],
+                    "username": user["username"],
-                    "password": make_password(user['password']),
+                    "password": make_password(user["password"]),
-                    "status": user['role'],
+                    "status": user["role"],
                 },
             )
             users_objs.append(user_obj)
@@ -647,19 +660,19 @@ class Command(BaseCommand):
 
         for i, competition in enumerate(competitions):
             competition_obj = Competition.objects.create(
-                title=competition['title'],
+                title=competition["title"],
-                description=competition['description'],
+                description=competition["description"],
-                start_date=competition['start_date'],
+                start_date=competition["start_date"],
-                end_date=competition['end_date'],
+                end_date=competition["end_date"],
-                type=competition['type'],
+                type=competition["type"],
-                participation_type=competition['participation_type'],
+                participation_type=competition["participation_type"],
             )
 
             if competition.get("image"):
-                competition_obj.image_url = competition['image']
+                competition_obj.image_url = competition["image"]
                 competition_obj.save()
 
-            competitions[i]['obj'] = competition_obj
+            competitions[i]["obj"] = competition_obj
             competition_obj.participants.add(*users)
             competitions_objs.append(competition_obj)
             self.stdout.write(f"Created competition: {competition['title']}")
@@ -673,45 +686,56 @@ class Command(BaseCommand):
             CompetitionTask.CompetitionTaskType.INPUT.value,
         ]
         for i, competition in enumerate(competitions):
-            for j, task in enumerate(competition['tasks']):
+            for j, task in enumerate(competition["tasks"]):
                 task_obj = CompetitionTask.objects.create(
-                    in_competition_position=j+1,
+                    in_competition_position=j + 1,
-                    competition=competition['obj'],
+                    competition=competition["obj"],
-                    title=task['title'],
+                    title=task["title"],
-                    description=task['description'],
+                    description=task["description"],
-                    type=task['type'],
+                    type=task["type"],
-                    points=task['points'],
+                    points=task["points"],
-                    submission_reviewers_count=task['submission_reviewers_count'],
+                    submission_reviewers_count=task[
-                    max_attempts=task.get('max_attempts'),
+                        "submission_reviewers_count"
+                    ],
+                    max_attempts=task.get("max_attempts"),
                 )
-                competitions[i]['tasks'][j]['obj'] = task_obj
+                competitions[i]["tasks"][j]["obj"] = task_obj
 
-                if task['type'] == CompetitionTask.CompetitionTaskType.INPUT.value:
+                if (
-                    task_obj.correct_answer_file = task['correct_answer_file']
+                    task["type"]
+                    == CompetitionTask.CompetitionTaskType.INPUT.value
+                ):
+                    task_obj.correct_answer_file = task["correct_answer_file"]
                 if task.get("attachment"):
                     CompetitionTaskAttachment.objects.create(
                         task=task_obj,
-                        file=task['attachment'],
+                        file=task["attachment"],
-                        bind_at=task['attachment_path'],
+                        bind_at=task["attachment_path"],
-                        public=True
+                        public=True,
                     )
 
                 if (
-                    task['type']
+                    task["type"]
                     == CompetitionTask.CompetitionTaskType.REVIEW.value
                 ):
-                    for k, criteria in enumerate(task['criteries']):
+                    for k, criteria in enumerate(task["criteries"]):
                         criteria_obj = CompetitionTaskCriteria.objects.create(
                             task=task_obj,
-                            name=criteria['name'],
+                            name=criteria["name"],
-                            slug=criteria['slug'],
+                            slug=criteria["slug"],
-                            description=criteria['description'],
+                            description=criteria["description"],
-                            max_value=criteria['max_value'],
+                            max_value=criteria["max_value"],
+                        )
+                        competitions[i]["tasks"][j]["criteries"][k]["obj"] = (
+                            criteria_obj
+                        )
+                        self.stdout.write(
+                            f"Created criteria: {criteria['slug']}"
                         )
-                        competitions[i]['tasks'][j]['criteries'][k]['obj'] = criteria_obj
-                        self.stdout.write(f"Created criteria: {criteria['slug']}")
                 tasks_objs.append(task_obj)
-                self.stdout.write(f"Created task: {task['title']} (type: {task['type']})")
+                self.stdout.write(
+                    f"Created task: {task['title']} (type: {task['type']})"
+                )
         self.add_reviewers_to_task(tasks_objs)
         return tasks_objs
 
@@ -723,22 +747,29 @@ class Command(BaseCommand):
     def create_incorrect_submissions(self, tasks, users):
         for user in users:
             for task in tasks:
-                if task.type == CompetitionTask.CompetitionTaskType.REVIEW.value:
+                if (
+                    task.type
+                    == CompetitionTask.CompetitionTaskType.REVIEW.value
+                ):
                     num_submissions = random.randint(1, 3)
                     for m in range(num_submissions):
                         dummy_content_txt = ContentFile(
-                            b"otvet: 112 sto proc" ,
+                            b"otvet: 112 sto proc",
                             name=f"submission_{uuid.uuid4().hex}.txt",
                         )
 
                         content_dir = f"{settings.BASE_DIR}/apps/core/contents"
-                        with open(f"{content_dir}/presentation.pptx", "rb") as f:
+                        with open(
+                            f"{content_dir}/presentation.pptx", "rb"
+                        ) as f:
                             pptx = File(f, name="presentation.pptx")
                             files = [pptx, pptx, dummy_content_txt]
-                            submission = CompetitionTaskSubmission.objects.create(
+                            submission = (
-                                user=user,
+                                CompetitionTaskSubmission.objects.create(
-                                task=task,
+                                    user=user,
-                                content=random.choice(files),
+                                    task=task,
+                                    content=random.choice(files),
+                                )
                             )
                         submission.save()
                         submission.send_on_review()
@@ -754,9 +785,10 @@ class Command(BaseCommand):
                     competition=comp,
                     defaults={
                         "state": "started",
-                        "changed_at": timezone.now() - timedelta(days=random.randint(1, 30)),
+                        "changed_at": timezone.now()
+                        - timedelta(days=random.randint(1, 30)),
                     },
                 )
                 self.stdout.write(
                     f"Created state '{state_obj.state}' for user '{user.username}' in competition '{comp.title}'"
-                )
+                )