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