Felix Baumann
39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
from django import forms
|
|
from .models import Income, Expense
|
|
from core.forms.base import BaseForm
|
|
from django.utils import timezone
|
|
|
|
class IncomeForm(BaseForm):
|
|
class Meta:
|
|
model = Income
|
|
fields = ["name", "amount", "date", "category", "payment_method", "note"]
|
|
widgets = {
|
|
"date": forms.DateTimeInput(attrs={"type": "datetime-local"}),
|
|
}
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
if not self.instance.pk:
|
|
self.fields["date"].initial = timezone.now()
|
|
|
|
self.apply_style()
|
|
|
|
|
|
class ExpenseForm(BaseForm):
|
|
class Meta:
|
|
model = Expense
|
|
fields = ["name", "amount", "date", "category", "supplier", "payment_method", "receipt", "note"]
|
|
|
|
widgets = {
|
|
"date": forms.DateTimeInput(attrs={"type": "datetime-local"}),
|
|
}
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
if not self.instance.pk:
|
|
self.fields["date"].initial = timezone.now()
|
|
|
|
self.apply_style()
|