# -*- coding: utf-8 -*- from django_filters.rest_framework import filters from django_filters.rest_framework.filterset import FilterSet from .models import Task class TaskFilterSet(FilterSet): name = filters.CharFilter(field_name="name", lookup_expr="icontains", label="任务名称") task_type = filters.NumberFilter(field_name="task_type", lookup_expr="exact", label="任务类型") class Meta: model = Task fields = ( "name", "task_type" )
class TaskViewset(ModelViewSet): filter_class = TaskFilterSet