individual answers assignable to experts - some fixes

[edumed.git] / 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 b0f2a4f..dcf0466 100644 (file)
--- a/wtem/management/commands/wtem_assign_submissions.py
+++ b/wtem/management/commands/wtem_assign_submissions.py
@@ -1,22 +1,24 @@
+# -*- coding: utf-8 -*-

 from optparse import make_option
 
 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 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 + (
 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'),
             action='store_true',
             dest='attachments_only',
             default=False,
             help='Take into account only submissions with attachments'),

-        make_option('--without-attachments-only',
+        make_option(
+            '--without-attachments-only',

             action='store_true',
             dest='no_attachments_only',
             default=False,
             action='store_true',
             dest='no_attachments_only',
             default=False,


@@ -29,17 +31,16 @@ class Command(BaseCommand):
         limit_to = int(args[1])
         examiner_names = args[2:]
 
         limit_to = int(args[1])
         examiner_names = args[2:]
 

-        users = User.objects.filter(username__in = examiner_names)
-        submissions_query = Submission.objects.annotate(examiners_count = Count('examiners'))
+        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)
-        

         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()
         if options['attachments_only']:

-            submissions = submissions.filter(id__in = with_attachment_ids)
+            submissions = submissions.filter(id__in=with_attachment_ids)

         if options['no_attachments_only']:
         if options['no_attachments_only']:

-            submissions = submissions.exclude(id__in = with_attachment_ids)
+            submissions = submissions.exclude(id__in=with_attachment_ids)

 
         for submission in submissions.order_by('id')[limit_from:limit_to]:
             submission.examiners.add(*users)
 
         for submission in submissions.order_by('id')[limit_from:limit_to]:
             submission.examiners.add(*users)


@@ -47,7 +48,7 @@ class Command(BaseCommand):
             self.stdout.write('added to %s:%s' % (submission.id, submission.email))
 
         count_by_examiners = dict()
             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)
             count_by_examiners[submission.examiners_count] = \
                 count_by_examiners.get(submission.examiners_count, 0) + 1
         self.stdout.write('%s' % count_by_examiners)