# -*- coding: utf-8 -*-
from django.contrib import admin
from sigi.apps.contatos.models import (UnidadeFederativa, Municipio, Telefone,
                                       Contato)
from sigi.apps.utils import queryset_ascii

class UnidadeFederativaAdmin(admin.ModelAdmin):
    actions = None
    list_display = ('codigo_ibge', 'nome', 'sigla', 'regiao', 'populacao')
    list_display_links = ('codigo_ibge', 'nome')
    list_filter = ('regiao', 'populacao')
    search_fields = ('search_text', 'codigo_ibge',  'sigla', 'regiao')
    queryset = queryset_ascii

class MunicipioAdmin(admin.ModelAdmin):
    actions = None
    list_display = ('codigo_ibge', 'codigo_tse', 'nome', 'uf', 'is_capital', 'populacao', 'is_polo')
    list_display_links = ('codigo_ibge', 'codigo_tse', 'nome')
    list_filter = ('is_capital', 'is_polo', 'populacao', 'uf')
    queryset = queryset_ascii
    fieldsets = (
        (None, {
            'fields': ('codigo_ibge', 'codigo_tse', 'codigo_mesorregiao',
                       'codigo_microrregiao', 'nome', 'uf', 'is_capital',
                       'populacao', 'is_polo')
        }),
        ('Posição geográfica', {
            'fields': ('latitude', 'longitude'),
        }),
    )
    search_fields = ('search_text', 'codigo_ibge', 'codigo_tse', 'codigo_mesorregiao',
                     'codigo_microrregiao', 'uf__sigla')

class TelefoneAdmin(admin.ModelAdmin):
    list_display = ('numero', 'tipo', 'nota')
    list_display_links = ('numero',)
    list_filter = ('tipo',)
    radio_fields = {'tipo': admin.VERTICAL}
    search_fields = ('numero', 'tipo', 'nota')

class ContatoAdmin(admin.ModelAdmin):
    list_display = ('nome', 'nota', 'email', 'municipio')
    list_display_links = ('nome',)
    list_filter = ('nome',)
    search_fields = ('nome', 'nota', 'email', 'municipio__nome', 'municipio__uf__nome')

admin.site.register(UnidadeFederativa, UnidadeFederativaAdmin)
admin.site.register(Municipio, MunicipioAdmin)
admin.site.register(Telefone, TelefoneAdmin)
admin.site.register(Contato, ContatoAdmin)