Move to src dir.

[edumed.git] / src / wtem / management / commands / wtem_send_results_csv.py

diff --git a/src/wtem/management/commands/wtem_send_results_csv.py b/src/wtem/management/commands/wtem_send_results_csv.py
new file mode 100644 (file)
index 0000000..97a98eb
--- /dev/null
+++ b/src/wtem/management/commands/wtem_send_results_csv.py
@@ -0,0 +1,115 @@
+# -*- coding: utf-8 -*-
+
+from optparse import make_option
+
+from django.core.management.base import BaseCommand
+from django.template.loader import render_to_string
+from django.utils import translation
+
+from wtem.management.commands import send_mail
+from wtem.models import Submission
+
+
+def get_submissions():
+    return sorted(Submission.objects.exclude(answers=None).all(), key=lambda s: -s.final_result)
+
+minimum = 52
+
+
+class Command(BaseCommand):
+    args = 'csv_filename'
+
+    option_list = BaseCommand.option_list + (
+        make_option(
+            '--to-teachers',
+            action='store_true',
+            dest='to_teachers',
+            default=False,
+            help='Send emails to teachers'),
+        make_option(
+            '--to-students',
+            action='store_true',
+            dest='to_students',
+            default=False,
+            help='Send emails to students'),
+        make_option(
+            '--only-to',
+            action='store',
+            dest='only_to',
+            default=None,
+            help='Send emails only to listed addresses'),
+    )
+
+    def handle(self, csv_filename, *args, **options):
+        translation.activate('pl')
+        self.results = [line.decode('utf-8').strip('\n').split(',') for line in open(csv_filename)]
+        for target in ['to_teachers', 'to_students']:
+            if options[target]:
+                self.sent = 0
+                self.failed = 0
+                getattr(self, 'handle_' + target)(*args, **options)
+
+    def handle_to_students(self, *args, **options):
+        self.stdout.write('>>> Sending results to students')
+        subject = 'Wyniki II etapu Wielkiego Turnieju Edukacji Medialnej'
+
+        for email, last_name, first_name, final_result in self.results:
+            if options['only_to'] and email != options['only_to']:
+                continue
+            if final_result == 'dyskwalifikacja':
+                template = 'results_student_disqualified.txt'
+            elif float(final_result) < minimum:
+                template = 'results_student_failed.txt'
+            else:
+                template = 'results_student_passed.txt'
+            message = render_to_string('wtem/' + template, dict(final_result=final_result))
+            self.send_message(message, subject, email)
+
+        self.sum_up()
+
+    def handle_to_teachers(self, *args, **options):
+        self.stdout.write('>>> Sending results to teachers')
+        subject = 'Wyniki II etapu Wielkiego Turnieju Edukacji Medialnej'
+
+        submissions_by_contact = dict()
+
+        from decimal import Decimal, InvalidOperation
+
+        def dec_or_0(s):
+            try:
+                return Decimal(s)
+            except InvalidOperation:
+                return Decimal(0)
+
+        sorted_results = sorted(self.results, key=lambda r: dec_or_0(r[3]), reverse=True)
+        for email, last_name, first_name, final_result in sorted_results:
+            submission = Submission.objects.get(email=email)
+            teacher_email = submission.contact.contact
+            if options['only_to'] and teacher_email != options['only_to']:
+                continue
+            submissions_by_contact.setdefault(teacher_email, []).append({
+                'first_name': first_name,
+                'last_name': last_name,
+                'final_result': final_result,
+            })
+
+        for email, submissions in submissions_by_contact.items():
+            # contact = Contact.objects.get(id=contact_id)
+            message = render_to_string('wtem/results_teacher.txt', dict(submissions=submissions))
+            self.send_message(message, subject, email)
+
+        self.sum_up()
+
+    def sum_up(self):        
+        self.stdout.write('sent: %s, failed: %s' % (self.sent, self.failed))
+
+    def send_message(self, message, subject, email):
+        self.stdout.write('>>> sending results to %s' % email)
+        try:
+            send_mail(subject=subject, body=message, to=[email])
+        except BaseException, e:
+            self.failed += 1
+            self.stdout.write('failed sending to: ' + email + ': ' + str(e))
+        else:
+            self.sent += 1
+            self.stdout.write('message sent to: ' + email)