from django import forms

from noah.models import Debt, Payment

def validate_code_for_count(code, count):
    #
    if count == 1 and not code == 'foo':
        raise forms.ValidationError("Code invalid for potato count")


class DebtForm(forms.ModelForm):
    class Meta:
        model = Debt

    def clean(self):
        cleaned_data = super(DebtForm, self).clean()
        validate_code_for_count(cleaned_data.get('secret_code'), cleaned_data.get('potato_count'))

        return cleaned_data


class PaymentForm(forms.ModelForm):
    secret_code = forms.CharField(max_length=12)
    potato_count = forms.IntegerField()

    # so you can filter the select queryset for the user who owns the debt or whatever
    def __init__(self, *args, **kwargs):
        super(PaymentForm, self).__init__(*args, **kwargs)
        # just change queryset to whatever
        self.fields['apply_to_debt'] = forms.ModelChoiceField(queryset=Debt.objects.all(), required=False)

    class Meta:
        model = Payment
        exclude = 'debt'

    def clean(self):
        cleaned_data = super(PaymentForm, self).clean()

        if not cleaned_data.get('apply_to_debt'):
            validation_form = DebtForm(cleaned_data)
            if not validation_form.is_valid():
                self._errors.update(validation_form.errors)
            else:
                self.cleaned_data['apply_to_debt'] = validation_form.save()

        return cleaned_data

    def save(self, commit=True):
        payment = super(PaymentForm, self).save(commit=False)
        payment.debt = self.cleaned_data['apply_to_debt']
        payment.save()
        return payment