from django import forms

from .models import Cuenta, Apunte, Tipo


class CuentaForm(forms.ModelForm):
    class Meta:
        model = Cuenta
        fields = ['nombre', 'saldo_inicial', 'tipo']

    nombre = forms.CharField(
        widget=forms.TextInput(attrs={'class': 'form-control'}))
    saldo_inicial = forms.DecimalField(
        widget=forms.TextInput(attrs={'class': 'form-control'}))
    tipo = forms.ModelChoiceField(
        queryset=Tipo.objects.all(),
        widget=forms.Select(attrs={'class': 'form-control'}))


class ApunteForm(forms.ModelForm):
    class Meta:
        model = Apunte
        fields = ['fecha', 'concepto', 'cta_origen', 'cta_destino', 'importe']

    fecha = forms.DateField(
            widget=forms.DateInput(attrs={'type': 'date', 'class': 'form-control'}))

    concepto = forms.CharField(
        widget=forms.TextInput(attrs={'class': 'form-control'}))

    cta_origen = forms.ModelChoiceField(
        queryset=Cuenta.objects.all(),
        widget=forms.Select(attrs={'class': 'form-control'}))

    cta_destino = forms.ModelChoiceField(
        queryset=Cuenta.objects.all(),
        widget=forms.Select(attrs={'class': 'form-control'}))

    importe = forms.DecimalField(
        widget=forms.NumberInput(attrs={'class': 'form-control'}))