X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/f4780d18f68d0608ae164a84f6f530dd0e2d35d2..6ddc5071a71f2388b9a6b9b1f8261e591cfe66ab:/wtem/management/commands/wtem_assign_submissions.py diff --git a/wtem/management/commands/wtem_assign_submissions.py b/wtem/management/commands/wtem_assign_submissions.py index 2ec816e..dcf0466 100644 --- a/wtem/management/commands/wtem_assign_submissions.py +++ b/wtem/management/commands/wtem_assign_submissions.py @@ -1,30 +1,54 @@ -from django.core.management.base import BaseCommand, CommandError +# -*- coding: utf-8 -*- +from optparse import make_option + +from django.core.management.base import BaseCommand from django.db.models import Count from django.contrib.auth.models import User -from contact.models import Contact -from wtem.models import Submission +from wtem.models import Submission, Attachment class Command(BaseCommand): + option_list = BaseCommand.option_list + ( + make_option( + '--with-attachments-only', + action='store_true', + dest='attachments_only', + default=False, + help='Take into account only submissions with attachments'), + make_option( + '--without-attachments-only', + action='store_true', + dest='no_attachments_only', + default=False, + help='Take into account only submissions without attachments'), + ) + def handle(self, *args, **options): - how_many = int(args[0]) - examiner_names = args[1:] - users = User.objects.filter(username__in = examiner_names) - submissions_query = Submission.objects.annotate(examiners_count = Count('examiners')) + limit_from = int(args[0]) + limit_to = int(args[1]) + examiner_names = args[2:] + + users = User.objects.filter(username__in=examiner_names) + all_submissions = Submission.objects.annotate(examiners_count=Count('examiners')) + + submissions = all_submissions.exclude(answers=None) - submissions = submissions_query \ - .filter(examiners_count__lt=2).exclude(answers = None)[0:how_many] + with_attachment_ids = Attachment.objects.values_list('submission_id', flat=True).all() + if options['attachments_only']: + submissions = submissions.filter(id__in=with_attachment_ids) + if options['no_attachments_only']: + submissions = submissions.exclude(id__in=with_attachment_ids) - for submission in submissions: + for submission in submissions.order_by('id')[limit_from:limit_to]: submission.examiners.add(*users) submission.save() self.stdout.write('added to %s:%s' % (submission.id, submission.email)) count_by_examiners = dict() - for submission in submissions_query.all(): + for submission in all_submissions.all(): count_by_examiners[submission.examiners_count] = \ count_by_examiners.get(submission.examiners_count, 0) + 1 self.stdout.write('%s' % count_by_examiners)