from django_filters import FilterSet
from .models import InvoiceHeader, InvoiceRecord

class InvoiceHeaderFilter(FilterSet):
    class Meta:
        model = InvoiceHeader
        fields = {
            'name': ['icontains', 'exact'],
            'tax_id': ['icontains', 'exact'],
            'address': ['icontains', 'exact'],
            'bank': ['icontains', 'exact'],
            'account': ['icontains', 'exact'],
            'create_time': ['exact', 'lt', 'gt'],
            'update_time': ['exact', 'lt', 'gt'],
        }

class InvoiceRecordFilter(FilterSet):
    class Meta:
        model = InvoiceRecord
        fields = {
            'amount': ['exact', 'lt', 'gt'],
            'date': ['exact', 'lt', 'gt'],
            'purpose': ['icontains', 'exact'],
            'project': ['icontains', 'exact'],
            'actual_paid': ['exact', 'lt', 'gt'],
            'is_reported': ['exact'],
            'create_time': ['exact', 'lt', 'gt'],
            'update_time': ['exact', 'lt', 'gt'],
        }