NDOJ/judge/tasks/experiment.py

from judge.models import SubmissionTestCase, Problem

from collections import defaultdict


def generate_report(problem):
    testcases = SubmissionTestCase.objects.filter(submission__problem=problem).all()

    score = defaultdict(int)
    total = defaultdict(int)
    rate = defaultdict(int)

    for case in testcases.iterator():
        score[case.case] += int(case.status == "AC")
        total[case.case] += 1

    for i in score:
        rate[i] = score[i] / total[i]

    for i, _ in sorted(rate.items(), key=lambda x: x[1], reverse=True):
        print(i, score[i], total[i], rate[i])