fix: fixed data generation (generatuion sumbissions for all types of task)

This commit is contained in:
Андрей Сумин
2025-03-03 00:30:17 +03:00
parent 219cba64c0
commit 082d24acef
@@ -132,28 +132,29 @@ class Command(BaseCommand):
def create_submissions(self, tasks, users): def create_submissions(self, tasks, users):
for task in tasks: for task in tasks:
# Each task will get between 1 and 3 submissions if task.type == CompetitionTask.CompetitionTaskType.REVIEW.value:
num_submissions = random.randint(1, 3) # Each task will get between 1 and 3 submissions
for _ in range(num_submissions): num_submissions = random.randint(1, 3)
user = random.choice(users) for _ in range(num_submissions):
# Create a dummy content file user = random.choice(users)
dummy_content = ContentFile( # Create a dummy content file
b"Submission content", dummy_content = ContentFile(
name=f"submission_{uuid.uuid4().hex}.txt", b"Submission content",
) name=f"submission_{uuid.uuid4().hex}.txt",
submission = CompetitionTaskSubmission.objects.create( )
user=user, submission = CompetitionTaskSubmission.objects.create(
task=task, user=user,
earned_points=random.randint( task=task,
0, task.points if task.points else 10 earned_points=random.randint(
), 0, task.points if task.points else 10
content=dummy_content, ),
) content=dummy_content,
submission.save() )
submission.send_on_review() submission.save()
self.stdout.write( submission.send_on_review()
f"Created submission for task '{task.title}' by user '{user.username}'" self.stdout.write(
) f"Created submission for task '{task.title}' by user '{user.username}'"
)
def create_states(self, competitions, users): def create_states(self, competitions, users):
# For each competition, create a State for some of its participants # For each competition, create a State for some of its participants