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 django.views.decorators.http import require_POST
 
  11 from unidecode import unidecode
 
  13 from stage2.forms import AttachmentForm, MarkForm
 
  14 from stage2.models import Participant, Assignment, Answer, Attachment, Mark
 
  17 def all_assignments(participant):
 
  18     assignments = Assignment.objects.all()
 
  19     for assignment in assignments:
 
  20         assignment.answer = assignment.answer_set.filter(participant=participant).first()
 
  22             (AttachmentForm(assignment=assignment, file_no=i, label=label, extensions=ext),
 
  23              assignment.answer.attachment_set.filter(file_no=i).first() if assignment.answer else None)
 
  24             for i, (label, ext) in enumerate(assignment.file_descriptions, 1)]
 
  29 def participant_view(request, participant_id, key):
 
  30     participant = get_object_or_404(Participant, id=participant_id)
 
  31     if not participant.check(key):
 
  33     response = render(request, 'stage2/participant.html', {
 
  34         'participant': participant,
 
  35         'assignments': all_assignments(participant)})
 
  36     # not needed in Django 1.8
 
  37     patch_cache_control(response, no_cache=True, no_store=True, must_revalidate=True)
 
  42 def upload(request, assignment_id, participant_id, key):
 
  43     participant = get_object_or_404(Participant, id=participant_id)
 
  44     if not participant.check(key):
 
  46     assignment = get_object_or_404(Assignment, id=assignment_id)
 
  48     if assignment.deadline < now:
 
  49         raise Http404  # TODO za późno
 
  50     for i, (label, ext) in enumerate(assignment.file_descriptions, 1):
 
  51         answer, created = Answer.objects.get_or_create(participant=participant, assignment=assignment)
 
  52         attachment, created = Attachment.objects.get_or_create(answer=answer, file_no=i)
 
  53         form = AttachmentForm(
 
  54             data=request.POST, files=request.FILES,
 
  55             assignment=assignment, file_no=i, label=label, instance=attachment, extensions=ext)
 
  58     return HttpResponseRedirect(reverse('stage2_participant', args=(participant_id, key)))
 
  61 def attachment_download(attachment):
 
  62     response = HttpResponse(content_type='application/force-download')
 
  63     response.write(attachment.file.read())
 
  64     # workaround to this: https://code.djangoproject.com/ticket/20889
 
  65     response['Content-Disposition'] = 'attachment; filename="%s"' % unidecode(attachment.filename().replace('\n', ' '))
 
  66     response['Content-Length'] = response.tell()
 
  70 def get_file(request, assignment_id, file_no, participant_id, key):
 
  71     """We want to serve submitted files back to participants, but also validate their keys,
 
  72        so static files are not good"""
 
  73     participant = get_object_or_404(Participant, id=participant_id)
 
  74     if not participant.check(key):
 
  76     assignment = get_object_or_404(Assignment, id=assignment_id)
 
  77     answer = get_object_or_404(Answer, participant=participant, assignment=assignment)
 
  78     attachment = get_object_or_404(Attachment, answer=answer, file_no=file_no)
 
  79     return attachment_download(attachment)
 
  83 def assignment_list(request):
 
  84     assignments = request.user.stage2_assignments.all()
 
  86         return HttpResponseForbidden('Not allowed')
 
  87     for assignment in assignments:
 
  88         assignment.marked_count = Mark.objects.filter(expert=request.user, answer__assignment=assignment).count()
 
  89         assignment.to_mark_count = assignment.available_answers(request.user).count()
 
  90         assignment.supervisor = request.user in assignment.supervisors.all()
 
  91         assignment.arbiter_count = assignment.answer_set.filter(need_arbiter=True).count()
 
  93     non_empty_assignments = [ass for ass in assignments if ass.marked_count > 0 or ass.to_mark_count > 0]
 
  94     if len(non_empty_assignments) == 1 and non_empty_assignments[0].to_mark_count > 0:
 
  95         return HttpResponseRedirect(reverse('stage2_answer_list', args=[non_empty_assignments[0].id]))
 
  96     return render(request, 'stage2/assignment_list.html', {'assignments': assignments})
 
  99 def available_answers(assignment, expert, answer_with_errors=None, form_with_errors=None, marked=False):
 
 101         answers = Answer.objects.filter(mark__expert=expert, assignment=assignment)
 
 103         answers = assignment.available_answers(expert)
 
 104     answers = answers.order_by('participant__last_name').prefetch_related('attachment_set')
 
 105     for answer in answers:
 
 106         attachments = answer.attachment_set.all()
 
 107         attachments_by_file_no = {attachment.file_no: attachment for attachment in attachments}
 
 108         answer.attachments = [
 
 109             (desc, attachments_by_file_no.get(i))
 
 110             for (i, (desc, ext)) in enumerate(assignment.file_descriptions, 1)]
 
 111         if answer == answer_with_errors:
 
 112             answer.form = form_with_errors
 
 114             answer.form = MarkForm(
 
 115                 answer=answer, instance=answer.mark_set.filter(expert=expert).first(), prefix='ans%s' % answer.id)
 
 120 def answer_list(request, assignment_id):
 
 121     assignment = get_object_or_404(Assignment, id=assignment_id)
 
 122     if request.user not in assignment.experts.all():
 
 123         return HttpResponseForbidden('Not allowed')
 
 124     return render(request, 'stage2/answer_list.html',
 
 125                   {'answers': available_answers(assignment, request.user), 'assignment': assignment})
 
 129 def marked_answer_list(request, assignment_id):
 
 130     assignment = get_object_or_404(Assignment, id=assignment_id)
 
 131     if request.user not in assignment.experts.all():
 
 132         return HttpResponseForbidden('Not allowed')
 
 133     return render(request, 'stage2/answer_list.html', {
 
 134         'answers': available_answers(assignment, request.user, marked=True),
 
 135         'assignment': assignment,
 
 141 def expert_download(request, attachment_id):
 
 142     attachment = get_object_or_404(Attachment, id=attachment_id)
 
 143     return attachment_download(attachment)
 
 148 def mark_answer(request, answer_id):
 
 149     answer = get_object_or_404(Answer, id=answer_id)
 
 150     if request.user not in answer.assignment.experts.all():
 
 151         return HttpResponseForbidden('Not allowed')
 
 152     if answer.assignment.is_active():
 
 153         return HttpResponseForbidden('Not allowed')
 
 154     mark, created = Mark.objects.get_or_create(answer=answer, expert=request.user, defaults={'points': 0})
 
 155     form = MarkForm(data=request.POST, answer=answer, instance=mark, prefix='ans%s' % answer.id)
 
 161     return HttpResponseRedirect(reverse(
 
 162         'stage2_answer_list' if created else 'stage2_marked_answers', args=[answer.assignment_id]))