import django_filters
from .models import Data, DataMs, TbFgPressinfoLotlist, LotSummary

class DataFilter(django_filters.FilterSet):
    lot_no = django_filters.CharFilter(field_name='lot_no', lookup_expr='icontains')
    code = django_filters.CharFilter(field_name='code', lookup_expr='icontains')

    class Meta:
        model = Data
        fields = ['lot_no', 'code']  # Add fields you want to filter

class DataMsFilter(django_filters.FilterSet):
    lot_no = django_filters.CharFilter(field_name='lot_no', lookup_expr='icontains')
    code = django_filters.CharFilter(field_name='code', lookup_expr='icontains')

    class Meta:
        model = DataMs
        fields = ['lot_no', 'code']  # Add fields you want to filter

class TbFgPressFilter(django_filters.FilterSet):
    productcode = django_filters.CharFilter(field_name='productcode', lookup_expr='icontains')
    product_year = django_filters.CharFilter(field_name='product_year', lookup_expr='icontains')

    class Meta:
        model = TbFgPressinfoLotlist
        fields = ['productcode', 'product_year']  # Add fields you want to filter



class LotSummaryFilter(django_filters.FilterSet):
    lot_no = django_filters.CharFilter(field_name='lot_no', lookup_expr='icontains')
    code = django_filters.CharFilter(field_name='code', lookup_expr='icontains')

    class Meta:
        model = LotSummary
        fields = ['lot_no', 'code']  # Add fields you want to filter