From 20e0d1333bd72692f991323bca13e3297ef34825 Mon Sep 17 00:00:00 2001 From: Administrator Date: Sat, 5 Oct 2024 11:32:00 +0300 Subject: [PATCH] fix sample --- Main/views/MainView.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Main/views/MainView.py b/Main/views/MainView.py index eacfaae..bccaae3 100644 --- a/Main/views/MainView.py +++ b/Main/views/MainView.py @@ -36,11 +36,11 @@ class MainView(BaseView): all_tasks = Task.objects.filter(solution__user=self.request.user).distinct() ok_tasks = all_tasks.filter(solution__result="OK").distinct() undone_tasks = set(all_tasks) - set(ok_tasks) - self.context['undone_tasks'] = sample(undone_tasks, k=min(5, len(undone_tasks))) + self.context['undone_tasks'] = sample(list(undone_tasks), k=min(5, len(undone_tasks))) for task in self.context['undone_tasks']: setattr(task, 'solution', Solution.objects.filter(user=self.request.user, task=task).last()) new_tasks = set(Task.objects.filter(public=True)) - set(all_tasks) - self.context['new_tasks'] = sample(new_tasks, k=min(5, len(new_tasks))) + self.context['new_tasks'] = sample(list(new_tasks), k=min(5, len(new_tasks))) self.context['groups'] = Group.objects.filter( Q(editors__in=self.request.user.username) | Q(creator=self.request.user) | Q( users=self.request.user)).distinct()