# -*- 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)