X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/729ebf7bb4bc619157b064e4e04dd9c1f0fbe8d9..8045f792684345fa3d82331c7d39696b6e2016f4:/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 6d33ec8..dcf0466 100644 --- a/wtem/management/commands/wtem_assign_submissions.py +++ b/wtem/management/commands/wtem_assign_submissions.py @@ -1,45 +1,54 @@ +# -*- coding: utf-8 -*- from optparse import make_option -from django.core.management.base import BaseCommand, CommandError +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, Attachment class Command(BaseCommand): option_list = BaseCommand.option_list + ( - make_option('--with-attachments-only', + 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:] + 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')) - users = User.objects.filter(username__in = examiner_names) - submissions_query = Submission.objects.annotate(examiners_count = Count('examiners')) + submissions = all_submissions.exclude(answers=None) - submissions = submissions_query \ - .filter(examiners_count__lt=2).exclude(answers = None) - + with_attachment_ids = Attachment.objects.values_list('submission_id', flat=True).all() if options['attachments_only']: - with_attachment_ids = Attachment.objects.values_list('submission_id', flat=True).all() - submissions = submissions.filter(id__in = with_attachment_ids) + 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[0:how_many]: + 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)