diff --git a/apps/mtm/views.py b/apps/mtm/views.py index 14fbd5ae..18a6df84 100644 --- a/apps/mtm/views.py +++ b/apps/mtm/views.py @@ -69,7 +69,11 @@ class MaterialViewSet(CustomModelViewSet): 'week_esitimate_consume', None) obj.save() return Response() - + + @action(methods=['get'], detail=False, serializer_class=Serializer, perms_map={'get': '*'}) + def cates(self, request, *args, **kwargs): + res = Material.objects.exclude(cate='').exclude(cate=None).values_list('cate', flat=True).distinct() + return Response(set(res)) class ShiftViewSet(ListModelMixin, CustomGenericViewSet): """