# vim: set fileencoding=utf-8 ts=4 shiftwidth=4 softtabstop=4 expandtab:
# Copyright (c) 2012 Joaquin Aguirrezabalaga <kinote@kinote.org>

from tastypie.resources import ModelResource
from tastypie import fields
from encuestas.models import Encuesta, Respuesta
from tablon.models import Mensaje


class RespuestaResource(ModelResource):
    class Meta:
        queryset = Respuesta.objects.all()
        resource_name = 'respuesta'
        

class EncuestaResource(ModelResource):
    class Meta:
        queryset = Encuesta.objects.all()
        respuestas = fields.ToManyField(
            RespuestaResource,
            attribute=lambda bundle: Respuesta.objects.filter(
                notes=bundle.obj, name__startswith='Personal'))
        
        resource_name = 'encuesta'
        

class MensajeResource(ModelResource):
    class Meta:
        queryset = Mensaje.objects.all()
        resource_name = 'mensaje'