from django_filters import rest_framework as filters

from LiveStock.models import LiveStock


class LiveStockFilterSet(filters.FilterSet):
    class Meta:
        model = LiveStock
        fields = [
            'national_id_livestock_code',
            'herd_code',
            'type',
            'gender',
            'contractor_code',
            'unique_identifier',
            'agent',
            'registering_user',
            'rancher__user__mobile',
            'rancher__user__first_name',
            'rancher__user__last_name',
            'rancher__user__fullname'
        ]
