accounting/expenses/forms.py

from django import forms
from django.contrib.auth.models import User
from django.core.exceptions import ValidationError

from expenses.models import Category, Expense, MultiplePaymentExepense, LoneExpense, RawExpense


class CategoryForm(forms.ModelForm):
    class Meta:
        model = Category
        fields = "__all__"

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)


class ExpenseForm(forms.ModelForm):
    class Meta:
        model = Expense
        fields = "__all__"

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)


class MultiplePaymentExepenseForm(forms.ModelForm):
    class Meta:
        model = MultiplePaymentExepense
        fields = ("name", "first_payment_date", "amount", "number_of_payment", "category", "source")

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)


class SubExpenseForm(forms.ModelForm):
    class Meta:
        model = RawExpense
        fields = ("amount",)
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)



class LoneExpenseForm(forms.ModelForm):
    class Meta:
        model = LoneExpense
        fields = "__all__"

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)