from django_filters import rest_framework as filters
from .models import Task

class TaskFilter(filters.FilterSet):
    servicio = filters.NumberFilter(field_name='servicio__id')
    pedimento_app = filters.CharFilter(field_name='pedimento__pedimento_app')
    pedimento = filters.UUIDFilter(field_name='pedimento__id')
    timestamp_gte = filters.DateTimeFilter(field_name='timestamp', lookup_expr='gte')
    timestamp_lte = filters.DateTimeFilter(field_name='timestamp', lookup_expr='lte')
    status = filters.CharFilter(field_name='status')

    class Meta:
        model = Task
        fields = ['servicio', 'pedimento_app', 'pedimento', 'timestamp_gte', 'timestamp_lte', 'status']