7ff1af20e1f6b4ca97e5137cc809ce60e1690fd5
[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         make_option('--without-attachments-only',
20             action='store_true',
21             dest='no_attachments_only',
22             default=False,
23             help='Take into account only submissions without attachments'),
24         )
25
26     def handle(self, *args, **options):
27
28         limit_from = int(args[0])
29         limit_to = int(args[1])
30         examiner_names = args[2:]
31
32         users = User.objects.filter(username__in = examiner_names)
33         submissions_query = Submission.objects.annotate(examiners_count = Count('examiners'))
34
35         submissions = submissions_query.exclude(answers = None)
36         
37         with_attachment_ids = Attachment.objects.values_list('submission_id', flat=True).all()
38         if options['attachments_only']:
39             submissions = submissions.filter(id__in = with_attachment_ids)
40         if options['no_attachments_only']:
41             submissions = submissions.exclude(id__in = with_attachment_ids)
42
43         for submission in submissions.order_by('id')[limit_from:limit_to]:
44             submission.examiners.add(*users)
45             submission.save()
46             self.stdout.write('added to %s:%s' % (submission.id, submission.email))
47
48         count_by_examiners = dict()
49         for submission in submissions_query.all():
50             count_by_examiners[submission.examiners_count] = \
51                 count_by_examiners.get(submission.examiners_count, 0) + 1
52         self.stdout.write('%s' % count_by_examiners)