from django.contrib import admin

from encuesta.models import Pregunta, Respuesta
    
def custom_titled_filter(title):
    class Wrapper(admin.FieldListFilter):
        def __new__(cls, *args, **kwargs):
            instance = admin.FieldListFilter.create(*args, **kwargs)
            instance.title = title
            return instance
    return Wrapper

class PreguntaAdmin(admin.ModelAdmin):
    list_display = ('pregunta_text', 'MediaDeRespuestas')
    search_fields = ['pregunta_text']
class RespuestaAdmin(admin.ModelAdmin):
    date_hierarchy = 'respuesta_creacion_date'
    list_display = ('respuesta_localizador', 'respuesta_media','contestada','respuesta_caducidad_date','respuesta_eliminada','respuesta_aviso_date')
    search_fields = ['book__number']
    list_filter = (('prop__name', custom_titled_filter('Establecimiento')), 'respuesta_eliminada','respuesta_aviso_date')
admin.site.register(Pregunta, PreguntaAdmin)
admin.site.register(Respuesta, RespuestaAdmin)