Gustavosdo/summum

from django.contrib import admin
from django.contrib.admin.views.main import IS_POPUP_VAR
from django.utils.translation import ugettext_lazy as _

from parametros_financeiros.models import FormaPagamento, GrupoEncargo
from parametros_financeiros.forms import GrupoEncargoForm
from app_global.admin import GlobalAdmin


class FormaPagamentoAdmin(GlobalAdmin):

    model = FormaPagamento
    list_display = ('nome', 'quant_parcelas', 'prazo_entre_parcelas', 'status')
    list_filter = ('status', 'quant_parcelas', 'prazo_entre_parcelas', 'tipo_prazo', 'carencia', 'tipo_carencia',)
    popup_list_filter = ('quant_parcelas', 'prazo_entre_parcelas', 'tipo_prazo', 'carencia', 'tipo_carencia',)
    search_fields = ['nome', 'descricao',]

    fieldsets = (
        (None, {
            'classes': ('suit-tab suit-tab-geral',),
            'fields': ('nome', 'descricao', 'quant_parcelas', 'prazo_entre_parcelas', 'tipo_prazo', 'carencia', 'tipo_carencia', 'status')
        }),
    )

    suit_form_tabs = (
        ('geral', _(u"Geral")),
    )


    def get_readonly_fields(self, request, obj=None):
        """ Define somente alguns campos da forma de pagamento como somente leitura caso o registro seja salvo no BD """

        if obj:
            return ['quant_parcelas', 'prazo_entre_parcelas', 'tipo_prazo', 'carencia', 'tipo_carencia',]
        else:
            return []


    def suit_row_attributes(self, obj, request):
        rowclass = ''
        if not obj.status:
            rowclass = 'error'

        return {'class': rowclass}


    def get_queryset(self, request):
        qs = super(FormaPagamentoAdmin, self).get_queryset(request)
        
        if IS_POPUP_VAR in request.GET:  
            return qs.filter(status=True)
        return qs



class GrupoEncargoAdmin(GlobalAdmin):

    model = GrupoEncargo
    form = GrupoEncargoForm
    list_display = ('nome', 'multa', 'juros', 'tipo_juros', 'status', 'padrao')
    list_filter = ('status',)
    search_fields = ['nome',]

    fieldsets = (
        (None, {
            'classes': ('suit-tab suit-tab-geral',),
            'fields': ('nome', 'multa', 'juros', 'tipo_juros', 'status', 'padrao')
        }),
    )

    suit_form_tabs = (
        ('geral', _(u"Geral")),
    )


    def get_readonly_fields(self, request, obj=None):
        """ Define somente alguns campos da forma de pagamento como somente leitura caso o registro seja salvo no BD """

        if obj:
            return ['multa', 'tipo_juros', 'juros',]
        else:
            return []


    def save_model(self, request, obj, form, change):
        if not obj.multa:
            obj.multa = 0
            
        if not obj.juros:
            obj.juros = 0

        obj.save()


    def suit_row_attributes(self, obj, request):
        rowclass = ''
        if not obj.status:
            rowclass = 'error'

        return {'class': rowclass}


    def get_queryset(self, request):
        qs = super(GrupoEncargoAdmin, self).get_queryset(request)
        
        if IS_POPUP_VAR in request.GET:  
            return qs.filter(status=True)
        return qs


admin.site.register(FormaPagamento, FormaPagamentoAdmin)
admin.site.register(GrupoEncargo, GrupoEncargoAdmin)