zcdsj
/
factory
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: goal增加查询条件

This commit is contained in:
caoqianming 2023-08-02 17:42:19 +08:00
parent 67d04a9a41
commit 8ab973749d
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
apps/mtm/filters.py Normal file
View File

@ -0,0 +1,13 @@
from django_filters import rest_framework as filters
from apps.mtm.models import Goal
class GoalFilter(filters.FilterSet):
class Meta:
model = Goal
fields = {
"mgroup": ["exact", "in"],
"mgroup__name": ["exact", "in"],
"goal_cate": ["exact", "in"],
"goal_cate__code": ["exact", "in"],
"year": ["exact"]
}

3
apps/mtm/views.py
View File

@ -4,6 +4,7 @@ from rest_framework.mixins import ListModelMixin
from apps.utils.viewsets import CustomGenericViewSet, CustomModelViewSet from apps.utils.viewsets import CustomGenericViewSet, CustomModelViewSet
from apps.mtm.models import Material, Mgroup, Shift, Team, Goal from apps.mtm.models import Material, Mgroup, Shift, Team, Goal
from apps.mtm.serializers import MaterialSerializer, MgroupSerializer, ShiftSerializer, TeamSerializer, GoalSerializer from apps.mtm.serializers import MaterialSerializer, MgroupSerializer, ShiftSerializer, TeamSerializer, GoalSerializer
from apps.mtm.filters import GoalFilter
# Create your views here. # Create your views here.
class MaterialViewSet(CustomModelViewSet): class MaterialViewSet(CustomModelViewSet):
@ -67,5 +68,5 @@ class GoalViewSet(CustomModelViewSet):
queryset = Goal.objects.all() queryset = Goal.objects.all()
serializer_class = GoalSerializer serializer_class = GoalSerializer
select_related_fields = ['mgroup', 'goal_cate'] select_related_fields = ['mgroup', 'goal_cate']
filterset_fields = ['mgroup', 'goal_cate', 'year', 'goal_cate__code'] filterset_class = GoalFilter
search_fields = ['name'] search_fields = ['name']