from django import forms
from django.contrib.auth.forms import UserCreationForm, UserChangeForm

from .models import Cuentas, Apuntes, ReyMotaUser, Tipos


class CuentasForm(forms.ModelForm):
    class Meta:
        model = Cuentas
        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=Tipos.objects.all(),
        widget=forms.TextInput(attrs={'class': 'form-control'}))


class ApuntesForm(forms.ModelForm):
    class Meta:
        model = Apuntes
        fields = ['fecha', 'cta_origen', 'cta_destino', 'importe']

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

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

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

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

class ReyMotaUserCreationForm(UserCreationForm):

    class Meta:
        model = ReyMotaUser
        fields = ("email", "nombre", "foto")
        labels = {'email': 'Dirección de correo'}


class ReyMotaUserChangeForm(UserChangeForm):

    class Meta:
        model = ReyMotaUser
        fields = ("email", "foto")