lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add download AC submissions

Browse source
This commit is contained in:
cuom1999 2020-07-26 00:11:03 -05:00
parent e8ef46ffcf
commit 859c8f62ed
3 changed files with 27 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

21
judge/views/problem.py
View file

@ -4,6 +4,7 @@ import shutil
from datetime import timedelta
from operator import itemgetter
from random import randrange
import zipfile, tempfile
from django.conf import settings
from django.contrib.auth.decorators import login_required
@ -676,4 +677,22 @@ class ProblemClone(ProblemMixin, PermissionRequiredMixin, TitleMixin, SingleObje
problem.language_limits.set(language_limits)
problem.types.set(types)
return HttpResponseRedirect(reverse('admin:judge_problem_change', args=(problem.id,)))
return HttpResponseRedirect(reverse('admin:judge_problem_change', args=(problem.id,)))
def download_submissions(request, problem):
submissions = Submission.objects.filter(problem__code=problem, result='AC')
with tempfile.SpooledTemporaryFile() as tmp:
with zipfile.ZipFile(tmp, 'w', zipfile.ZIP_DEFLATED) as archive:
for submission in submissions:
file_name = str(submission.id) + '.' + str(submission.language.key)
archive.writestr(file_name, submission.source.source)
# Reset file pointer
tmp.seek(0)
# Write file data to response
response = HttpResponse(tmp.read(), content_type='application/x-zip-compressed')
response['Content-Disposition'] = 'attachment; filename="%s"' % (str(problem) + '_submissions.zip')
return response