Fix out of index in stat page

This commit is contained in:
cuom1999 2021-06-02 08:31:58 -05:00
parent 04a20626b8
commit 5c5bba5f7b

View file

@ -561,16 +561,16 @@ class ContestStats(TitleMixin, ContestMixin, DetailView):
result_data[category['code']][i] = category['count']
problem_points = [[] for _ in range(num_problems)]
point_count = queryset.values('contest__problem__order', 'contest__points')\
.annotate(count=Count('contest__points')) \
.filter(points__isnull=False) \
.order_by('contest__problem__order', 'contest__points')
point_count_queryset = list(queryset.values('problem__code', 'contest__points', 'contest__problem__points')
.annotate(count=Count('contest__points'))
.filter(points__isnull=False)
.order_by('problem__code', 'contest__points')
.values_list('problem__code', 'contest__points', 'contest__problem__points', 'count'))
counter = [[0 for _ in range(self.POINT_BIN + 1)] for _ in range(num_problems)]
for sub in point_count.iterator():
problem_idx = sub['contest__problem__order'] - 1
bin_idx = math.floor(sub['contest__points'] * self.POINT_BIN / 100)
counter[problem_idx][bin_idx] += sub['count']
for problem_code, point, max_point, count in point_count_queryset:
problem_idx = codes.index(problem_code)
bin_idx = math.floor(point * self.POINT_BIN / max_point)
counter[problem_idx][bin_idx] += count
for i in range(num_problems):
problem_points[i] = [(j * 100 / self.POINT_BIN, counter[i][j])
for j in range(len(counter[i]))]