From 082d24acefeb8d5b94537a87770c604d3a0bf717 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=90=D0=BD=D0=B4=D1=80=D0=B5=D0=B9=20=D0=A1=D1=83=D0=BC?=
 =?UTF-8?q?=D0=B8=D0=BD?= <dreamonovich@yandex.ru>
Date: Mon, 3 Mar 2025 00:30:17 +0300
Subject: [PATCH] fix: fixed data generation (generatuion sumbissions for all
 types of task)

---
 .../core/management/commands/generate_data.py | 45 ++++++++++---------
 1 file changed, 23 insertions(+), 22 deletions(-)

diff --git a/services/backend/apps/core/management/commands/generate_data.py b/services/backend/apps/core/management/commands/generate_data.py
index 4d2a916..d6e6157 100644
--- a/services/backend/apps/core/management/commands/generate_data.py
+++ b/services/backend/apps/core/management/commands/generate_data.py
@@ -132,28 +132,29 @@ class Command(BaseCommand):
 
     def create_submissions(self, tasks, users):
         for task in tasks:
-            # Each task will get between 1 and 3 submissions
-            num_submissions = random.randint(1, 3)
-            for _ in range(num_submissions):
-                user = random.choice(users)
-                # Create a dummy content file
-                dummy_content = ContentFile(
-                    b"Submission content",
-                    name=f"submission_{uuid.uuid4().hex}.txt",
-                )
-                submission = CompetitionTaskSubmission.objects.create(
-                    user=user,
-                    task=task,
-                    earned_points=random.randint(
-                        0, task.points if task.points else 10
-                    ),
-                    content=dummy_content,
-                )
-                submission.save()
-                submission.send_on_review()
-                self.stdout.write(
-                    f"Created submission for task '{task.title}' by user '{user.username}'"
-                )
+            if task.type == CompetitionTask.CompetitionTaskType.REVIEW.value:
+                # Each task will get between 1 and 3 submissions
+                num_submissions = random.randint(1, 3)
+                for _ in range(num_submissions):
+                    user = random.choice(users)
+                    # Create a dummy content file
+                    dummy_content = ContentFile(
+                        b"Submission content",
+                        name=f"submission_{uuid.uuid4().hex}.txt",
+                    )
+                    submission = CompetitionTaskSubmission.objects.create(
+                        user=user,
+                        task=task,
+                        earned_points=random.randint(
+                            0, task.points if task.points else 10
+                        ),
+                        content=dummy_content,
+                    )
+                    submission.save()
+                    submission.send_on_review()
+                    self.stdout.write(
+                        f"Created submission for task '{task.title}' by user '{user.username}'"
+                    )
 
     def create_states(self, competitions, users):
         # For each competition, create a State for some of its participants