stage2/views.py

   1 # -*- coding: utf-8 -*-
   2 from django.contrib.auth.decorators import login_required
   3 from django.core.urlresolvers import reverse
   4 from django.http import Http404
   5 from django.http.response import HttpResponseRedirect, HttpResponse, HttpResponseForbidden
   6 from django.shortcuts import get_object_or_404, render
   7 from django.utils import timezone
   8 from django.utils.cache import patch_cache_control
   9 from django.views.decorators.cache import never_cache
  10 from unidecode import unidecode
  11
  12 from stage2.forms import AttachmentForm, MarkForm, AssignmentFieldForm
  13 from stage2.models import Participant, Assignment, Answer, Attachment, Mark
  14
  15
  16 def all_assignments(participant, sent_forms):
  17     assignments = Assignment.objects.all()
  18     if sent_forms:
  19         sent_assignment, field_forms, attachment_forms = sent_forms
  20     else:
  21         sent_assignment = field_forms = attachment_forms = None
  22     for assignment in assignments:
  23         assignment.answer, created = Answer.objects.get_or_create(participant=participant, assignment=assignment)
  24         if assignment == sent_assignment:
  25             assignment.field_forms = field_forms
  26         else:
  27             assignment.field_forms = [
  28                 AssignmentFieldForm(label=label, field_no=i, options=options, answer=assignment.answer)
  29                 for i, (label, options) in enumerate(assignment.field_descriptions, 1)]
  30         # in theory, if assignment == sent_assignment, it should be copied like field_forms,
  31         # but somehow it doesn't work as expected
  32         assignment.attachment_forms = [
  33             (AttachmentForm(assignment=assignment, file_no=i, label=label, options=options),
  34              assignment.answer.attachment_set.filter(file_no=i).first() if assignment.answer else None)
  35             for i, (label, options) in enumerate(assignment.file_descriptions, 1)]
  36     return assignments
  37
  38
  39 @never_cache
  40 def participant_view(request, participant_id, key):
  41     participant = get_object_or_404(Participant, id=participant_id)
  42     if not participant.check(key):
  43         raise Http404
  44     if request.POST:
  45         # ugly :/
  46         assignment_id = None
  47         for post_key, value in request.POST.iteritems():
  48             if post_key.endswith('assignment_id'):
  49                 assignment_id = int(value)
  50         assert assignment_id
  51
  52         assignment = get_object_or_404(Assignment, id=assignment_id)
  53         now = timezone.now()
  54         if assignment.deadline < now:
  55             return HttpResponseForbidden('Not Allowed')
  56         attachments_valid, attachment_forms = get_attachment_forms(assignment, participant, request)
  57         fields_valid, field_forms = get_field_forms(assignment, participant, request)
  58         # tutaj w zasadzie powinno być też sprawdzenie attachments_valid, ale to powoduje,
  59         # że jeśli jakiś plik nie został wysłany, to traktujemy to jako błąd, a tego nie chcemy.
  60         # trzeba by znaleźć sensowny sposób odrózniania błędnego pliku od braku pliku.
  61         # na szczęście pliki walidujemy też javascriptem, więc jakoś ujdzie
  62         if fields_valid:
  63             return HttpResponseRedirect(reverse('stage2_participant', args=(participant_id, key)))
  64         else:
  65             sent_forms = (assignment, field_forms, attachment_forms)
  66     else:
  67         sent_forms = None
  68     response = render(request, 'stage2/participant.html', {
  69         'participant': participant,
  70         'assignments': all_assignments(participant, sent_forms)})
  71     # not needed in Django 1.8
  72     patch_cache_control(response, no_cache=True, no_store=True, must_revalidate=True)
  73     return response
  74
  75
  76 def get_attachment_forms(assignment, participant, request):
  77     all_valid = True
  78     attachment_forms = []
  79     for i, (label, options) in enumerate(assignment.file_descriptions, 1):
  80         answer, created = Answer.objects.get_or_create(participant=participant, assignment=assignment)
  81         attachment, created = Attachment.objects.get_or_create(answer=answer, file_no=i)
  82         form = AttachmentForm(
  83             data=request.POST, files=request.FILES,
  84             assignment=assignment, file_no=i, label=label, instance=attachment, options=options)
  85         if form.is_valid():
  86             form.save()
  87         else:
  88             all_valid = False
  89         attachment_forms.append(form)
  90     return all_valid, attachment_forms
  91
  92
  93 def get_field_forms(assignment, participant, request):
  94     all_valid = True
  95     field_forms = []
  96     for i, (label, options) in enumerate(assignment.field_descriptions, 1):
  97         answer = Answer.objects.get(participant=participant, assignment=assignment)
  98         form = AssignmentFieldForm(data=request.POST, label=label, field_no=i, options=options, answer=answer)
  99         if form.is_valid():
 100             form.save()
 101         else:
 102             all_valid = False
 103         field_forms.append(form)
 104     return all_valid, field_forms
 105
 106
 107 def attachment_download(attachment):
 108     response = HttpResponse(content_type='application/force-download')
 109     response.write(attachment.file.read())
 110     # workaround to this: https://code.djangoproject.com/ticket/20889
 111     response['Content-Disposition'] = 'attachment; filename="%s"' % unidecode(attachment.filename().replace('\n', ' '))
 112     response['Content-Length'] = response.tell()
 113     return response
 114
 115
 116 def get_file(request, assignment_id, file_no, participant_id, key):
 117     """We want to serve submitted files back to participants, but also validate their keys,
 118        so static files are not good"""
 119     participant = get_object_or_404(Participant, id=participant_id)
 120     if not participant.check(key):
 121         raise Http404
 122     assignment = get_object_or_404(Assignment, id=assignment_id)
 123     answer = get_object_or_404(Answer, participant=participant, assignment=assignment)
 124     attachment = get_object_or_404(Attachment, answer=answer, file_no=file_no)
 125     return attachment_download(attachment)
 126
 127
 128 @login_required
 129 def assignment_list(request):
 130     expert = request.user
 131     assignments = expert.stage2_assignments.all()
 132     if not assignments:
 133         return HttpResponseForbidden('Not allowed')
 134     for assignment in assignments:
 135         assignment.marked_count = assignment.available_answers(expert, marked=True).count()
 136         assignment.to_mark_count = assignment.available_answers(expert).count()
 137         assignment.supervisor = expert in assignment.supervisors.all()
 138         assignment.arbiter_count = assignment.needing_arbiter().count()
 139
 140     non_empty_assignments = [ass for ass in assignments if ass.marked_count > 0 or ass.to_mark_count > 0]
 141     if len(non_empty_assignments) == 1 and non_empty_assignments[0].to_mark_count > 0:
 142         return HttpResponseRedirect(reverse('stage2_answer_list', args=[non_empty_assignments[0].id]))
 143     return render(request, 'stage2/assignment_list.html', {'assignments': assignments})
 144
 145
 146 def available_answers(assignment, expert, sent_forms=None, marked=False):
 147     if marked:
 148         answers = assignment.available_answers(expert, marked=True)
 149     else:
 150         answers = assignment.available_answers(expert).order_by('participant__last_name')
 151     answers = answers.prefetch_related('attachment_set')
 152     if sent_forms:
 153         sent_answer_id, mark_forms = sent_forms
 154     else:
 155         sent_answer_id = mark_forms = None
 156     for answer in answers:
 157         attachments = answer.attachment_set.all()
 158         attachments_by_file_no = {attachment.file_no: attachment for attachment in attachments}
 159         answer.attachments = [
 160             (desc, attachments_by_file_no.get(i))
 161             for (i, (desc, ext)) in enumerate(assignment.file_descriptions, 1)]
 162         if answer.id == sent_answer_id:
 163             answer.forms = mark_forms
 164         else:
 165             answer.forms = []
 166             for criterion in assignment.markcriterion_set.all():
 167                 answer.forms.append(MarkForm(
 168                     answer=answer,
 169                     criterion=criterion,
 170                     instance=answer.mark_set.filter(expert=expert, criterion=criterion).first(),
 171                     prefix='mark%s-%s' % (answer.id, criterion.id)))
 172     return answers
 173
 174
 175 @login_required
 176 def answer_list(request, assignment_id, marked=False):
 177     assignment = get_object_or_404(Assignment, id=assignment_id)
 178     expert = request.user
 179     if expert not in assignment.experts.all():
 180         return HttpResponseForbidden('Not allowed')
 181     if request.POST:
 182         # ugly :/
 183         answer_id = None
 184         for post_key, value in request.POST.iteritems():
 185             if post_key.endswith('answer_id'):
 186                 answer_id = int(value)
 187         answer = get_object_or_404(Answer, id=answer_id)
 188
 189         if answer not in assignment.available_answers(expert, marked=marked):
 190             return HttpResponseForbidden('Not allowed')
 191         if answer.assignment.is_active():
 192             return HttpResponseForbidden('Not allowed')
 193         all_valid, forms = get_mark_forms(answer, request)
 194         if all_valid:
 195             if marked:
 196                 return HttpResponseRedirect(reverse('stage2_marked_answers', args=[answer.assignment_id]))
 197             else:
 198                 return HttpResponseRedirect(reverse('stage2_answer_list', args=[answer.assignment_id]))
 199         else:
 200             sent_forms = answer_id, forms
 201     else:
 202         sent_forms = None
 203     answers = available_answers(assignment, expert, sent_forms=sent_forms, marked=marked)
 204     return render(request, 'stage2/answer_list.html', {
 205         'answers': answers,
 206         'assignment': assignment,
 207         'field_counts': assignment.field_counts(answers) if not marked else None,
 208         'supervisor': expert in assignment.supervisors.all(),
 209         'marked': marked
 210     })
 211
 212
 213 def get_mark_forms(answer, request):
 214     all_valid = True
 215     created_marks = []
 216     forms = []
 217     for criterion in answer.assignment.markcriterion_set.all():
 218         mark, created = Mark.objects.get_or_create(
 219             answer=answer, criterion=criterion, expert=request.user, defaults={'points': 0})
 220         if created:
 221             created_marks.append(mark)
 222         form = MarkForm(
 223             data=request.POST, answer=answer, criterion=criterion, instance=mark,
 224             prefix='mark%s-%s' % (answer.id, criterion.id))
 225         if form.is_valid():
 226             form.save()
 227         else:
 228             all_valid = False
 229         forms.append(form)
 230     if not all_valid:
 231         for mark in created_marks:
 232             mark.delete()
 233     return all_valid, forms
 234
 235
 236 @login_required
 237 def expert_download(request, attachment_id):
 238     attachment = get_object_or_404(Attachment, id=attachment_id)
 239     return attachment_download(attachment)
 240
 241
 242 def write_row(row, writer):
 243     writer.writerow([unicode(item).encode('utf-8') for item in row])
 244
 245
 246 @login_required
 247 def csv_results(request):
 248     import csv
 249     response = HttpResponse(content_type='text/csv')
 250     writer = csv.writer(response)
 251     assignments = Assignment.objects.all()
 252     participants = Participant.objects.filter(complete_set=True)
 253     headers = [u'imię', u'nazwisko', u'szkoła', u'adres szkoły']
 254     assignments_experts = []
 255     for assignment in assignments:
 256         for expert in assignment.experts.filter(mark__answer__assignment=assignment).distinct():
 257             assignments_experts.append((assignment, expert))
 258             headers.append(u'%s %s' % (assignment.title, expert.last_name))
 259     for assignment in assignments:
 260         headers.append(u'%s - średnia' % assignment.title.encode('utf-8'))
 261     headers.append(u'ostateczny wynik')
 262     write_row(headers, writer)
 263     for participant in participants:
 264         row = [
 265             participant.first_name,
 266             participant.last_name,
 267             participant.contact.body['school'],
 268             participant.contact.body['school_address'],
 269         ]
 270         for assignment, expert in assignments_experts:
 271             marks = expert.mark_set.filter(answer__assignment=assignment, answer__participant=participant)
 272             if marks:
 273                 row.append(sum(mark.points for mark in marks))
 274             else:
 275                 row.append('')
 276         for assignment in assignments:
 277             row.append('%.2f' % participant.answer_set.get(assignment=assignment).score())
 278         row.append('%.2f' % participant.score())
 279         write_row(row, writer)
 280     response['Content-Disposition'] = 'attachment; filename="wyniki.csv"'
 281     return response
 282
 283
 284 @login_required
 285 def csv_details(request, assignment_id):
 286     import csv
 287     response = HttpResponse(content_type='text/csv')
 288     writer = csv.writer(response)
 289     assignment = get_object_or_404(Assignment, id=assignment_id)
 290     criteria = tuple(assignment.markcriterion_set.all())
 291     expert_headers = (u"ekspert",) + criteria
 292     headers = (u"imię", u"nazwisko", u"numer", u"wynik") + expert_headers * 2
 293     write_row(headers, writer)
 294
 295     for a in assignment.answer_set.filter(participant__complete_set=True).order_by('participant__last_name'):
 296         row = (
 297             a.participant.first_name,
 298             a.participant.last_name,
 299             str(a.participant.id),
 300             '%.2f' % a.score(),
 301         )
 302         experts = a.mark_set.values_list('expert__username', flat=True).order_by().distinct()
 303         for expert in experts:
 304             row += (expert,)
 305             marks = a.mark_set.filter(expert__username=expert).order_by('criterion__order')
 306             row += tuple(marks.values_list('points', flat=True))
 307         write_row(row, writer)
 308     response['Content-Disposition'] = 'attachment; filename="%s.csv"' % assignment.title
 309     return response