path: root/webapp/sociologist/views.py

# -*- coding: utf-8 -*-
from bson import ObjectId, DBRef

from django.views.generic.list import ListView
from django.views.generic.edit import FormView
from django.http import HttpResponseRedirect, Http404
from django.core.urlresolvers import reverse
from django.contrib import messages
from django.template.loader import render_to_string

from utils.mongo_connection import get_db
from utils.forms import BadFormValidation

from pollster.models import Pollster
from polls.models import Poll, Option
from pollster.forms import PollsterForm, PollAsignationForm
from sociologist.forms import SociologistForm


__all__ = [
    'PollsterListView',
    'PollsterFormView',
    'PollAsignationFormView',
    'PollResultListView',
    'SociologistFormView',
]


# TODO: Refactoring!
class PollResultListView(ListView):

    template_name = "poll-result-list.html"
    context_object_name = "results"

    def dispatch(self, *args, **kwargs):

        self.poll_type = "monitoreo"
        self.do_no_respond = []

        # If id and Wrong id or poll is not open => 404
        _id = kwargs.get('poll_id', None)
        self.poll = Poll.get(id=_id) if _id else Poll()
        if _id and not self.poll:
            raise Http404()

        return super(PollResultListView, self).dispatch(*args, **kwargs)

    def get_context_data(self, **kwargs):
        context = super(PollResultListView, self).get_context_data(
            **kwargs)

        column_names = [
            {'ng_class': "polled_name", 'label': "Encuestado"},
            {'ng_class': "completed_questions", 'classes': "hide"},
            {'ng_class': "polled_escuela", 'label': "Escuela"},
            {'ng_class': "polled_grado", 'label': "Grado"},
            {'ng_class': "polled_grupo", 'label': "Grupo"},
        ]
        searchable_fields = [
            'polled_name',
            'completed_questions',
            'polled_escuela',
            'polled_grado',
            'polled_grupo',
        ]

        sorting_order = "polled_name"

        if self.poll_type == "general":
            searchable_fields[0] = 'polled_id'
            column_names[0] = {
                'ng_class': "polled_id", 'classes': "", 'label': "Encuesta ID"
            }
            sorting_order = "polled_grado"

        context.update({
            'poll': self.poll,
            'do_no_respond': self.do_no_respond,
            'searchable_fields': searchable_fields,
            'column_names': column_names,
            'sorting_order': sorting_order
        })

        return context

    def get_queryset(self, *args, **kwargs):
        _list = []

        poll_result = self.poll.get_result()

        questions = map(Poll.clean_question, self.poll.get_questions())

        self.poll_type = poll_result.to_python()['poll_type']

        index = 0
        for id_polled, data in poll_result.to_python()['result'].iteritems():
            item = {
                "answers": {},
                "polled_name": data['polled'].get('NOMBRE', index),
                "polled_escuela": data['polled']['NUM_ESC'],
                "polled_grado": data['polled']['GRADO'],
                "polled_grupo": data['polled']['GRUPO']
            }

            if self.poll_type == "general":
                del item['polled_name']
                item.update({
                    "polled_id": "Encuesta #%s" % index
                })

            answers = data['answers']
            if len(answers):
                completed_questions = []
                for group_order, fields in answers.iteritems():
                    fields = fields['fields']
                    for field_order, field_data in fields.iteritems():
                        _answers = []
                        _weight = 0
                        answer = field_data['answer']
                        for option_id, option_selected in answer.iteritems():
                            if "Image" in field_data['widget_type']:
                                img_src, media_url = Option.get_img(
                                    str(self.poll.id), option_id)
                                img = render_to_string(
                                    'image_option_thumbnail.html',
                                    {"img": open(img_src)}
                                )
                                _answers.append(img)
                            else:
                                _answers.append(
                                    option_selected.get('text', ""))
                            _weight += option_selected.get('weight', 0)

                        if len(_answers):
                            completed_questions.append(field_data['name'])
                            field_name = Poll.clean_question(
                                field_data['name'])
                            item['answers'].update({
                                field_name: {
                                    "weight": str(_weight),
                                    "options": ", ".join(_answers)
                                }
                            })

                        item['completed_questions'] = " ".join(
                            completed_questions)

                for field_name in questions:
                    if field_name not in item['answers'].keys():
                        item['answers'].update({
                            field_name: {
                                "weight": "-",
                                "options": "No responde."
                            }
                        })

                _list.append(item)
                index += 1
            elif not len(answers) and self.poll_type == "monitoreo":
                self.do_no_respond.append({
                    "polled_name": data['polled'].get('NOMBRE', index),
                    "polled_escuela": data['polled']['NUM_ESC'],
                    "polled_grado": data['polled']['GRADO'],
                    "polled_grupo": data['polled']['GRUPO']
                })

        return _list


class PollsterListView(ListView):

    template_name = "pollster-list.html"
    context_object_name = "pollsters"

    def get_queryset(self, *args, **kwargs):
        return [
            {
                'username': pe.get_user().username,
                'email': pe.get_user().email,
                'action_assign': reverse(
                    'sociologist:poll_assignment', kwargs={'id': str(pe.id)})
            } for pe in Pollster.all()
        ]


class PollsterFormView(FormView):

    template_name = "pollster-form.html"
    form_class = PollsterForm

    def form_valid(self, form):
        try:
            form.save()
        except Exception:
            msg = u'El encuestador no se pudo guardar.'
            messages.add_message(self.request, messages.ERROR, msg)
        else:
            msg = u'El encuestador fué guardado con éxito.'
            messages.add_message(self.request, messages.SUCCESS, msg)

            return HttpResponseRedirect(reverse('sociologist:pollster_list'))

        return self.form_invalid(form)


class SociologistFormView(FormView):

    template_name = "sociologist-form.html"
    form_class = SociologistForm

    def dispatch(self, *args, **kwargs):

        if not args[0].user.is_superuser:
            raise Http404()

        return super(SociologistFormView, self).dispatch(*args, **kwargs)

    def form_valid(self, form):
        try:
            form.save()
        except Exception:
            msg = u'El investigador no se pudo guardar.'
            messages.add_message(self.request, messages.ERROR, msg)
        else:
            msg = u'El investigador fué guardado con éxito.'
            messages.add_message(self.request, messages.SUCCESS, msg)

            return HttpResponseRedirect(reverse('index'))

        return self.form_invalid(form)


class PollAsignationFormView(FormView):

    template_name = "poll-assignment-form.html"
    form_class = PollAsignationForm
    pollster = None

    def dispatch(self, *args, **kwargs):

        _id = kwargs.get('id', None)
        self.pollster = Pollster.get(id=_id) if _id else None
        if _id and not self.pollster:
            raise Http404()

        return super(PollAsignationFormView, self).dispatch(*args, **kwargs)

    def get_form_kwargs(self, *args, **kwargs):
        args_ = super(
            PollAsignationFormView, self).get_form_kwargs(*args, **kwargs)
        args_.update({'pollster_id': str(self.pollster.id)})
        return args_

    def get_context_data(self, **kwargs):
        context = super(PollAsignationFormView, self).get_context_data(
            **kwargs)

        context.update({
            'pollster': self.pollster,
            'assigned_to_pollster': Poll.assigned_to_pollster(self.pollster)
        })

        return context

    def form_valid(self, form):
        try:
            form.assign()
        except BadFormValidation:
            msg = u'Ocurrió un error, no fueron asignadas.'
            messages.add_message(self.request, messages.ERROR, msg)
            return self.render_to_response(self.get_context_data(form=form))
            # TODO: Logear
        except Exception:
            pass
            # TODO: Logear
        else:
            msg = u'Se realizaron las asignaciones correctamente.'
            messages.add_message(self.request, messages.SUCCESS, msg)

            return HttpResponseRedirect(reverse('sociologist:pollster_list'))

        return self.form_invalid(form)


def pollster_detach(request, poll_id, pollster_id):
    dbref = DBRef(Pollster.collection_name, ObjectId(pollster_id))
    get_db().polls.update(
        {"_id": ObjectId(poll_id), "pollsters": dbref},
        {"$pull": {"pollsters": dbref}},
    )
    return HttpResponseRedirect(
        reverse('sociologist:poll_assignment', kwargs={'id': pollster_id})
    )


def poll_clone(request, id):
    try:
        poll = Poll.get(id=id)
        poll_clone = poll.clone()
        msg = u'Se clonó la encuesta: %s a %s' % (poll.name, poll_clone.name)
        messages.add_message(request, messages.SUCCESS, msg)
    except:
        msg = u'No se pudo clonar la encuesta.'
        messages.add_message(request, messages.ERROR, msg)

    return HttpResponseRedirect(reverse('sociologist:poll_list'))


def poll_delete(request, id):
    try:
        poll = Poll.get(id=id)
        poll_name = poll.name
        poll.delete()

        msg = u'La encuesta %s fué eliminada.' % poll_name
        messages.add_message(request, messages.SUCCESS, msg)
    except:
        msg = u'La encuesta no pudo ser eliminada.'
        messages.add_message(request, messages.ERROR, msg)

    return HttpResponseRedirect(reverse('sociologist:poll_list'))