fad2d3e8861b86f0c33dd3aa64ac5ad6344f309d
[edumed.git] / wtem / management / commands / wtem_assign_submissions.py
1 from optparse import make_option
2
3 from django.core.management.base import BaseCommand, CommandError
4 from django.db.models import Count
5 from django.contrib.auth.models import User
6
7 from contact.models import Contact
8 from wtem.models import Submission, Attachment
9
10
11 class Command(BaseCommand):
12
13     option_list = BaseCommand.option_list + (
14         make_option('--with-attachments-only',
15             action='store_true',
16             dest='attachments_only',
17             default=False,
18             help='Take into account only submissions with attachments'),
19         )
20
21     def handle(self, *args, **options):
22         how_many = int(args[0])
23         examiner_names = args[1:]
24
25
26         users = User.objects.filter(username__in = examiner_names)
27         submissions_query = Submission.objects.annotate(examiners_count = Count('examiners'))
28
29         submissions = submissions_query \
30             .filter(examiners_count__lt=2).exclude(answers = None)
31         
32         if options['attachments_only']:
33             with_attachment_ids = Attachment.objects.values_list('submission_id', flat=True).all()
34             submissions = submissions.filter(id__in = with_attachment_ids)
35
36         for submission in submissions.order_by('id')[0:how_many]:
37             submission.examiners.add(*users)
38             submission.save()
39             self.stdout.write('added to %s:%s' % (submission.id, submission.email))
40
41         count_by_examiners = dict()
42         for submission in submissions_query.all():
43             count_by_examiners[submission.examiners_count] = \
44                 count_by_examiners.get(submission.examiners_count, 0) + 1
45         self.stdout.write('%s' % count_by_examiners)