# coding: utf-8

from django.contrib import admin
from .models import Dossier, Proposal, Vote


class DossierAdmin(admin.ModelAdmin):
    list_display = ('id', 'fingerprint', 'reference', 'title', 'link')
    search_fields = ('reference', 'title', 'fingerprint')


class ProposalAdmin(admin.ModelAdmin):
    list_display = ('id', 'fingerprint', 'reference', 'dossier_reference', 'title', 'datetime', 'kind', 'total_abstain', 'total_against', 'total_for')
    search_fields = ('reference', 'dossier__reference', 'title', 'fingerprint')
    def dossier_reference(self, obj):
        return obj.dossier.reference


class NoneMatchingFilter(admin.SimpleListFilter):
    title = 'Representative'
    parameter_name = 'representative'

    def lookups(self, request, model_admin):
        return [('None', 'Unknown')]
    
    def queryset(self, request, queryset):
        if self.value() == 'None':
            return queryset.filter(representative=None)
        else:
            return queryset            


class VoteAdmin(admin.ModelAdmin):
    list_display = ('id', 'proposal_reference', 'position', 'representative', 'representative_name')
    list_filter = (NoneMatchingFilter,)
    
    def proposal_reference(self, obj):
        return obj.proposal.reference

admin.site.register(Dossier, DossierAdmin)
admin.site.register(Proposal, ProposalAdmin)
admin.site.register(Vote, VoteAdmin)