39 lines
1.6 KiB
Python
39 lines
1.6 KiB
Python
from Main.models import Solution
|
|
from SprintLib.BaseView import BaseView, AccessError
|
|
|
|
|
|
class SolutionsTableView(BaseView):
|
|
view_file = "solutions_table.html"
|
|
required_login = True
|
|
endpoint = "solutions_table"
|
|
page_size = 20
|
|
page = None
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.queryset = Solution.objects.all()
|
|
|
|
def pre_handle(self):
|
|
if 'page' not in self.request.GET:
|
|
raise AccessError()
|
|
self.page = int(self.request.GET['page'])
|
|
|
|
def get(self):
|
|
if 'teacher' in self.request.GET.keys():
|
|
if 'set_id' in self.request.GET.keys():
|
|
self.queryset = self.queryset.filter(set_id=self.request.GET['set_id'])
|
|
elif 'task_id' in self.request.GET.keys():
|
|
self.queryset = self.queryset.filter(task_id=self.request.GET['task_id'], set=None)
|
|
else:
|
|
raise AccessError()
|
|
else:
|
|
if hasattr(self.entities, 'setTask'):
|
|
self.queryset = self.queryset.filter(user=self.request.user, task=self.entities.setTask.task, set=self.entities.setTask.set)
|
|
else:
|
|
self.queryset = self.queryset.filter(user=self.request.user, task=self.entities.task, set=None)
|
|
offset = self.page_size * (self.page - 1)
|
|
limit = self.page_size
|
|
self.context["solutions"] = self.queryset.order_by("-id")[offset:offset + limit]
|
|
self.context["count_pages"] = range(1, (len(self.queryset) - 1) // self.page_size + 2)
|
|
self.context["need_pagination"] = len(self.context["count_pages"]) > 1
|